File tree Expand file tree Collapse file tree 1 file changed +55
-0
lines changed Expand file tree Collapse file tree 1 file changed +55
-0
lines changed Original file line number Diff line number Diff line change @@ -58,6 +58,61 @@ def completion_with_websearch():
5858
5959 print (response )
6060
61+ def completion_with_mcp_server_url ():
62+ import os
63+
64+ # Initialize client
65+ client = ZaiClient ()
66+
67+ # Create chat completion with MCP server URL
68+ response = client .chat .completions .create (
69+ model = 'glm-4' ,
70+ stream = False ,
71+ messages = [{'role' : 'user' , 'content' : 'Hello, please introduce GPT?' }],
72+ tools = [
73+ {
74+ 'type' : 'mcp' ,
75+ 'mcp' : {
76+ 'server_url' : 'https://open.bigmodel.cn/api/mcp/sogou/sse' ,
77+ 'server_label' : 'sougou' ,
78+ 'transport_type' : 'sse' ,
79+ 'headers' : {
80+ 'Authorization' : f'Bearer { os .getenv ("ZAI_API_KEY" )} '
81+ }
82+ }
83+ }
84+ ],
85+ temperature = 0.7 ,
86+ )
87+
88+ print (response )
89+
90+ def completion_with_mcp_server_label ():
91+ import os
92+
93+ # Initialize client
94+ client = ZaiClient ()
95+
96+ # Create chat completion with MCP server label
97+ response = client .chat .completions .create (
98+ model = 'glm-4' ,
99+ stream = False ,
100+ messages = [{'role' : 'user' , 'content' : 'Hello, please introduce GPT?' }],
101+ tools = [
102+ {
103+ 'type' : 'mcp' ,
104+ 'mcp' : {
105+ 'server_label' : 'sougou_search' ,
106+ 'headers' : {
107+ 'Authorization' : f'Bearer { os .getenv ("ZAI_API_KEY" )} '
108+ }
109+ }
110+ }
111+ ],
112+ temperature = 0.7 ,
113+ )
114+
115+ print (response )
61116
62117def multi_modal_chat ():
63118 import base64
You can’t perform that action at this time.
0 commit comments