273 lines
7.7 KiB
Plaintext
273 lines
7.7 KiB
Plaintext
{
|
||
"cells": [
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 1,
|
||
"id": "b646cf4a-18a8-4eb0-882e-4f087b779ffe",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": [
|
||
"import os\n",
|
||
"import json\n",
|
||
"os.environ['http_proxy'] = 'http://10.10.9.50:3000'\n",
|
||
"os.environ['https_proxy'] = 'http://10.10.9.50:3000'\n",
|
||
"os.environ['no_proxy'] = 'localhost,127.0.0.1'"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "8bb7790a-b2c5-46ac-a283-491529b2c0e5",
|
||
"metadata": {},
|
||
"source": [
|
||
"## 1.描述外部函数"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 2,
|
||
"id": "72f06bfc-67d2-4c7b-a394-ffb6df2c5d9b",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": [
|
||
"tools = [\n",
|
||
" {\n",
|
||
" \"type\": \"function\",\n",
|
||
" \"function\": {\n",
|
||
" \"name\": \"get_weather_info\",\n",
|
||
" \"description\": \"Get the weather information of a city\",\n",
|
||
" \"parameters\": {\n",
|
||
" \"type\": \"object\",\n",
|
||
" \"properties\": {\n",
|
||
" \"city\": {\n",
|
||
" \"type\": \"string\",\n",
|
||
" \"description\": \"The name of the city, e.g. Shanghai\",\n",
|
||
" },\n",
|
||
" },\n",
|
||
" \"required\": [\"city\"],\n",
|
||
" },\n",
|
||
" }\n",
|
||
" }\n",
|
||
"]"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "48d76891-be79-4b3a-ae4d-9a4fd1f8b4c0",
|
||
"metadata": {},
|
||
"source": [
|
||
"#### 初始化ZhipuAI"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 3,
|
||
"id": "e868149d-e207-4fa0-be87-2bd834778dfa",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": [
|
||
"from zhipuai import ZhipuAI\n",
|
||
"client = ZhipuAI(api_key=\"cc28e0d694973a7276d02d6822e5958c.5h0H2TcNId2qBGBi\")"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "10b89c40-9e78-4aaf-900c-4501101a4bd9",
|
||
"metadata": {},
|
||
"source": [
|
||
"## 2. 与模型交互，触发模型对函数的调用"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 4,
|
||
"id": "47b1fdaf-c013-42d2-b57b-51acc5c4bfed",
|
||
"metadata": {},
|
||
"outputs": [
|
||
{
|
||
"name": "stdout",
|
||
"output_type": "stream",
|
||
"text": [
|
||
"{'content': None, 'role': 'assistant', 'tool_calls': [{'id': 'call_9106863387107282637', 'function': {'arguments': '{\"city\": \"Beijing\"}', 'name': 'get_weather_info'}, 'type': 'function', 'index': 0}]}\n"
|
||
]
|
||
}
|
||
],
|
||
"source": [
|
||
"query = \"What is the weather like in Beijing?\"\n",
|
||
"messages=[{\"role\": \"user\", \"content\": query}]\n",
|
||
"response = client.chat.completions.create(\n",
|
||
" model=\"GLM-4-Plus\",\n",
|
||
" messages=messages,\n",
|
||
" tools=tools,\n",
|
||
" #这个地方我注释掉了，因为默认就是auto。\n",
|
||
" #tool_choice=\"auto\", \n",
|
||
" )\n",
|
||
"message = response.dict()[\"choices\"][0][\"message\"]\n",
|
||
"print(message)"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "6837d5b3-568c-4c2b-b3e3-29e55a1edc6b",
|
||
"metadata": {},
|
||
"source": [
|
||
"## 3. 使用模型生成的参数调用函数"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "90c428ef-6e79-42b3-b6fd-5e135a0aef3c",
|
||
"metadata": {},
|
||
"source": [
|
||
"### 3.1 先实现函数"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 5,
|
||
"id": "2e292953-6420-4120-aaba-88c0daca7030",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": [
|
||
"import requests\n",
|
||
"\n",
|
||
"def get_weather_info(city=\"Beijing\"):\n",
|
||
" return {\"weather in beijing\": \"sunny\"}"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "markdown",
|
||
"id": "14ee8b16-0a31-45b9-b86a-b9fd3e50b381",
|
||
"metadata": {},
|
||
"source": [
|
||
"### 3.2 定义处理 Function call 的函数："
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 6,
|
||
"id": "51b83a05-efeb-4532-849c-19e0f1c3fe65",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": [
|
||
"def parse_function_call(model_response,messages):\n",
|
||
" # 处理函数调用结果，根据模型返回参数，调用对应的函数。\n",
|
||
" # 调用函数返回结果后构造tool message，再次调用模型，将函数结果输入模型\n",
|
||
" # 模型会将函数调用结果以自然语言格式返回给用户。\n",
|
||
" if model_response.choices[0].message.tool_calls:\n",
|
||
" tool_call = model_response.choices[0].message.tool_calls[0]\n",
|
||
" args = tool_call.function.arguments\n",
|
||
" function_result = {}\n",
|
||
" if tool_call.function.name == \"get_weather_info\":\n",
|
||
" function_result = get_weather_info(**json.loads(args))\n",
|
||
" messages.append({\n",
|
||
" \"role\": \"tool\",\n",
|
||
" \"content\": f\"{json.dumps(function_result)}\",\n",
|
||
" \"tool_call_id\":tool_call.id\n",
|
||
" })\n",
|
||
" response = client.chat.completions.create(\n",
|
||
" model=\"glm-4\", # 填写需要调用的模型名称\n",
|
||
" messages=messages,\n",
|
||
" tools=tools,\n",
|
||
" )\n",
|
||
" print(response.choices[0].message)\n",
|
||
" messages.append(response.choices[0].message.model_dump())"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 7,
|
||
"id": "a3183e74-944d-4596-91b3-4caadbdf5718",
|
||
"metadata": {},
|
||
"outputs": [
|
||
{
|
||
"name": "stdout",
|
||
"output_type": "stream",
|
||
"text": [
|
||
"CompletionMessage(content=None, role='assistant', tool_calls=[CompletionMessageToolCall(id='call_9106863215308436029', function=Function(arguments='{\"city\":\"Beijing\"}', name='get_weather_info'), type='function', index=0)])\n",
|
||
"CompletionMessage(content='According to the weather information API, the weather in Beijing is currently sunny.', role='assistant', tool_calls=None)\n"
|
||
]
|
||
}
|
||
],
|
||
"source": [
|
||
"# 清空对话\n",
|
||
"messages = []\n",
|
||
" \n",
|
||
"messages.append({\"role\": \"user\", \"content\": \"What is the weather like in Beijing?\"})\n",
|
||
" \n",
|
||
"response = client.chat.completions.create(\n",
|
||
" model=\"glm-4\", # 填写需要调用的模型名称\n",
|
||
" messages=messages,\n",
|
||
" tools=tools,\n",
|
||
")\n",
|
||
"print(response.choices[0].message)\n",
|
||
"messages.append(response.choices[0].message.model_dump())\n",
|
||
" \n",
|
||
"parse_function_call(response,messages)"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": 8,
|
||
"id": "04d00aae-254f-4120-b657-b493d3a4f410",
|
||
"metadata": {},
|
||
"outputs": [
|
||
{
|
||
"data": {
|
||
"text/plain": [
|
||
"{'content': None,\n",
|
||
" 'role': 'assistant',\n",
|
||
" 'tool_calls': [{'id': 'call_9106863215308436029',\n",
|
||
" 'function': {'arguments': '{\"city\":\"Beijing\"}', 'name': 'get_weather_info'},\n",
|
||
" 'type': 'function',\n",
|
||
" 'index': 0}]}"
|
||
]
|
||
},
|
||
"execution_count": 8,
|
||
"metadata": {},
|
||
"output_type": "execute_result"
|
||
}
|
||
],
|
||
"source": [
|
||
"response.choices[0].message.model_dump()"
|
||
]
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": null,
|
||
"id": "87c09ed4-b719-4fd9-a151-6f0ffd93aed9",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": []
|
||
},
|
||
{
|
||
"cell_type": "code",
|
||
"execution_count": null,
|
||
"id": "c9ab6bb0-1ebf-4358-ba09-12950bd4d107",
|
||
"metadata": {},
|
||
"outputs": [],
|
||
"source": []
|
||
}
|
||
],
|
||
"metadata": {
|
||
"kernelspec": {
|
||
"display_name": "python3.10(base)",
|
||
"language": "python",
|
||
"name": "python3"
|
||
},
|
||
"language_info": {
|
||
"codemirror_mode": {
|
||
"name": "ipython",
|
||
"version": 3
|
||
},
|
||
"file_extension": ".py",
|
||
"mimetype": "text/x-python",
|
||
"name": "python",
|
||
"nbconvert_exporter": "python",
|
||
"pygments_lexer": "ipython3",
|
||
"version": "3.10.8"
|
||
}
|
||
},
|
||
"nbformat": 4,
|
||
"nbformat_minor": 5
|
||
}
|