Update minstral client in function_calling.ipynb

github · mor10 · Aug 12, 2024 · Aug 12, 2024 · Aug 12, 2024 · Aug 12, 2024
commit 516b3447a8566c42ca297d0b9953cf6d38cef539
diff --git a/cookbooks/python/mistralai/function_calling.ipynb b/cookbooks/python/mistralai/function_calling.ipynb
@@ -190,10 +190,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from mistralai.models.chat_completion import ChatMessage\n",
+    "from mistralai import Mistral\n",
     "\n",
     "messages = [\n",
-    "    ChatMessage(role=\"user\", content=\"What's the status of my transaction T1001?\")\n",
+    "    {\"role\":\"user\", \"content\":\"What's the status of my transaction T1001?\"}\n",
     "]\n"
    ]
   },
@@ -214,13 +214,11 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from mistralai.client import MistralClient\n",
-    "\n",
     "model = \"mistral-large\"\n",
     "\n",
-    "client = MistralClient(api_key=github_token, endpoint=endpoint)\n",
+    "client = Mistral(api_key=github_token, server_url=endpoint)\n",
     "\n",
-    "response = client.chat(\n",
+    "response = client.chat.complete(\n",
     "    model=model,\n",
     "    messages=messages,\n",
     "    tools=tools,\n",
@@ -294,7 +292,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "messages.append(ChatMessage(role=\"tool\", name=function_name, content=function_result, tool_call_id=tool_call.id))"
+    "messages.append({\"role\":\"tool\", \"name\":function_name, \"content\":function_result, \"tool_call_id\":tool_call.id})"
    ]
   },
   {
@@ -324,7 +322,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "response = client.chat(\n",
+    "response = client.chat.complete(\n",
     "    model=model,\n",
     "    messages=messages\n",
     ")\n",