Update weipu api

PaddlePaddle · Feb 22, 2024 · ddceb1e · ddceb1e
1 parent 4aca819
commit ddceb1e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py b/erniebot-agent/src/erniebot_agent/chat_models/erniebot.py
@@ -328,10 +328,10 @@ def convert_response_to_output(response: ChatCompletionResponse, output_type: Ty
         )
     else:
         return output_type(
-            content=response.result,
+            content=response.rbody,
             function_call=None,
             plugin_info=None,
             search_info=None,
-            token_usage=response.usage,
+            token_usage=None,
             clarify=clarify,
         )
diff --git a/erniebot/src/erniebot/backends/aistudio.py b/erniebot/src/erniebot/backends/aistudio.py
@@ -91,7 +91,7 @@ async def arequest(
 
     @classmethod
     def handle_response(cls, resp: EBResponse) -> EBResponse:
-        if resp["errorCode"] != 0:
+        if "errorCode" in resp and resp["errorCode"] != 0:
             ecode = resp["errorCode"]
             emsg = resp["errorMsg"]
             if ecode in (4, 17):
@@ -117,5 +117,6 @@ def _add_aistudio_fields_to_headers(self, headers: HeadersType) -> HeadersType:
                 "Key 'Authorization' already exists in `headers`: %r",
                 headers["Authorization"],
             )
-        headers["Authorization"] = f"token {self._access_token}"
+        # headers["Authorization"] = f"token {self._access_token}"
+        headers["Authorization"] = f"{self._access_token}"
         return headers