portugalfuturista/z-ai-sdk-python
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: chatCompletion with mcp example (#27)

Browse source
This commit is contained in:
Lighthousexx 2025-08-15 10:49:02 +08:00 committed by GitHub
parent 2645c01967
commit 9c4860ac59
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 55 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

55
examples/basic_usage.py
View file

@ -58,6 +58,61 @@ def completion_with_websearch():
print(response)
def completion_with_mcp_server_url():
import os
# Initialize client
client = ZaiClient()
# Create chat completion with MCP server URL
response = client.chat.completions.create(
model='glm-4',
stream=False,
messages=[{'role': 'user', 'content': 'Hello, please introduce GPT?'}],
tools=[
{
'type': 'mcp',
'mcp': {
'server_url': 'https://open.bigmodel.cn/api/mcp/sogou/sse',
'server_label': 'sougou',
'transport_type': 'sse',
'headers': {
'Authorization': f'Bearer {os.getenv("ZAI_API_KEY")}'
}
}
}
],
temperature=0.7,
)
print(response)
def completion_with_mcp_server_label():
import os
# Initialize client
client = ZaiClient()
# Create chat completion with MCP server label
response = client.chat.completions.create(
model='glm-4',
stream=False,
messages=[{'role': 'user', 'content': 'Hello, please introduce GPT?'}],
tools=[
{
'type': 'mcp',
'mcp': {
'server_label': 'sougou_search',
'headers': {
'Authorization': f'Bearer {os.getenv("ZAI_API_KEY")}'
}
}
}
],
temperature=0.7,
)
print(response)
def multi_modal_chat():
import base64