langgraph-agent-toolkit 0.5.0__tar.gz → 0.7.0__tar.gz

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/CHANGELOG.md

@@ -6,6 +6,33 @@ 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.7.0]
+
+### Fixed
+
+- React Agent with SO
+
+### Updated
+
+- Dependencies
+- Name of default configurable parameters
+
+## [0.6.0]
+
+### Fixed
+
+- React Agent with SO
+
+### Added
+
+- Complex input
+
+### Updated
+
+- Dependencies
+- Error handling
+- Tests
+
 ## [0.5.0]
 
 ### Fixed

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.5.0
+Version: 0.7.0
 Summary: Full toolkit for running an AI agent service built with LangGraph, FastAPI and Streamlit
 Project-URL: documentation, https://kryvokhyzha.github.io/langgraph-agent-toolkit
 Project-URL: repository, https://github.com/kryvokhyzha/langgraph-agent-toolkit
@@ -27,20 +27,20 @@ Requires-Dist: fire~=0.7.0
 Requires-Dist: grpcio>=1.68.0
 Requires-Dist: httpx~=0.28.1
 Requires-Dist: jinja2~=3.1.6
-Requires-Dist: jiter~=0.9.0
+Requires-Dist: jiter~=0.10.0
 Requires-Dist: joblib>=1.4.2
-Requires-Dist: langchain-community~=0.3.23
-Requires-Dist: langchain-core~=0.3.59
+Requires-Dist: langchain-community~=0.3.24
+Requires-Dist: langchain-core~=0.3.60
 Requires-Dist: langgraph-checkpoint-postgres~=2.0.21
 Requires-Dist: langgraph-checkpoint-sqlite~=2.0.7
-Requires-Dist: langgraph-cli[inmem]~=0.2.8
+Requires-Dist: langgraph-cli[inmem]~=0.2.10
 Requires-Dist: langgraph-supervisor~=0.0.21
-Requires-Dist: langgraph~=0.4.3
+Requires-Dist: langgraph~=0.4.5
 Requires-Dist: loguru~=0.7.3
 Requires-Dist: multidict>6.3.2
 Requires-Dist: numpy~=1.26.4; python_version <= '3.12'
 Requires-Dist: numpy~=2.2.3; python_version >= '3.13'
-Requires-Dist: onnxruntime~=1.21.1
+Requires-Dist: onnxruntime~=1.22.0
 Requires-Dist: pandas~=2.2.3
 Requires-Dist: psycopg[binary,pool]~=3.2.4
 Requires-Dist: pyarrow>=19.0.1

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/docs/usage.rst

@@ -76,7 +76,7 @@ The toolkit includes ``AgentClient`` for interacting with the agent service:
    from client import AgentClient
    client = AgentClient()
 
-   response = client.invoke("Tell me a brief joke?")
+   response = client.invoke({"message": "Tell me a brief joke?"})
    response.pretty_print()
    # ================================== Ai Message ==================================
    #

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/agents/agent_executor.py

@@ -1,8 +1,8 @@
 import asyncio
 import functools
 import importlib
-import json
 import os
+import sys
 from pathlib import Path
 from typing import Any, AsyncGenerator, Callable, Dict, List, Optional, Tuple, TypeVar
 from uuid import UUID, uuid4
@@ -143,9 +143,9 @@ class AgentExecutor:
         def _handle_error(e: Exception):
             """Handle and re-raise errors with logging."""
             if isinstance(e, GraphRecursionError):
-                logger.error(f"GraphRecursionError occurred: {e}")
+                logger.opt(exception=sys.exc_info()).error(f"GraphRecursionError occurred: {e}")
             else:
-                logger.error(f"Error during agent execution: {e}")
+                logger.opt(exception=sys.exc_info()).error(f"Error during agent execution: {e}")
 
             raise e
 
@@ -171,7 +171,7 @@ class AgentExecutor:
     async def _setup_agent_execution(
         self,
         agent_id: str,
-        message: str,
+        input: Dict[str, Any],
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
         model_name: Optional[str] = None,
@@ -184,7 +184,7 @@ class AgentExecutor:
 
         Args:
             agent_id: ID of the agent to invoke
-            message: User message to send to the agent
+            input: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
             model_name: Optional model name to override the default
@@ -250,12 +250,14 @@ class AgentExecutor:
         state = await agent_graph.aget_state(config=config)
         interrupted_tasks = [task for task in state.tasks if hasattr(task, "interrupts") and task.interrupts]
 
+        _input = input.model_dump()
         input_data: Command | dict[str, Any]
         if interrupted_tasks:
             # User input is a response to resume agent execution from interrupt
-            input_data = Command(resume=message)
+            input_data = Command(resume=_input)
         else:
-            input_data = {"messages": [HumanMessage(content=message)]}
+            message = _input.pop("message", "")
+            input_data = {"messages": [HumanMessage(content=message)], **_input}
 
         return agent, input_data, config, run_id
 
@@ -263,7 +265,7 @@ class AgentExecutor:
     async def invoke(
         self,
         agent_id: str,
-        message: str,
+        input: Dict[str, Any],
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
         model_name: Optional[str] = None,
@@ -276,7 +278,7 @@ class AgentExecutor:
 
         Args:
             agent_id: ID of the agent to invoke
-            message: User message to send to the agent
+            input: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
             model_name: Optional model name to override the default
@@ -291,12 +293,12 @@ class AgentExecutor:
         """
         agent, input_data, config, run_id = await self._setup_agent_execution(
             agent_id=agent_id,
-            message=message,
+            input=input,
             thread_id=thread_id,
             user_id=user_id,
             model_name=model_name,
             model_provider=model_provider,
-            model_config_key=model_config_key,  # Pass this parameter
+            model_config_key=model_config_key,
             agent_config=agent_config,
             recursion_limit=recursion_limit,
         )
@@ -309,9 +311,14 @@ class AgentExecutor:
         )
 
         response_type, response = response_events[-1]
+
         if response_type == "values":
+            generated_message = response.get("structured_response")
+            if not generated_message:
+                generated_message = response["messages"][-1]
+
             # Normal response, the agent completed successfully
-            output = langchain_to_chat_message(response["messages"][-1])
+            output = langchain_to_chat_message(generated_message)
         elif response_type == "updates" and "__interrupt__" in response:
             # The last thing to occur was an interrupt
             # Return the value of the first interrupt as an AIMessage
@@ -326,7 +333,7 @@ class AgentExecutor:
     async def stream(
         self,
         agent_id: str,
-        message: str,
+        input: Dict[str, Any],
         thread_id: Optional[str] = None,
         user_id: Optional[str] = None,
         model_name: Optional[str] = None,
@@ -340,7 +347,7 @@ class AgentExecutor:
 
         Args:
             agent_id: ID of the agent to invoke
-            message: User message to send to the agent
+            input: User message to send to the agent
             thread_id: Optional thread ID for conversation history
             user_id: Optional user ID for the agent
             model_name: Optional model name to override the default
@@ -356,7 +363,7 @@ class AgentExecutor:
         """
         agent, input_data, config, run_id = await self._setup_agent_execution(
             agent_id=agent_id,
-            message=message,
+            input=input,
             thread_id=thread_id,
             user_id=user_id,
             model_name=model_name,
@@ -445,7 +452,7 @@ class AgentExecutor:
             if current_message:
                 processed_messages.append(create_ai_message(current_message))
 
-            for msg in new_messages:
+            for msg in processed_messages:
                 try:
                     chat_message = langchain_to_chat_message(msg)
                     chat_message.run_id = str(run_id)

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/agents/blueprints/interrupt_agent/agent.py

@@ -1,5 +1,5 @@
 from datetime import datetime
-from typing import Any, cast
+from typing import Any
 
 from langchain.prompts import SystemMessagePromptTemplate
 from langchain_core.language_models.base import LanguageModelInput
@@ -83,20 +83,31 @@ async def determine_birthdate(state: AgentState, config: RunnableConfig) -> Agen
     m = ModelFactory.create(
         model_provider=config["configurable"].get("model_provider", ModelProvider.OPENAI),
         model_name=config["configurable"].get("model_name", settings.OPENAI_MODEL_NAME),
+        config_prefix="",
+        configurable_fields=(),
+        model_parameter_values=(("temperature", 0.0), ("top_p", 0.7), ("streaming", False)),
         openai_api_base=settings.OPENAI_API_BASE_URL,
         openai_api_key=settings.OPENAI_API_KEY,
     )
     model_runnable = wrap_model(
-        m.with_structured_output(BirthdateExtraction),
+        m.with_structured_output(BirthdateExtraction, strict=True, include_raw=True),
         birthdate_extraction_prompt.format(),
     ).with_config(tags=["skip_stream"])
-    response: BirthdateExtraction = await model_runnable.ainvoke(state, config)
+    raw_response: BirthdateExtraction = await model_runnable.ainvoke(state, config)
+
+    if raw_response["parsed"] is not None:
+        response: BirthdateExtraction = raw_response["parsed"]
+    elif raw_response["raw"].tool_calls is not None:
+        raw_result = raw_response["raw"].tool_calls[-1]["args"]
+        response = BirthdateExtraction(**raw_result)
+    else:
+        raise ValueError("No valid response from the model")
 
     # If no birthdate found, interrupt
     if response.birthdate is None:
         birthdate_input = interrupt(f"{response.reasoning}\nPlease tell me your birthdate?")
         # Re-run extraction with the new input
-        state["messages"].append(HumanMessage(birthdate_input))
+        state["messages"].append(HumanMessage(birthdate_input["message"]))
         return await determine_birthdate(state, config)
 
     # Birthdate found - convert string to datetime
@@ -107,7 +118,7 @@ async def determine_birthdate(state: AgentState, config: RunnableConfig) -> Agen
         birthdate_input = interrupt(
             "I couldn't understand the date format. Please provide your birthdate in YYYY-MM-DD format."
         )
-        state["messages"].append(HumanMessage(birthdate_input))
+        state["messages"].append(HumanMessage(birthdate_input["message"]))
         return await determine_birthdate(state, config)
 
     # Birthdate found

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/agents/blueprints/react_so/agent.py

@@ -15,7 +15,7 @@ from langgraph_agent_toolkit.schema.models import ModelProvider
 
 
 class ResponseSchema(BaseModel):
-    response: list[str] = Field(
+    response: str = Field(
         description="The response on user query.",
     )
     alternative_response: str = Field(
@@ -30,6 +30,9 @@ react_agent_so = Agent(
         model=ModelFactory.create(
             model_provider=ModelProvider.OPENAI,
             model_name=settings.OPENAI_MODEL_NAME,
+            config_prefix="",
+            configurable_fields=(),
+            model_parameter_values=(("temperature", 0.0), ("top_p", 0.7), ("streaming", False)),
             openai_api_base=settings.OPENAI_API_BASE_URL,
             openai_api_key=settings.OPENAI_API_KEY,
         ),
@@ -40,7 +43,7 @@ react_agent_so = Agent(
             "You can also ask clarifying questions to the user. "
         ),
         pre_model_hook=pre_model_hook_standard,
-        # response_format=ResponseSchema,
+        response_format=ResponseSchema,
         state_schema=AgentStateWithStructuredResponseAndRemainingSteps,
         checkpointer=MemorySaver(),
         immediate_step_threshold=5,

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/agents/components/creators/create_react_agent.py

@@ -319,7 +319,9 @@ def create_react_agent(
             messages = [SystemMessage(content=system_prompt)] + list(messages)
 
         model_with_structured_output = _get_model(model, config).with_structured_output(
-            cast(StructuredResponseSchema, structured_response_schema)
+            cast(StructuredResponseSchema, structured_response_schema),
+            strict=True,
+            # include_raw=True,
         )
         response = model_with_structured_output.invoke(messages, config)
         return {"structured_response": response}
@@ -332,7 +334,9 @@ def create_react_agent(
             messages = [SystemMessage(content=system_prompt)] + list(messages)
 
         model_with_structured_output = _get_model(model, config).with_structured_output(
-            cast(StructuredResponseSchema, structured_response_schema)
+            cast(StructuredResponseSchema, structured_response_schema),
+            strict=True,
+            # include_raw=True,
         )
         response = await model_with_structured_output.ainvoke(messages, config)
         return {"structured_response": response}

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/agents/components/utils.py

@@ -20,7 +20,7 @@ class AgentStateWithStructuredResponseAndRemainingSteps(AgentStateWithStructured
 
 
 def pre_model_hook_standard(state: T, config: RunnableConfig):
-    _max_messages = config.get("configurable", {}).get("memory_saver_params", {}).get("k", None)
+    _max_messages = config.get("configurable", {}).get("checkpointer_params", {}).get("k", None)
 
     updated_messages = trim_messages(
         state["messages"],

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/client/client.py

@@ -1,7 +1,7 @@
 import json
 import os
 from collections.abc import AsyncGenerator, Generator
-from typing import Any
+from typing import Any, Dict
 
 import httpx
 
@@ -18,6 +18,7 @@ from langgraph_agent_toolkit.schema import (
     MessageInput,
     ServiceMetadata,
     StreamInput,
+    UserComplexInput,
     UserInput,
 )
 
@@ -92,7 +93,7 @@ class AgentClient:
 
     async def ainvoke(
         self,
-        message: str,
+        input: Dict[str, Any],
         model_name: str | None = None,
         model_provider: str | None = None,
         model_config_key: str | None = None,
@@ -104,7 +105,7 @@ class AgentClient:
         """Invoke the agent asynchronously. Only the final message is returned.
 
         Args:
-            message (str): The message to send to the agent
+            input (Dict[str, Any]): The input to send to 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
             model_config_key (str, optional): Key for predefined model configuration
@@ -120,7 +121,7 @@ class AgentClient:
         if not self.agent:
             raise AgentClientError("No agent selected. Use update_agent() to select an agent.")
 
-        request = UserInput(message=message)
+        request = UserInput(input=UserComplexInput(**input))
         if thread_id:
             request.thread_id = thread_id
         if model_name:
@@ -152,7 +153,7 @@ class AgentClient:
 
     def invoke(
         self,
-        message: str,
+        input: Dict[str, Any],
         model_name: str | None = None,
         model_provider: str | None = None,
         model_config_key: str | None = None,
@@ -164,7 +165,7 @@ class AgentClient:
         """Invoke the agent synchronously. Only the final message is returned.
 
         Args:
-            message (str): The message to send to the agent
+            input (Dict[str, Any]): The input to send to 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
             model_config_key (str, optional): Key for predefined model configuration
@@ -180,7 +181,7 @@ class AgentClient:
         if not self.agent:
             raise AgentClientError("No agent selected. Use update_agent() to select an agent.")
 
-        request = UserInput(message=message)
+        request = UserInput(input=UserComplexInput(**input))
         if thread_id:
             request.thread_id = thread_id
         if model_name:
@@ -236,7 +237,7 @@ class AgentClient:
 
     def stream(
         self,
-        message: str,
+        input: Dict[str, Any],
         model_name: str | None = None,
         model_provider: str | None = None,
         model_config_key: str | None = None,
@@ -253,7 +254,7 @@ class AgentClient:
         content tokens from streaming models as they are generated.
 
         Args:
-            message (str): The message to send to the agent
+            input (Dict[str, Any]): The input to send to 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
             model_config_key (str, optional): Key for predefined model configuration
@@ -271,7 +272,7 @@ class AgentClient:
         if not self.agent:
             raise AgentClientError("No agent selected. Use update_agent() to select an agent.")
 
-        request = StreamInput(message=message, stream_tokens=stream_tokens)
+        request = StreamInput(input=UserComplexInput(**input), stream_tokens=stream_tokens)
         if thread_id:
             request.thread_id = thread_id
         if model_name:
@@ -307,7 +308,7 @@ class AgentClient:
 
     async def astream(
         self,
-        message: str,
+        input: Dict[str, Any],
         model_name: str | None = None,
         model_provider: str | None = None,
         model_config_key: str | None = None,
@@ -324,7 +325,7 @@ class AgentClient:
         content tokens from streaming models as they are generated.
 
         Args:
-            message (str): The message to send to the agent
+            input (Dict[str, Any]): The input to send to 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
             model_config_key (str, optional): Key for predefined model configuration
@@ -342,7 +343,7 @@ class AgentClient:
         if not self.agent:
             raise AgentClientError("No agent selected. Use update_agent() to select an agent.")
 
-        request = StreamInput(message=message, stream_tokens=stream_tokens)
+        request = StreamInput(input=UserComplexInput(**input), stream_tokens=stream_tokens)
         if thread_id:
             request.thread_id = thread_id
         if model_name:

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/core/models/factory.py

@@ -87,7 +87,7 @@ class ModelFactory:
         model_name: Optional[str] = None,
         configurable_fields: Optional[Union[Literal["any"], List[str], Tuple[str, ...]]] = None,
         config_prefix: Optional[str] = None,
-        model_parameter_values: Optional[Tuple[Tuple[str, Any], ...]] = None,  # Changed to tuple of tuples
+        model_parameter_values: Optional[Tuple[Tuple[str, Any], ...]] = None,
         **kwargs: Any,
     ) -> ModelT:
         """Create and return a model instance.

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/helper/constants.py

@@ -17,8 +17,8 @@ def set_default_agent(agent_name):
 
 
 DEFAULT_MAX_MESSAGE_HISTORY_LENGTH = os.getenv("DEFAULT_MAX_MESSAGE_HISTORY_LENGTH", 18)
-DEFAULT_RECURSION_LIMIT = 25
-DEFAULT_CONFIG_PREFIX = "agent"
+DEFAULT_RECURSION_LIMIT = os.getenv("DEFAULT_RECURSION_LIMIT", 25)
+DEFAULT_CONFIG_PREFIX = os.getenv("DEFAULT_CONFIG_PREFIX", "agent")
 DEFAULT_CONFIGURABLE_FIELDS = ("temperature", "max_tokens", "top_p", "streaming")
 DEFAULT_MODEL_PARAMETER_VALUES = dict(
     temperature=0.0,

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/helper/utils.py

@@ -1,4 +1,5 @@
 import inspect
+import json
 
 from langchain_core.messages import (
     AIMessage,
@@ -9,7 +10,7 @@ from langchain_core.messages import (
 from langchain_core.messages import (
     ChatMessage as LangchainChatMessage,
 )
-from pydantic import HttpUrl, TypeAdapter
+from pydantic import BaseModel, HttpUrl, TypeAdapter
 
 from langgraph_agent_toolkit.schema import ChatMessage
 
@@ -22,7 +23,9 @@ def check_str_is_http(x: str) -> str:
 def convert_message_content_to_string(content: str | list[str | dict]) -> str:
     if isinstance(content, str):
         return content
+
     text: list[str] = []
+
     for content_item in content:
         if isinstance(content_item, str):
             text.append(content_item)
@@ -32,8 +35,14 @@ def convert_message_content_to_string(content: str | list[str | dict]) -> str:
     return "".join(text)
 
 
-def langchain_to_chat_message(message: BaseMessage) -> ChatMessage:
+def langchain_to_chat_message(message: BaseMessage | dict | BaseModel) -> ChatMessage:
     """Create a ChatMessage from a LangChain message."""
+    if not isinstance(message, (BaseMessage, AIMessage, HumanMessage, ToolMessage, LangchainChatMessage)):
+        if isinstance(message, BaseModel):
+            message = message.model_dump()
+        elif isinstance(message, dict):
+            message = message["raw"].content
+
     match message:
         case HumanMessage():
             human_message = ChatMessage(
@@ -68,6 +77,11 @@ def langchain_to_chat_message(message: BaseMessage) -> ChatMessage:
                 return custom_message
             else:
                 raise ValueError(f"Unsupported chat message role: {message.role}")
+        case str() | dict():
+            return ChatMessage(
+                type="ai",
+                content=message,
+            )
         case _:
             raise ValueError(f"Unsupported message type: {message.__class__.__name__}")
 

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/run_agent.py

@@ -47,7 +47,7 @@ async def main() -> None:
                 "agent_temperature": 0.9,
                 "agent_top_p": 0.75,
                 "agent_max_tokens": 512,
-                "memory_saver_params": {"k": 3},
+                "checkpointer_params": {"k": 3},
             },
             callbacks=[handler],
             recursion_limit=15,

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/run_service.py

@@ -27,6 +27,7 @@ def run_service(
                 "langgraph_agent_toolkit.agents.blueprints.react_so.agent:react_agent_so",
                 "langgraph_agent_toolkit.agents.blueprints.supervisor_agent.agent:supervisor_agent",
                 "langgraph_agent_toolkit.agents.blueprints.chatbot.agent:chatbot_agent",
+                "langgraph_agent_toolkit.agents.blueprints.interrupt_agent.agent:interrupt_agent",
             ]
         ),
     )

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/schema/__init__.py

@@ -13,6 +13,7 @@ from langgraph_agent_toolkit.schema.schema import (
     MessageInput,
     ServiceMetadata,
     StreamInput,
+    UserComplexInput,
     UserInput,
 )
 
@@ -21,6 +22,7 @@ __all__ = [
     "AddMessagesInput",
     "AddMessagesResponse",
     "AgentInfo",
+    "UserComplexInput",
     "UserInput",
     "ChatMessage",
     "ClearHistoryInput",

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/schema/schema.py

@@ -1,4 +1,4 @@
-from typing import Any, Literal, NotRequired
+from typing import Any, Dict, Literal, NotRequired
 
 from pydantic import BaseModel, Field
 from typing_extensions import TypedDict
@@ -35,13 +35,30 @@ class ServiceMetadata(BaseModel):
     )
 
 
-class UserInput(BaseModel):
-    """Basic user input for the agent."""
+class UserComplexInput(BaseModel):
+    """Basic user input for the agent, supporting dynamic fields."""
 
     message: str = Field(
         description="User input to the agent.",
         examples=["What is the weather in Tokyo?"],
     )
+
+    model_config = {
+        "extra": "allow"  # allow unknown fields
+    }
+
+
+class UserInput(BaseModel):
+    """Basic user input for the agent."""
+
+    input: UserComplexInput = Field(
+        description="Structured input from the user, including a message and optional dynamic fields.",
+        examples=[
+            {
+                "message": "What is the weather in Tokyo?",
+            }
+        ],
+    )
     model_name: str | None = Field(
         title="Model",
         description="LLM Model Name to use for the agent.",
@@ -75,7 +92,7 @@ class UserInput(BaseModel):
         default={},
         examples=[
             {
-                "memory_saver_params": {"k": 6},
+                "checkpointer_params": {"k": 6},
                 **DEFAULT_MODEL_PARAMETER_VALUES,
             },
         ],
@@ -115,7 +132,7 @@ class ChatMessage(BaseModel):
         description="Role of the message.",
         examples=["human", "ai", "tool", "custom"],
     )
-    content: str = Field(
+    content: str | Dict[str, Any] = Field(
         description="Content of the message.",
         examples=["Hello, world!"],
     )

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/service/routes.py

@@ -1,5 +1,3 @@
-from typing import Optional
-
 from fastapi import APIRouter, Depends, HTTPException, Request, status
 from fastapi.responses import RedirectResponse, StreamingResponse
 from langchain_core.messages import AnyMessage, RemoveMessage
@@ -83,7 +81,7 @@ async def invoke(user_input: UserInput, agent_id: str = None, request: Request =
     try:
         return await executor.invoke(
             agent_id=agent_id,
-            message=user_input.message,
+            input=user_input.input,
             thread_id=user_input.thread_id,
             user_id=user_input.user_id,
             model_name=user_input.model_name,

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/service/utils.py

@@ -78,7 +78,7 @@ async def message_generator(
     try:
         async for message in executor.stream(
             agent_id=agent_id,
-            message=stream_input.message,
+            input=stream_input.input,
             thread_id=stream_input.thread_id,
             user_id=stream_input.user_id,
             model_name=stream_input.model_name,

{langgraph_agent_toolkit-0.5.0 → langgraph_agent_toolkit-0.7.0}/langgraph_agent_toolkit/streamlit_app.py

@@ -187,14 +187,14 @@ async def main() -> None:
         try:
             if use_streaming:
                 stream = agent_client.astream(
-                    message=user_input,
+                    input=dict(message=user_input),
                     thread_id=st.session_state.thread_id,
                     user_id=DEFAULT_STREAMLIT_USER_ID,
                 )
                 await draw_messages(stream, is_new=True)
             else:
                 response = await agent_client.ainvoke(
-                    message=user_input,
+                    input=dict(message=user_input),
                     thread_id=st.session_state.thread_id,
                     user_id=DEFAULT_STREAMLIT_USER_ID,
                 )