Skip to content

Commit 778ad60

Browse files
author
zhengweijun
committed
add test
1 parent 32c7ec7 commit 778ad60

File tree

1 file changed

+34
-0
lines changed

1 file changed

+34
-0
lines changed

tests/unit_tests/test_agents.py

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
def test_agents_completion_error_field():
2+
from zhipuai.types.agents.agents_completion import AgentsCompletion, AgentsError, AgentsCompletionChoice, AgentsCompletionMessage, AgentsCompletionUsage
3+
4+
# 构造一个 AgentsError
5+
error = AgentsError(code="404", message="Not Found")
6+
7+
# 构造一个完整的 AgentsCompletion
8+
completion = AgentsCompletion(
9+
agent_id="test_agent",
10+
conversation_id="conv_1",
11+
status="failed",
12+
choices=[
13+
AgentsCompletionChoice(
14+
index=0,
15+
finish_reason="error",
16+
message=AgentsCompletionMessage(content="error", role="system")
17+
)
18+
],
19+
request_id="req_1",
20+
id="id_1",
21+
usage=AgentsCompletionUsage(prompt_tokens=1, completion_tokens=1, total_tokens=2),
22+
error=error
23+
)
24+
25+
# 检查 error 字段是否为 AgentsError 实例
26+
assert isinstance(completion.error, AgentsError)
27+
assert completion.error.code == "404"
28+
assert completion.error.message == "Not Found"
29+
30+
# 检查序列化
31+
as_dict = completion.model_dump()
32+
assert as_dict["error"]["code"] == "404"
33+
assert as_dict["error"]["message"] == "Not Found"
34+
print("test_agents_completion_error_field passed.")

0 commit comments

Comments
 (0)