From 2e7ba112fe4f57e46caa8e306047f1ab65da7f39 Mon Sep 17 00:00:00 2001
From: Lighthousexx <xxyyuki0323@gmail.com>
Date: Thu, 14 Aug 2025 19:32:39 +0800
Subject: [PATCH] feat: chatCompletion with mcp example

---
 examples/basic_usage.py | 55 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 55 insertions(+)

diff --git a/examples/basic_usage.py b/examples/basic_usage.py
index 11f00fc..3a65216 100644
--- a/examples/basic_usage.py
+++ b/examples/basic_usage.py
@@ -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