mirror of
https://github.com/ianarawjo/ChainForge.git
synced 2025-03-17 09:35:49 +00:00
36 lines
1.4 KiB
Python
36 lines
1.4 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')
|
|
|
|
# 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='?')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Currently only support the 'serve' command...
|
|
if not args.serve:
|
|
parser.print_help()
|
|
exit(0)
|
|
|
|
port = 8000 # args.port if args.port else 8000
|
|
|
|
print(f"Serving Flask server on port {port}...")
|
|
run_server(host="localhost", port=port, cmd_args=args)
|
|
|
|
if __name__ == "__main__":
|
|
main() |