上传文件至 /

function call作业基础版.ipynb

@@ -0,0 +1,272 @@
+{
+ "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
+}