Update model

Author: rlancemartin

examples/customer_support/src/agent/customer_support.py

@@ -32,7 +32,6 @@
     }
 ]
 
-
 # Flight tools
 def search_flights(
     departure_airport: str,

examples/research/pyproject.toml

@@ -12,6 +12,7 @@ dependencies = [
     "httpx>=0.28.1",
     "markdownify>=1.1.0",
     "langchain-anthropic>=0.3.10",
+    "langchain-openai>=0.3.11",
     "langchain-mcp-adapters>=0.0.5",
     "langgraph>=0.3.21",
     "langgraph-swarm>=0.0.7",

examples/research/src/agent/agent.ipynb

@@ -10,7 +10,7 @@
    },
    "outputs": [],
    "source": [
-    "! pip install langchain-mcp-adapters langgraph \"langchain[anthropic]\" langgraph-swarm httpx markdownify"
+    "! pip install langchain-mcp-adapters langgraph \"langchain[anthropic]\" \"langchain[openai]\" langgraph-swarm httpx markdownify"
    ]
   },
   {
@@ -32,7 +32,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -43,12 +43,16 @@
     "from langgraph.prebuilt import create_react_agent\n",
     "from langgraph_swarm import create_handoff_tool, create_swarm\n",
     "\n",
-    "# LLM\n",
+    "# Reasoning model (Anthropic)\n",
     "model = init_chat_model(model=\"claude-3-7-sonnet-latest\", \n",
     "                        model_provider=\"anthropic\", \n",
     "                        max_tokens=20_000, \n",
     "                        thinking={\"type\": \"enabled\", \"budget_tokens\": 16_000})\n",
     "\n",
+    "# Chat model (OpenAI)\n",
+    "model = init_chat_model(model=\"gpt-4o\", \n",
+    "                        model_provider=\"openai\")\n",
+    "\n",
     "# Handoff tools\n",
     "transfer_to_planner_agent = create_handoff_tool(\n",
     "    agent_name=\"planner_agent\",\n",
@@ -231,13 +235,6 @@
    "source": [
     "result"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {

examples/research/src/agent/agent.py

@@ -1,6 +1,5 @@
 
 from langchain.chat_models import init_chat_model
-
 from langgraph.prebuilt import create_react_agent
 from langgraph_swarm import create_handoff_tool, create_swarm
 
@@ -9,10 +8,8 @@
 from swarm_researcher.utils import fetch_doc
 
 # LLM
-model = init_chat_model(model="claude-3-7-sonnet-latest", 
-                        model_provider="anthropic", 
-                        max_tokens=20_000, 
-                        thinking={"type": "enabled", "budget_tokens": 16_000})
+model = init_chat_model(model="gpt-4o", 
+                        model_provider="openai")
 
 # Handoff tools
 transfer_to_planner_agent = create_handoff_tool(
@@ -29,7 +26,6 @@
 num_urls = 3
 planner_prompt_formatted = planner_prompt.format(llms_txt=llms_txt, num_urls=num_urls)
 
-
 # Planner agent
 planner_agent = create_react_agent(model,
                                    prompt=planner_prompt_formatted,