Merge pull request #74 from MetaGLM/feature/agent-api-0622

Feature/agent api 0622

Author: haiyang679

tests/agent/glm3_agent.py

@@ -0,0 +1,66 @@
+import os.path
+
+from zhipuai import ZhipuAI
+import zhipuai
+import time
+
+import logging
+import logging.config
+
+
+def test_completions_sync(logging_conf):
+    logging.config.dictConfig(logging_conf)  # type: ignore
+    client = ZhipuAI()  # 填写您自己的APIKey
+    try:
+        # 生成request_id
+        request_id = time.time()
+        print(f"request_id:{request_id}")
+        response = client.agents.invoke(
+            request_id=request_id,
+            agent_id="general_translation",
+            messages=[
+                {
+                    "role": "user",
+                    "content": "tell me a joke"
+                }
+            ],
+            user_id="12345678"
+        )
+        print(response)
+
+    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)
+
+def test_completions_stream(logging_conf):
+    logging.config.dictConfig(logging_conf)  # type: ignore
+    client = ZhipuAI()  # 填写您自己的APIKey
+    try:
+        # 生成request_id
+        request_id = time.time()
+        print(f"request_id:{request_id}")
+        response = client.agents.invoke(
+            request_id=request_id,
+            agent_id="general_translation",
+            messages=[
+                {
+                    "role": "user",
+                    "content": "tell me a joke"
+                }
+            ],
+            user_id="12345678",
+            stream=True
+        )
+        for item in response:
+            print(item)
+
+
+    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)

tests/agent/test_agent.py

@@ -68,6 +68,7 @@ def __init__(
         self.web_search = api_resource.WebSearchApi(self)
         self.audio = api_resource.audio.Audio(self)
         self.moderation = api_resource.moderation.Moderation(self)
+        self.agents = api_resource.agents.Agents(self)
 
     @property
     @override

tests/agent/test_create_structured_funcation_call.py

@@ -45,6 +45,10 @@
     WebSearchApi
 )
 
+from .agents import (
+    Agents
+)
+
 __all__ = [
     'Videos',
     'AsyncCompletions',

tests/agent/test_prompt_parser_agent.py

@@ -0,0 +1,5 @@
+from zhipuai.api_resource.agents.agents import Agents
+
+__all__= [
+    "Agents"
+]