tcsenpai/agenticSeek
1
0
Fork 0
mirror of https://github.com/tcsenpai/agenticSeek.git synced 2025-06-07 03:25:32 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix app.py problem for server

Browse Source
This commit is contained in:
martin legrand 2025-03-29 12:13:03 +01:00
parent de76fe14ea
commit c995d2a47c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
server/app.py
View File

@ -12,12 +12,15 @@ log.setLevel(logging.ERROR)
parser = argparse.ArgumentParser(description='AgenticSeek server script') 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('--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() args = parser.parse_args()
app = Flask(__name__) app = Flask(__name__)
generator = None 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']) @app.route('/generate', methods=['POST'])
def start_generation(): def start_generation():
if generator is None: if generator is None:
@ -49,4 +52,4 @@ def get_updated_sentence():
return generator.get_status() return generator.get_status()
if __name__ == '__main__': if __name__ == '__main__':
app.run(host='0.0.0.0', threaded=True, debug=True, port=3333) app.run(host='0.0.0.0', threaded=True, debug=True, port=args.port)