feat: support glm-4.6v (#54)

This commit is contained in:
Tomsun28 2025-12-08 14:46:40 +08:00 committed by GitHub
parent b39f4b290f
commit 84d48d48f7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 16 additions and 16 deletions

View file

@ -169,7 +169,7 @@ client = ZaiClient(api_key="your-api-key")
# Create chat completion # Create chat completion
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4', model='glm-4.6',
messages=[ messages=[
{'role': 'system', 'content': 'You are a helpful assistant.'}, {'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': 'Tell me a story about AI.'}, {'role': 'user', 'content': 'Tell me a story about AI.'},
@ -192,7 +192,7 @@ client = ZaiClient(api_key="your-api-key")
# Create chat completion # Create chat completion
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4', model='glm-4.6',
messages=[ messages=[
{'role': 'system', 'content': 'You are a helpful assistant.'}, {'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': 'What is artificial intelligence?'}, {'role': 'user', 'content': 'What is artificial intelligence?'},
@ -228,7 +228,7 @@ client = ZaiClient(api_key="your-api-key")
base64_image = encode_image('examples/test_multi_modal.jpeg') base64_image = encode_image('examples/test_multi_modal.jpeg')
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4v', model='glm-4.6v',
messages=[ messages=[
{ {
'role': 'user', 'role': 'user',
@ -338,7 +338,7 @@ client = ZaiClient(api_key="your-api-key")
try: try:
response = client.chat.completions.create( response = client.chat.completions.create(
model="glm-4", model="glm-4.6",
messages=[ messages=[
{"role": "user", "content": "Hello, Z.ai!"} {"role": "user", "content": "Hello, Z.ai!"}
] ]

View file

@ -107,7 +107,7 @@ client = ZhipuAiClient(api_key="your-api-key")
# Create chat completion # Create chat completion
response = client.chat.completions.create( response = client.chat.completions.create(
model="glm-4", model="glm-4.6",
messages=[ messages=[
{"role": "user", "content": "Hello, Z.ai!"} {"role": "user", "content": "Hello, Z.ai!"}
] ]
@ -175,7 +175,7 @@ client = ZaiClient(api_key="your-api-key")
# 创建对话 # 创建对话
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4', model='glm-4.6',
messages=[ messages=[
{'role': 'system', 'content': 'You are a helpful assistant.'}, {'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': 'Tell me a story about AI.'}, {'role': 'user', 'content': 'Tell me a story about AI.'},
@ -198,7 +198,7 @@ client = ZaiClient(api_key="your-api-key")
# 创建对话 # 创建对话
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4', model='glm-4.6',
messages=[ messages=[
{'role': 'system', 'content': 'You are a helpful assistant.'}, {'role': 'system', 'content': 'You are a helpful assistant.'},
{'role': 'user', 'content': 'What is artificial intelligence?'}, {'role': 'user', 'content': 'What is artificial intelligence?'},
@ -234,7 +234,7 @@ client = ZaiClient(api_key="your-api-key")
base64_image = encode_image('examples/test_multi_modal.jpeg') base64_image = encode_image('examples/test_multi_modal.jpeg')
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4v', model='glm-4.6v',
messages=[ messages=[
{ {
'role': 'user', 'role': 'user',
@ -344,7 +344,7 @@ client = ZaiClient(api_key="your-api-key") # 请填写您自己的APIKey
try: try:
response = client.chat.completions.create( response = client.chat.completions.create(
model="glm-4", model="glm-4.6",
messages=[ messages=[
{"role": "user", "content": "你好, Z.ai "} {"role": "user", "content": "你好, Z.ai "}
] ]

View file

@ -126,7 +126,7 @@ def multi_modal_chat():
base64_image = encode_image('examples/test_multi_modal.jpeg') base64_image = encode_image('examples/test_multi_modal.jpeg')
response = client.chat.completions.create( response = client.chat.completions.create(
model='glm-4v', model='glm-4.6v',
messages=[ messages=[
{ {
'role': 'user', 'role': 'user',

View file

@ -1,6 +1,6 @@
[tool.poetry] [tool.poetry]
name = "zai-sdk" name = "zai-sdk"
version = "0.0.4.3" version = "0.1.0"
description = "A SDK library for accessing big model apis from Z.ai" description = "A SDK library for accessing big model apis from Z.ai"
authors = ["Z.ai"] authors = ["Z.ai"]
readme = "README.md" readme = "README.md"

View file

@ -1,2 +1,2 @@
__title__ = 'Z.ai' __title__ = 'Z.ai'
__version__ = '0.0.4.3' __version__ = '0.1.0'

View file

@ -68,7 +68,7 @@ class ChoiceDelta(BaseModel):
Attributes: Attributes:
content: Content delta content: Content delta
role: Role of the message sender role: Role of the message sender
reasoning_content: Reasoning content delta reasoning_content: Reasoning content delta, it's recommended to return the model's reasoning_content in next request to achieve better results in multi-turn conversations.
tool_calls: List of tool call deltas tool_calls: List of tool call deltas
audio: Audio completion chunk audio: Audio completion chunk
""" """

View file

@ -323,7 +323,7 @@ def test_completions_stream_with_tools(logging_conf):
print(f'request_id:{request_id}') print(f'request_id:{request_id}')
response = client.chat.completions.create( response = client.chat.completions.create(
request_id=request_id, request_id=request_id,
model='glm-4v', # Fill in the model name to call model='glm-4.6v', # Fill in the model name to call
extra_body={'temperature': 0.5, 'max_tokens': 50}, extra_body={'temperature': 0.5, 'max_tokens': 50},
messages=[ messages=[
{ {
@ -361,7 +361,7 @@ def test_completions_vis_base64(test_file_path, logging_conf):
print(f'request_id:{request_id}') print(f'request_id:{request_id}')
response = client.chat.completions.create( response = client.chat.completions.create(
request_id=request_id, request_id=request_id,
model='glm-4v', # Fill in the model name to call model='glm-4.6v', # Fill in the model name to call
extra_body={'temperature': 0.5, 'max_tokens': 50}, extra_body={'temperature': 0.5, 'max_tokens': 50},
messages=[ messages=[
{ {
@ -402,7 +402,7 @@ def test_async_completions(logging_conf):
print(f'request_id:{request_id}') print(f'request_id:{request_id}')
response = client.chat.completions.create( response = client.chat.completions.create(
request_id=request_id, request_id=request_id,
model='glm-4v', # Fill in the model name to call model='glm-4.6v', # Fill in the model name to call
messages=[ messages=[
{ {
'role': 'user', 'role': 'user',