restore erniebot

PaddlePaddle · Feb 26, 2024 · d0fdbff · d0fdbff
1 parent e14086b
commit d0fdbff
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 15 deletions.
diff --git a/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py b/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py
@@ -328,20 +328,20 @@ def convert_response_to_output(response: ChatCompletionResponse, output_type: Ty
         )
     else:
 
-        # return output_type(
-        #     content=response.result,
-        #     function_call=None,
-        #     plugin_info=None,
-        #     search_info=None,
-        #     token_usage=response.usage,
-        #     clarify=clarify,
-        # )
-
         return output_type(
-            content=response.rbody,
+            content=response.result,
             function_call=None,
             plugin_info=None,
             search_info=None,
-            token_usage=None,
+            token_usage=response.usage,
             clarify=clarify,
         )
+
+        # return output_type(
+        #     content=response.rbody,
+        #     function_call=None,
+        #     plugin_info=None,
+        #     search_info=None,
+        #     token_usage=None,
+        #     clarify=clarify,
+        # )
diff --git a/erniebot/src/erniebot/backends/bce.py b/erniebot/src/erniebot/backends/bce.py
@@ -337,7 +337,7 @@ def handle_response(cls, resp: EBResponse) -> EBResponse:
             else:
                 raise errors.APIError(emsg, ecode=ecode)
         else:
-            return EBResponse(resp.rcode, resp.result, resp.rheaders)
+            return resp
 
 
 class YinianBackend(_BCELegacyBackend):

diff --git a/erniebot/src/erniebot/resources/chat_completion.py b/erniebot/src/erniebot/resources/chat_completion.py
@@ -84,9 +84,6 @@ class ChatCompletion(EBResource, CreatableWithStreaming):
                 "ernie-longtext": {
                     "model_id": "ernie_bot_8k",
                 },
-                "ernie-speed": {
-                    "model_id": "ernie_speed",
-                },
             },
         },
         APIType.CUSTOM: {