Update version to 2.1.5.20250724 in pyproject.toml and __version__.py; refactor AgentsError class to inherit from BaseModel; add new tests for correction functionality in test_agents.py.

Author: zhengweijun

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zhipuai"
-version = "2.1.5.20250716"
+version = "2.1.5.20250724"
 description = "A SDK library for accessing big model apis from ZhipuAI"
 authors = ["Zhipu AI"]
 readme = "README.md"
@@ -45,15 +45,13 @@ cli = ["typer"]
 # An extra used to be able to add extended testing.
 # Please use new-line on formatting to make it easier to add new packages without
 # merge-conflicts
-extended_testing = [
-    "langchain",
-]
+extended_testing = ["langchain"]
 
 [tool.ruff.lint]
 select = [
-    "E", # pycodestyle
-    "F", # pyflakes
-    "I", # isort
+    "E",    # pycodestyle
+    "F",    # pyflakes
+    "I",    # isort
     "T201", # print
 ]
 
@@ -66,9 +64,7 @@ indent-style = "tab"
 docstring-code-format = true
 
 [tool.coverage.run]
-omit = [
-    "tests/*",
-]
+omit = ["tests/*"]
 
 [build-system]
 requires = ["poetry-core>=1.0.0", "poetry-plugin-pypi-mirror==0.4.2"]

tests/integration_tests/test_agents.py

@@ -52,3 +52,46 @@ def test_completions_stream(logging_conf):
 		print(err)
 	except zhipuai.core._errors.APIStatusError as err:
 		print(err)
+
+def test_correction():
+	client = ZhipuAI()  # 请替换为实际API密钥
+
+	response = client.agents.invoke(
+		agent_id="intelligent_education_correction_agent",
+		messages=[
+				{
+					"role": "user",
+					"content": [
+						{
+							"type": "image_url",
+							"image_url": "https://b0.bdstatic.com/e24937f1f6b9c0ff6895e1012c981515.jpg"
+						}
+					]
+				}
+			]
+		)
+	print(response)
+ 
+def test_correction_result(image_id,uuids,trace_id):
+	client = ZhipuAI()
+
+	response = client.agents.async_result(
+		agent_id="intelligent_education_correction_polling",
+		custom_variables={
+				"images": [
+					{
+						"image_id": image_id,
+						"uuids": uuids
+					}
+				],
+				"trace_id": trace_id
+		}
+	)
+	print(response)
+ 
+def main():
+    test_correction()
+	# test_correction_result(image_id,uuids,trace_id)
+
+if __name__ == "__main__":
+	main()

zhipuai/__version__.py

@@ -1,2 +1 @@
-
-__version__ = 'v2.1.0'
+__version__ = 'v2.1.5.20250724'

zhipuai/types/agents/agents_completion.py

@@ -19,7 +19,7 @@ class AgentsCompletionChoice(BaseModel):
     finish_reason: str
     message: AgentsCompletionMessage
 
-class AgentsError:
+class AgentsError(BaseModel):
     code: Optional[str] = None
     message: Optional[str] = None