Change default model for InferenceClientModel to fix ToolCallingAgent incompatibility
#118
by
WilliamRabuel
- opened
unit2/smolagents/tool_calling_agents.ipynb
CHANGED
@@ -87,7 +87,7 @@
|
|
87 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
88 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"font-weight: bold\">Search for the best music recommendations for a party at the Wayne's mansion.</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
89 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
90 |
-
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β°β InferenceClientModel -
|
91 |
"</pre>\n"
|
92 |
],
|
93 |
"text/plain": [
|
@@ -95,7 +95,7 @@
|
|
95 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
96 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[1mSearch for the best music recommendations for a party at the Wayne's mansion.\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
97 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
98 |
-
"\u001b[38;2;212;183;2mβ°β\u001b[0m\u001b[38;2;212;183;2m InferenceClientModel -
|
99 |
]
|
100 |
},
|
101 |
"metadata": {},
|
@@ -552,7 +552,7 @@
|
|
552 |
"source": [
|
553 |
"from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel\n",
|
554 |
"\n",
|
555 |
-
"agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel())\n",
|
556 |
"\n",
|
557 |
"agent.run(\"Search for the best music recommendations for a party at the Wayne's mansion.\")"
|
558 |
]
|
|
|
87 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
88 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"font-weight: bold\">Search for the best music recommendations for a party at the Wayne's mansion.</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
89 |
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span> <span style=\"color: #d4b702; text-decoration-color: #d4b702\">β</span>\n",
|
90 |
+
"<span style=\"color: #d4b702; text-decoration-color: #d4b702\">β°β InferenceClientModel - mistralai/Mixtral-8x7B-Instruct-v0.1 βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ―</span>\n",
|
91 |
"</pre>\n"
|
92 |
],
|
93 |
"text/plain": [
|
|
|
95 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
96 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[1mSearch for the best music recommendations for a party at the Wayne's mansion.\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
97 |
"\u001b[38;2;212;183;2mβ\u001b[0m \u001b[38;2;212;183;2mβ\u001b[0m\n",
|
98 |
+
"\u001b[38;2;212;183;2mβ°β\u001b[0m\u001b[38;2;212;183;2m InferenceClientModel - mistralai/Mixtral-8x7B-Instruct-v0.1 \u001b[0m\u001b[38;2;212;183;2mβββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ\u001b[0m\u001b[38;2;212;183;2mββ―\u001b[0m\n"
|
99 |
]
|
100 |
},
|
101 |
"metadata": {},
|
|
|
552 |
"source": [
|
553 |
"from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, InferenceClientModel\n",
|
554 |
"\n",
|
555 |
+
"agent = ToolCallingAgent(tools=[DuckDuckGoSearchTool()], model=InferenceClientModel(model_id="mistralai/Mixtral-8x7B-Instruct-v0.1")))\n",
|
556 |
"\n",
|
557 |
"agent.run(\"Search for the best music recommendations for a party at the Wayne's mansion.\")"
|
558 |
]
|