This commit is contained in:
Timothy J. Baek 2024-06-11 00:24:25 -07:00
parent 3d6f5f418d
commit 049b3136e8

View File

@ -284,6 +284,7 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
context = ""
for tool_id in data["tool_ids"]:
print(tool_id)
response = await get_function_call_response(
prompt=prompt,
tool_id=tool_id,
@ -291,10 +292,9 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
task_model_id=task_model_id,
user=user,
)
print(response)
if response:
context = ("\n" if context != "" else "") + response
context += ("\n" if context != "" else "") + response
if context != "":
system_prompt = rag_template(