From 6271c2c5f5ffa8d8e73322f07f7637c0788f0ac6 Mon Sep 17 00:00:00 2001 From: Yangong Date: Mon, 3 Nov 2025 03:35:28 +0000 Subject: [PATCH] refactor(graph): apply dependency inversion to default model creation --- libs/deepagents/graph.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/libs/deepagents/graph.py b/libs/deepagents/graph.py index f2eeee59..bed393d4 100644 --- a/libs/deepagents/graph.py +++ b/libs/deepagents/graph.py @@ -8,7 +8,6 @@ from langchain.agents.middleware.summarization import SummarizationMiddleware from langchain.agents.middleware.types import AgentMiddleware from langchain.agents.structured_output import ResponseFormat -from langchain_anthropic import ChatAnthropic from langchain_anthropic.middleware import AnthropicPromptCachingMiddleware from langchain_core.language_models import BaseChatModel from langchain_core.tools import BaseTool @@ -25,12 +24,14 @@ BASE_AGENT_PROMPT = "In order to complete the objective that the user asks of you, you have access to a number of standard tools." -def get_default_model() -> ChatAnthropic: +def get_default_model() -> BaseChatModel: """Get the default model for deep agents. Returns: - ChatAnthropic instance configured with Claude Sonnet 4. + BaseChatModel instance configured with Claude Sonnet 4. """ + from langchain_anthropic import ChatAnthropic + return ChatAnthropic( model_name="claude-sonnet-4-5-20250929", max_tokens=20000,