add ernie-turbo

PaddlePaddle · Jan 11, 2024 · 540c8b6 · 540c8b6
1 parent 0c8a796
commit 540c8b6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py b/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py
@@ -184,6 +184,7 @@ async def chat(
         cfg_dict = self._generate_config(messages, functions=functions, **kwargs)
 
         response = await self._generate_response(cfg_dict, stream, functions)
+
         if not stream:
             assert isinstance(response, ChatCompletionResponse)
             return convert_response_to_output(response, AIMessage)
@@ -279,8 +280,8 @@ async def _generate_response(
 
 def convert_response_to_output(response: ChatCompletionResponse, output_type: Type[_T]) -> _T:
     clarify = False
-    if response["finish_reason"] == "plugin_clarify":
-        # clarify would not occur in function call
+    # ernie-turbo has no `finish_reason`
+    if hasattr(response, "finish_reason") and response["finish_reason"] == "plugin_clarify":
         clarify = True
 
     if hasattr(response, "function_call"):