소스 검색

Fixup API Server

StrayWarrior 2 달 전
부모
커밋
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)
         elif scene == 'response_type_detector':
             response = run_response_type_prompt(req_data)
-            return wrap_response(200, data=response)
+            return wrap_response(200, data=response.choices[0].message.content)
         else:
             response = run_chat_prompt(req_data)
             return wrap_response(200, data=response.choices[0].message.content)