mirror of
https://github.com/ianarawjo/ChainForge.git
synced 2025-03-14 08:16:37 +00:00
41 lines
1.7 KiB
Python
41 lines
1.7 KiB
Python
import argparse
|
|
from chainforge.flask_app import run_server
|
|
|
|
# Main Chainforge start
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Chainforge command line tool')
|
|
|
|
# Serve command
|
|
subparsers = parser.add_subparsers(dest='serve')
|
|
serve_parser = subparsers.add_parser('serve', help='Start Chainforge server')
|
|
|
|
# TODO: Add this back
|
|
# Turn on to disable all outbound LLM API calls and replace them with dummy calls
|
|
# that return random strings of ASCII characters. Useful for testing interface without wasting $$.
|
|
# serve_parser.add_argument('--dummy-responses',
|
|
# help="""Disables queries to LLMs, replacing them with spoofed responses composed of random ASCII characters.
|
|
# Produces each dummy response at random intervals between 0.1 and 3 seconds.""",
|
|
# dest='dummy_responses',
|
|
# action='store_true')
|
|
|
|
# TODO: Reimplement this where the React server is given the backend's port before loading.
|
|
serve_parser.add_argument('--port', help='The port to run the server on. Defaults to 8000.', type=int, default=8000, nargs='?')
|
|
serve_parser.add_argument('--host',
|
|
help="The host to run the server on. Defaults to 'localhost'.",
|
|
type=str, default="localhost", nargs='?')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Currently only support the 'serve' command...
|
|
if not args.serve:
|
|
parser.print_help()
|
|
exit(0)
|
|
|
|
port = args.port if args.port else 8000
|
|
host = args.host if args.host else "localhost"
|
|
|
|
print(f"Serving Flask server on {host} on port {port}...")
|
|
run_server(host=host, port=port, cmd_args=args)
|
|
|
|
if __name__ == "__main__":
|
|
main() |