【翻译api】翻译智能体支持

Author: jinhaiyang

tests/integration_tests/test_assistant.py

@@ -3,7 +3,7 @@
 import logging.config
 import zhipuai
 from zhipuai import ZhipuAI
-
+import json
 
 def test_assistant(logging_conf) -> None:
     logging.config.dictConfig(logging_conf)  # type: ignore
@@ -73,3 +73,40 @@ def test_assistant_query_conversation_usage(logging_conf) -> None:
         print(err)
     except zhipuai.core._errors.APIStatusError as err:
         print(err)
+
+
+def test_translate_api(logging_conf) -> None:
+    logging.config.dictConfig(logging_conf)  # type: ignore
+    client = ZhipuAI()  # 填写您自己的APIKey
+    try:
+        generate = client.assistant.conversation(
+            assistant_id="9996ijk789lmn012o345p999",
+            model="glm-4-assistant",
+            messages=[
+                {
+                    "role": "user",
+                    "content": [{
+                        "type": "text",
+                        "text": "你好呀"
+                    }]
+                }
+            ],
+            stream=False,
+            attachments=None,
+            metadata=None,
+            request_id="request_1790291013237211136",
+            user_id="12345678",
+            extra_parameters = {
+                "translate": {
+                    "from": "zh",
+                    "to": "en"
+                }
+            }
+        )
+        print(generate)
+    except zhipuai.core._errors.APIRequestFailedError as err:
+        print(err)
+    except zhipuai.core._errors.APIInternalError as err:
+        print(err)
+    except zhipuai.core._errors.APIStatusError as err:
+        print(err)

zhipuai/api_resource/assistant/assistant.py

@@ -36,16 +36,17 @@ def conversation(
             model: str,
             messages: List[assistant_create_params.ConversationMessage],
             *,
-            stream: bool = True,
+            stream: Optional[Literal[False]] | Literal[True] | NotGiven = NOT_GIVEN,
             conversation_id: Optional[str] = None,
             attachments: Optional[List[assistant_create_params.AssistantAttachments]] = None,
             metadata: dict | None = None,
             request_id: str = None,
             user_id: str = None,
+            extra_parameters: Optional[assistant_create_params.ExtraParameters] = None,
             extra_headers: Headers | None = None,
             extra_body: Body | None = None,
             timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> StreamResponse[AssistantCompletion]:
+    ) -> AssistantCompletion | StreamResponse[AssistantCompletion]:
         body = deepcopy_minimal(
             {
                 "assistant_id": assistant_id,
@@ -57,6 +58,7 @@ def conversation(
                 "metadata": metadata,
                 "request_id": request_id,
                 "user_id": user_id,
+                "extra_parameters": extra_parameters
             }
         )
         return self._post(
@@ -66,7 +68,7 @@ def conversation(
                 extra_headers=extra_headers, extra_body=extra_body, timeout=timeout
             ),
             cast_type=AssistantCompletion,
-            stream=stream or True,
+            stream=stream or False,
             stream_cls=StreamResponse[AssistantCompletion],
         )
 

zhipuai/types/assistant/assistant_create_params.py

@@ -28,3 +28,11 @@ class AssistantParameters(TypedDict, total=False):
     messages: List[ConversationMessage]  # 会话消息体
     attachments: Optional[List[AssistantAttachments]]  # 会话指定的文件，非必填
     metadata: Optional[dict]  # 元信息，拓展字段，非必填
+
+class TranslateParameters(TypedDict, total=False):
+    from_language: str
+    to_language: str
+
+class ExtraParameters(TypedDict, total=False):
+    translate: TranslateParameters
+