mirror of
https://github.com/tcsenpai/agenticSeek.git
synced 2025-06-06 11:05:26 +00:00
fix : response timeout
This commit is contained in:
parent
5c9ada9468
commit
90894f806a
@ -36,6 +36,17 @@ def setup():
|
||||
generator.set_model(model)
|
||||
return jsonify({"message": "Model set"}), 200
|
||||
|
||||
@app.route('/get_complete_sentence', methods=['GET'])
|
||||
def get_complete_sentence():
|
||||
if not generator:
|
||||
return jsonify({"error": "Generator not initialized"}), 400
|
||||
while True:
|
||||
status = generator.get_status()
|
||||
print(status)
|
||||
if status["is_complete"]:
|
||||
return jsonify(status)
|
||||
return None
|
||||
|
||||
@app.route('/get_updated_sentence')
|
||||
def get_updated_sentence():
|
||||
if not generator:
|
||||
|
@ -118,7 +118,7 @@ class Provider:
|
||||
requests.post(route_gen, json={"messages": history})
|
||||
is_complete = False
|
||||
while not is_complete:
|
||||
response = requests.get(f"http://{self.server_ip}/get_updated_sentence")
|
||||
response = requests.get(f"http://{self.server_ip}/get_complete_sentence")
|
||||
if "error" in response.json():
|
||||
pretty_print(response.json()["error"], color="failure")
|
||||
break
|
||||
|
Loading…
x
Reference in New Issue
Block a user