fix: thought param in /api/answer

This commit is contained in:
Alex
2025-04-07 11:03:49 +01:00
parent bea89b93eb
commit f6c88da81b

View File

@@ -611,6 +611,7 @@ class Answer(Resource):
source_log_docs = [] source_log_docs = []
tool_calls = [] tool_calls = []
stream_ended = False stream_ended = False
thought = ""
for line in complete_stream( for line in complete_stream(
question=question, question=question,
@@ -633,6 +634,8 @@ class Answer(Resource):
source_log_docs = event["source"] source_log_docs = event["source"]
elif event["type"] == "tool_calls": elif event["type"] == "tool_calls":
tool_calls = event["tool_calls"] tool_calls = event["tool_calls"]
elif event["type"] == "thought":
thought = event["thought"]
elif event["type"] == "error": elif event["type"] == "error":
logger.error(f"Error from stream: {event['error']}") logger.error(f"Error from stream: {event['error']}")
return bad_request(500, event["error"]) return bad_request(500, event["error"])
@@ -664,6 +667,7 @@ class Answer(Resource):
conversation_id, conversation_id,
question, question,
response_full, response_full,
thought,
source_log_docs, source_log_docs,
tool_calls, tool_calls,
llm, llm,