langgraph-agent-toolkit 0.1.2__tar.gz → 0.2.0__tar.gz

langgraph_agent_toolkit-0.2.0/.coveragerc

@@ -0,0 +1,6 @@
+[run]
+omit =
+    examples/*
+    scripts/*
+    tests/*
+    langgraph_agent_toolkit/agents/blueprints/*

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/.env.example

@@ -47,11 +47,38 @@ POSTGRES_MAX_IDLE=5
 # Use a fake model for testing
 USE_FAKE_MODEL=false
 
-# Set a default model
-DEFAULT_MODEL_TYPE="openai-compatible"
-
-# If MODEL is set to "openai-compatible", set the following
-# This is just a flexible solution. If you need multiple model options, you still need to add it to models.py
-MODEL_NAME=
-MODEL_API_KEY=
-MODEL_BASE_URL=http://litellm:4000/v1
+# OpenAI Settings
+OPENAI_API_KEY=
+OPENAI_API_BASE_URL=
+OPENAI_API_VERSION=
+OPENAI_MODEL_NAME=
+
+# Azure OpenAI Settings
+AZURE_OPENAI_API_KEY=
+AZURE_OPENAI_ENDPOINT=
+AZURE_OPENAI_API_VERSION=
+AZURE_OPENAI_MODEL_NAME=
+AZURE_OPENAI_DEPLOYMENT_NAME=
+
+# Anthropic Settings
+ANTHROPIC_API_KEY=
+ANTHROPIC_MODEL_NAME=
+
+ # Google VertexAI Settings
+GOOGLE_VERTEXAI_MODEL_NAME=
+GOOGLE_VERTEXAI_API_KEY=
+
+# Google GenAI Settings
+GOOGLE_GENAI_MODEL_NAME=
+GOOGLE_GENAI_API_KEY=
+
+# Bedrock Settings
+AWS_BEDROCK_MODEL_NAME=
+
+# DeepSeek Settings
+DEEPSEEK_MODEL_NAME=
+DEEPSEEK_API_KEY=
+
+# Ollama Settings
+OLLAMA_MODEL_NAME=
+OLLAMA_BASE_URL=

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/.github/PULL_REQUEST_TEMPLATE.md

@@ -1,3 +1,5 @@
+# PR Description
+
 ## Summary
 
 <!-- Provide a concise summary "Why are the changes needed"?

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/CHANGELOG.md

@@ -6,6 +6,17 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to
 [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.2.0]
+
+### Fixed
+
+- Refactored code structure for better maintainability
+- Refactored Model factory
+
+### Removed
+
+- Removed AllModels and added environment variables for different providers
+
 ## [0.1.2]
 
 ### Added

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.1.2
+Version: 0.2.0
 Summary: Full toolkit for running an AI agent service built with LangGraph, FastAPI and Streamlit
 Project-URL: repository, https://github.com/kryvokhyzha/langgraph-agent-toolkit
 Project-URL: PyPI, https://pypi.org/project/langgraph-agent-toolkit
@@ -133,7 +133,8 @@ Features include:
 
 ## 🏗️ Architecture
 
-<img src="docs/media/agent_architecture.png" width="800">
+<!-- <img src="docs/media/agent_architecture.png" width="800"> -->
+<img src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/agent_architecture.png?raw=true" width="800">
 
 ## ✨ Key Features
 
@@ -183,10 +184,29 @@ Features include:
    **🔑 LLM API Configuration**
 
    ```env
-   # LiteLLM configuration
-   MODEL_NAME="gpt4o"                         # LiteLLM model name
-   MODEL_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
-   MODEL_API_KEY=sk-any-string                # LiteLLM master key
+   # OpenAI Settings
+   OPENAI_API_KEY=sk-xxxxxxxxxxxxx
+   OPENAI_API_BASE_URL=http://litellm:4000/v1      # Can be OpenAI API or LiteLLM proxy
+   OPENAI_API_VERSION=2025-01-01
+   OPENAI_MODEL_NAME=gpt-4o-mini
+
+   # Azure OpenAI Settings (Optional)
+   AZURE_OPENAI_API_KEY=
+   AZURE_OPENAI_ENDPOINT=
+   AZURE_OPENAI_API_VERSION=
+   AZURE_OPENAI_MODEL_NAME=
+   AZURE_OPENAI_DEPLOYMENT_NAME=
+
+   # Anthropic Settings (Optional)
+   ANTHROPIC_API_KEY=
+   ANTHROPIC_MODEL_NAME=
+
+   # Other provider settings are also available:
+   # - Google VertexAI
+   # - Google GenAI
+   # - AWS Bedrock
+   # - DeepSeek
+   # - Ollama
    ```
 
    **🗄️ Database Configuration**

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/README.md

@@ -62,7 +62,8 @@ Features include:
 
 ## 🏗️ Architecture
 
-<img src="docs/media/agent_architecture.png" width="800">
+<!-- <img src="docs/media/agent_architecture.png" width="800"> -->
+<img src="https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/docs/media/agent_architecture.png?raw=true" width="800">
 
 ## ✨ Key Features
 
@@ -112,10 +113,29 @@ Features include:
    **🔑 LLM API Configuration**
 
    ```env
-   # LiteLLM configuration
-   MODEL_NAME="gpt4o"                         # LiteLLM model name
-   MODEL_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
-   MODEL_API_KEY=sk-any-string                # LiteLLM master key
+   # OpenAI Settings
+   OPENAI_API_KEY=sk-xxxxxxxxxxxxx
+   OPENAI_API_BASE_URL=http://litellm:4000/v1      # Can be OpenAI API or LiteLLM proxy
+   OPENAI_API_VERSION=2025-01-01
+   OPENAI_MODEL_NAME=gpt-4o-mini
+
+   # Azure OpenAI Settings (Optional)
+   AZURE_OPENAI_API_KEY=
+   AZURE_OPENAI_ENDPOINT=
+   AZURE_OPENAI_API_VERSION=
+   AZURE_OPENAI_MODEL_NAME=
+   AZURE_OPENAI_DEPLOYMENT_NAME=
+
+   # Anthropic Settings (Optional)
+   ANTHROPIC_API_KEY=
+   ANTHROPIC_MODEL_NAME=
+
+   # Other provider settings are also available:
+   # - Google VertexAI
+   # - Google GenAI
+   # - AWS Bedrock
+   # - DeepSeek
+   # - Ollama
    ```
 
    **🗄️ Database Configuration**

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/configs/litellm/config.example.yaml

@@ -1,5 +1,7 @@
-# https://docs.litellm.ai/docs/proxy/configs
+# Proxy doc: https://docs.litellm.ai/docs/proxy/configs
+# Caching doc: https://docs.litellm.ai/docs/proxy/config_settings
 
+# list of supported models on the server, with model-specific configs
 model_list:
   - model_name: gpt-4o-mini
     litellm_params:
@@ -16,6 +18,7 @@ model_list:
     model_info:
       supported_environments: ["development", "production", "staging"]
 
+# list of credentials to be used by the models
 credential_list:
   - credential_name: default_azure_credential
     credential_values:
@@ -25,12 +28,14 @@ credential_list:
     credential_info:
       description: "Production credentials for EU region"
 
+# litellm Module settings
 litellm_settings:
   drop_params: True
   num_retries: 2 # retry call 3 times on each model_name (e.g. zephyr-beta)
   request_timeout: 40 # raise Timeout error if call takes longer than 10s. Sets litellm.request_timeout
   allowed_fails: 3 # cooldown model if it fails > 1 call in a minute.
 
+# litellm Router settings
 router_settings: # router_settings are optional
   routing_strategy: latency-based-routing # Literal["simple-shuffle", "least-busy", "usage-based-routing","latency-based-routing"], default="simple-shuffle"
   model_group_alias: { "gpt-4": "gpt-3.5-turbo" } # all requests with `gpt-4` will be routed to models with `gpt-3.5-turbo`
@@ -40,8 +45,15 @@ router_settings: # router_settings are optional
   redis_password: os.environ/REDIS_AUTH
   redis_port: 1992
 
+# Server settings
 general_settings:
   master_key: os.environ/LITELLM_MASTER_KEY
   database_connection_pool_limit: 1000
+  database_connection_timeout: 60
   store_model_in_db: True
   store_prompts_in_spend_logs: True
+# environment_variables:
+#   # settings for using redis caching
+#   REDIS_HOST: redis-16337.c322.us-east-1-2.ec2.cloud.redislabs.com
+#   REDIS_PORT: "16337"
+#   REDIS_PASSWORD:

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/agent_executor.py

@@ -172,7 +172,8 @@ class AgentExecutor:
         message: str,
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
-        model: Optional[str] = None,
+        model_name: Optional[str] = None,
+        model_provider: Optional[str] = None,
         agent_config: Optional[Dict[str, Any]] = None,
         recursion_limit: Optional[int] = None,
     ) -> Tuple[Agent, Any, Any, UUID]:
@@ -183,7 +184,8 @@ class AgentExecutor:
             message: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
-            model: Optional model name to override the default
+            model_name: Optional model name to override the default
+            model_provider: Optional model provider to override the default
             agent_config: Optional additional configuration for the agent
             recursion_limit: Optional recursion limit for the agent
 
@@ -208,8 +210,11 @@ class AgentExecutor:
             "user_id": user_id,
         }
 
-        if model:
-            configurable["model"] = model
+        if model_name:
+            configurable["model_name"] = model_name
+
+        if model_provider:
+            configurable["model_provider"] = model_provider
 
         if agent_config:
             configurable.update(agent_config)
@@ -243,7 +248,8 @@ class AgentExecutor:
         message: str,
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
-        model: Optional[str] = None,
+        model_name: Optional[str] = None,
+        model_provider: Optional[str] = None,
         agent_config: Optional[Dict[str, Any]] = None,
         recursion_limit: Optional[int] = None,
     ) -> ChatMessage:
@@ -254,7 +260,8 @@ class AgentExecutor:
             message: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
-            model: Optional model name to override the default
+            model_name: Optional model name to override the default
+            model_provider: Optional model provider to override the default
             agent_config: Optional additional configuration for the agent
             recursion_limit: Optional recursion limit for the agent
 
@@ -267,7 +274,8 @@ class AgentExecutor:
             message=message,
             thread_id=thread_id,
             user_id=user_id,
-            model=model,
+            model_name=model_name,
+            model_provider=model_provider,
             agent_config=agent_config,
             recursion_limit=recursion_limit,
         )
@@ -300,7 +308,8 @@ class AgentExecutor:
         message: str,
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
-        model: Optional[str] = None,
+        model_name: Optional[str] = None,
+        model_provider: Optional[str] = None,
         stream_tokens: bool = True,
         agent_config: Optional[Dict[str, Any]] = None,
         recursion_limit: Optional[int] = None,
@@ -312,7 +321,8 @@ class AgentExecutor:
             message: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
-            model: Optional model name to override the default
+            model_name: Optional model name to override the default
+            model_provider: Optional model provider to override the default
             stream_tokens: Whether to stream individual tokens
             agent_config: Optional additional configuration for the agent
             recursion_limit: Optional recursion limit for the agent
@@ -326,7 +336,8 @@ class AgentExecutor:
             message=message,
             thread_id=thread_id,
             user_id=user_id,
-            model=model,
+            model_name=model_name,
+            model_provider=model_provider,
             agent_config=agent_config,
             recursion_limit=recursion_limit,
         )

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/bg_task_agent/agent.py

@@ -11,6 +11,7 @@ from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.agents.blueprints.bg_task_agent.task import Task
 from langgraph_agent_toolkit.core import settings
 from langgraph_agent_toolkit.core.models.factory import ModelFactory
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 class AgentState(MessagesState, total=False):
@@ -29,7 +30,12 @@ def wrap_model(model: BaseChatModel) -> RunnableSerializable[AgentState, AIMessa
 
 
 async def acall_model(state: AgentState, config: RunnableConfig) -> AgentState:
-    m = ModelFactory.create(config["configurable"].get("model", settings.DEFAULT_MODEL_TYPE))
+    m = ModelFactory.create(
+        mmodel_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
+        model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        openai_api_base=settings.OPENAI_API_BASE_URL,
+        openai_api_key=settings.OPENAI_API_KEY,
+    )
     model_runnable = wrap_model(m)
     response = await model_runnable.ainvoke(state, config)
 

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/chatbot/agent.py

@@ -1,12 +1,11 @@
 from langchain_core.messages import BaseMessage
 from langchain_core.runnables import RunnableConfig
-from langgraph.checkpoint.memory import MemorySaver
 from langgraph.func import entrypoint
-from langgraph.graph import add_messages
 
 from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.core import settings
 from langgraph_agent_toolkit.core.models.factory import ModelFactory
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 @entrypoint(
@@ -22,7 +21,12 @@ async def chatbot(
     if previous:
         messages = previous["messages"] + messages
 
-    model = ModelFactory.create(config["configurable"].get("model", settings.DEFAULT_MODEL_TYPE))
+    model = ModelFactory.create(
+        model_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
+        model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        openai_api_base=settings.OPENAI_API_BASE_URL,
+        openai_api_key=settings.OPENAI_API_KEY,
+    )
     response = await model.ainvoke(messages)
     return entrypoint.final(value={"messages": [response]}, save={"messages": messages + [response]})
 

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/interrupt_agent/agent.py

@@ -14,6 +14,7 @@ from pydantic import BaseModel, Field
 from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.core import settings
 from langgraph_agent_toolkit.core.models.factory import ModelFactory
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 class AgentState(MessagesState, total=False):
@@ -44,7 +45,12 @@ Don't tell the user what their sign is, you are just demonstrating your knowledg
 
 async def background(state: AgentState, config: RunnableConfig) -> AgentState:
     """Demonstrate doing work before the interrupt."""
-    m = ModelFactory.create(config["configurable"].get("model", settings.DEFAULT_MODEL_TYPE))
+    m = ModelFactory.create(
+        model_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
+        model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        openai_api_base=settings.OPENAI_API_BASE_URL,
+        openai_api_key=settings.OPENAI_API_KEY,
+    )
     model_runnable = wrap_model(m, background_prompt.format())
     response = await model_runnable.ainvoke(state, config)
 
@@ -74,7 +80,12 @@ async def determine_birthdate(state: AgentState, config: RunnableConfig) -> Agen
 
     If no birthdate is found, it will perform an interrupt before proceeding.
     """
-    m = ModelFactory.create(config["configurable"].get("model", settings.DEFAULT_MODEL_TYPE))
+    m = ModelFactory.create(
+        model_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
+        model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        openai_api_base=settings.OPENAI_API_BASE_URL,
+        openai_api_key=settings.OPENAI_API_KEY,
+    )
     model_runnable = wrap_model(
         m.with_structured_output(BirthdateExtraction),
         birthdate_extraction_prompt.format(),
@@ -117,7 +128,12 @@ async def determine_sign(state: AgentState, config: RunnableConfig) -> AgentStat
     if not state.get("birthdate"):
         raise ValueError("No birthdate found in state")
 
-    m = ModelFactory.create(config["configurable"].get("model", settings.DEFAULT_MODEL_TYPE))
+    m = ModelFactory.create(
+        model_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
+        model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        openai_api_base=settings.OPENAI_API_BASE_URL,
+        openai_api_key=settings.OPENAI_API_KEY,
+    )
     model_runnable = wrap_model(m, sign_prompt.format(birthdate=state["birthdate"].strftime("%Y-%m-%d")))
     response = await model_runnable.ainvoke(state, config)
 

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/react/agent.py

@@ -10,6 +10,7 @@ from langgraph_agent_toolkit.core.observability.factory import ObservabilityFact
 from langgraph_agent_toolkit.core.observability.types import ChatMessageDict, MessageRole, ObservabilityBackend
 from langgraph_agent_toolkit.core.prompts.chat_prompt_template import ObservabilityChatPromptTemplate
 from langgraph_agent_toolkit.core.settings import settings
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 PROMPT_NAME = "react-assistant"
@@ -50,7 +51,12 @@ react_agent = Agent(
     name="react-agent",
     description="A react agent.",
     graph=create_react_agent(
-        model=ModelFactory.create(settings.DEFAULT_MODEL_TYPE),
+        model=ModelFactory.create(
+            model_provider=ModelProvider.OPENAI,
+            model_name=settings.OPENAI_MODEL_NAME,
+            openai_api_base=settings.OPENAI_API_BASE_URL,
+            openai_api_key=settings.OPENAI_API_KEY,
+        ),
         tools=[add, multiply, DuckDuckGoSearchResults()],
         prompt=prompt,
         pre_model_hook=pre_model_hook_standard,

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/react_so/agent.py

@@ -8,6 +8,7 @@ from langgraph_agent_toolkit.agents.components.tools import add, multiply
 from langgraph_agent_toolkit.agents.components.utils import AgentStateWithRemainingSteps, pre_model_hook_standard
 from langgraph_agent_toolkit.core import settings
 from langgraph_agent_toolkit.core.models.factory import ModelFactory
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 class ResponseSchema(BaseModel):
@@ -23,7 +24,12 @@ react_agent_so = Agent(
     name="react-agent-so",
     description="A react agent with structured output.",
     graph=create_react_agent(
-        model=ModelFactory.create(settings.DEFAULT_MODEL_TYPE),
+        model=ModelFactory.create(
+            model_provider=ModelProvider.OPENAI,
+            model_name=settings.OPENAI_MODEL_NAME,
+            openai_api_base=settings.OPENAI_API_BASE_URL,
+            openai_api_key=settings.OPENAI_API_KEY,
+        ),
         tools=[add, multiply, DuckDuckGoSearchResults()],
         prompt=(
             "You are a team support agent that can perform calculations and search the web. "

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/agents/blueprints/supervisor_agent/agent.py

@@ -7,9 +7,15 @@ from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.agents.components.tools import add, multiply
 from langgraph_agent_toolkit.core import settings
 from langgraph_agent_toolkit.core.models.factory import ModelFactory
+from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
-model = ModelFactory.create(settings.DEFAULT_MODEL_TYPE)
+model = ModelFactory.create(
+    model_provider=ModelProvider.OPENAI,
+    model_name=settings.OPENAI_MODEL_NAME,
+    openai_api_base=settings.OPENAI_API_BASE_URL,
+    openai_api_key=settings.OPENAI_API_KEY,
+)
 
 math_agent = create_react_agent(
     model=model,

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/client/client.py

@@ -87,7 +87,8 @@ class AgentClient:
     async def ainvoke(
         self,
         message: str,
-        model: str | None = None,
+        model_name: str | None = None,
+        model_provider: str | None = None,
         thread_id: str | None = None,
         user_id: str | None = None,
         agent_config: dict[str, Any] | None = None,
@@ -96,7 +97,8 @@ class AgentClient:
 
         Args:
             message (str): The message to send to the agent
-            model (str, optional): LLM model to use for the agent
+            model_name (str, optional): LLM model to use for the agent
+            model_provider (str, optional): LLM model provider to use for the agent
             thread_id (str, optional): Thread ID for continuing a conversation
             user_id (str, optional): User ID for identifying the user
             agent_config (dict[str, Any], optional): Additional configuration to pass through to the agent
@@ -111,8 +113,10 @@ class AgentClient:
         request = UserInput(message=message)
         if thread_id:
             request.thread_id = thread_id
-        if model:
-            request.model = model
+        if model_name:
+            request.model_name = model_name
+        if model_provider:
+            request.model_provider = model_provider
         if agent_config:
             request.agent_config = agent_config
         if user_id:
@@ -135,7 +139,8 @@ class AgentClient:
     def invoke(
         self,
         message: str,
-        model: str | None = None,
+        model_name: str | None = None,
+        model_provider: str | None = None,
         thread_id: str | None = None,
         user_id: str | None = None,
         agent_config: dict[str, Any] | None = None,
@@ -144,7 +149,8 @@ class AgentClient:
 
         Args:
             message (str): The message to send to the agent
-            model (str, optional): LLM model to use for the agent
+            model_name (str, optional): LLM model to use for the agent
+            model_provider (str, optional): LLM model provider to use for the agent
             thread_id (str, optional): Thread ID for continuing a conversation
             user_id (str, optional): User ID for identifying the user
             agent_config (dict[str, Any], optional): Additional configuration to pass through to the agent
@@ -159,8 +165,10 @@ class AgentClient:
         request = UserInput(message=message)
         if thread_id:
             request.thread_id = thread_id
-        if model:
-            request.model = model
+        if model_name:
+            request.model_name = model_name
+        if model_provider:
+            request.model_provider = model_provider
         if agent_config:
             request.agent_config = agent_config
         if user_id:
@@ -207,7 +215,8 @@ class AgentClient:
     def stream(
         self,
         message: str,
-        model: str | None = None,
+        model_name: str | None = None,
+        model_provider: str | None = None,
         thread_id: str | None = None,
         user_id: str | None = None,
         agent_config: dict[str, Any] | None = None,
@@ -221,7 +230,8 @@ class AgentClient:
 
         Args:
             message (str): The message to send to the agent
-            model (str, optional): LLM model to use for the agent
+            model_name (str, optional): LLM model to use for the agent
+            model_provider (str, optional): LLM model provider to use for the agent
             thread_id (str, optional): Thread ID for continuing a conversation
             user_id (str, optional): User ID for identifying the user
             agent_config (dict[str, Any], optional): Additional configuration to pass through to the agent
@@ -238,8 +248,10 @@ class AgentClient:
         request = StreamInput(message=message, stream_tokens=stream_tokens)
         if thread_id:
             request.thread_id = thread_id
-        if model:
-            request.model = model
+        if model_name:
+            request.model_name = model_name
+        if model_provider:
+            request.model_provider = model_provider
         if agent_config:
             request.agent_config = agent_config
         if user_id:
@@ -266,7 +278,8 @@ class AgentClient:
     async def astream(
         self,
         message: str,
-        model: str | None = None,
+        model_name: str | None = None,
+        model_provider: str | None = None,
         thread_id: str | None = None,
         user_id: str | None = None,
         agent_config: dict[str, Any] | None = None,
@@ -280,7 +293,8 @@ class AgentClient:
 
         Args:
             message (str): The message to send to the agent
-            model (str, optional): LLM model to use for the agent
+            model_name (str, optional): LLM model to use for the agent
+            model_provider (str, optional): LLM model provider to use for the agent
             thread_id (str, optional): Thread ID for continuing a conversation
             user_id (str, optional): User ID for identifying the user
             agent_config (dict[str, Any], optional): Additional configuration to pass through to the agent
@@ -297,8 +311,10 @@ class AgentClient:
         request = StreamInput(message=message, stream_tokens=stream_tokens)
         if thread_id:
             request.thread_id = thread_id
-        if model:
-            request.model = model
+        if model_name:
+            request.model_name = model_name
+        if model_provider:
+            request.model_provider = model_provider
         if agent_config:
             request.agent_config = agent_config
         if user_id:
@@ -324,13 +340,22 @@ class AgentClient:
                 raise AgentClientError(f"Error: {e}")
 
     async def acreate_feedback(
-        self, run_id: str, key: str, score: float, kwargs: dict[str, Any] = {}, user_id: str | None = None
+        self,
+        run_id: str,
+        key: str,
+        score: float,
+        kwargs: dict[str, Any] = {},
+        user_id: str | None = None,
     ) -> None:
         """Create a feedback record for a run.
 
-        This is a simple wrapper for the LangSmith create_feedback API, so the
-        credentials can be stored and managed in the service rather than the client.
-        See: https://api.smith.langchain.com/redoc#tag/feedback/operation/create_feedback_api_v1_feedback_post
+        Args:
+            run_id (str): The ID of the run to provide feedback for
+            key (str): The key for the feedback
+            score (float): The score for the feedback
+            kwargs (dict[str, Any], optional): Additional metadata for the feedback
+            user_id (str, optional): User ID for identifying the user
+
         """
         request = Feedback(run_id=run_id, key=key, score=score, user_id=user_id, kwargs=kwargs)
         async with httpx.AsyncClient() as client:

{langgraph_agent_toolkit-0.1.2 → langgraph_agent_toolkit-0.2.0}/langgraph_agent_toolkit/core/memory/postgres.py

@@ -36,7 +36,8 @@ class PostgresMemoryBackend(BaseMemoryBackend):
             )
         return True
 
-    def get_connection_string(self) -> str:
+    @staticmethod
+    def get_connection_string() -> str:
         """Build and return the PostgreSQL connection string from settings."""
         return (
             f"postgresql://{settings.POSTGRES_USER}:"