瀏覽代碼

Update api_server: logging

StrayWarrior 1 周之前
父節點
當前提交
c5ba2a4b95
共有 1 個文件被更改,包括 4 次插入1 次删除
  1. 4 1
      api_server.py

+ 4 - 1
api_server.py

@@ -182,8 +182,11 @@ def run_extractor_prompt(req_data):
         if function_call.function.name == 'update_user_profile':
             profile_info = json.loads(function_call.function.arguments)
             return {k: v for k, v in profile_info.items() if v}
+        else:
+            logger.error("llm does not return update_user_profile")
+            return {}
     else:
-        raise Exception("llm does not return tool_calls")
+        raise {}
 
 def run_chat_prompt(req_data):
     prompt = req_data['prompt']