浏览代码

Fixup API Server

StrayWarrior 1 月之前
父节点
当前提交
e7243ea962
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      api_server.py

+ 1 - 1
api_server.py

@@ -246,7 +246,7 @@ def run_prompt():
             return wrap_response(200, data=response)
             return wrap_response(200, data=response)
         elif scene == 'response_type_detector':
         elif scene == 'response_type_detector':
             response = run_response_type_prompt(req_data)
             response = run_response_type_prompt(req_data)
-            return wrap_response(200, data=response)
+            return wrap_response(200, data=response.choices[0].message.content)
         else:
         else:
             response = run_chat_prompt(req_data)
             response = run_chat_prompt(req_data)
             return wrap_response(200, data=response.choices[0].message.content)
             return wrap_response(200, data=response.choices[0].message.content)