From c995d2a47ce4b95e5fbe088d705ab5aa773caf7a Mon Sep 17 00:00:00 2001 From: martin legrand Date: Sat, 29 Mar 2025 12:13:03 +0100 Subject: [PATCH] fix app.py problem for server --- server/app.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/server/app.py b/server/app.py index 85abcc8..f928cd7 100644 --- a/server/app.py +++ b/server/app.py @@ -12,12 +12,15 @@ log.setLevel(logging.ERROR) parser = argparse.ArgumentParser(description='AgenticSeek server script') parser.add_argument('--provider', type=str, help='LLM backend library to use. set to [ollama] or [llamacpp]', required=True) +parser.add_argument('--port', type=int, help='port to use', required=True) args = parser.parse_args() app = Flask(__name__) generator = None +assert args.provider in ["ollama", "llamacpp"], f"Provider {args.provider} does not exists. see --help for more information" + @app.route('/generate', methods=['POST']) def start_generation(): if generator is None: @@ -49,4 +52,4 @@ def get_updated_sentence(): return generator.get_status() if __name__ == '__main__': - app.run(host='0.0.0.0', threaded=True, debug=True, port=3333) \ No newline at end of file + app.run(host='0.0.0.0', threaded=True, debug=True, port=args.port) \ No newline at end of file