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

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

@@ -6,6 +6,22 @@ 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.5.0]
+
+### Fixed
+
+- Streamlit UI bugs
+- Windows compatibility issue
+- Enhance message handling inside `pre_hook_model`
+- Add prompt hash to Langfuse observability class
+- Rename few parameters
+
+## [0.4.5]
+
+### Fixed
+
+- Strucuted output and model factory
+
 ## [0.4.4]
 
 ### Updated

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.4.4
+Version: 0.5.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
@@ -40,6 +40,7 @@ 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: pandas~=2.2.3
 Requires-Dist: psycopg[binary,pool]~=3.2.4
 Requires-Dist: pyarrow>=19.0.1
@@ -299,10 +300,14 @@ and includes:
 - [LangGraph documentation](https://langchain-ai.github.io/langgraph/concepts/low_level/#multiple-schemas)
 - [LangGraph Memory Concept](https://langchain-ai.github.io/langgraph/concepts/memory/)
 - [LangGraph Memory Persistence](https://langchain-ai.github.io/langgraph/concepts/persistence/#memory)
+- [LangGraph Memory Template](https://github.com/langchain-ai/memory-template)
+- [LangGraph Human in the Loop](https://langchain-ai.github.io/langgraph/how-tos/human_in_the_loop/wait-user-input/)
 - [LangGraph 101 - blueprints](https://github.com/langchain-ai/langgraph-101)
 - [LangGraph - Examples](https://github.com/langchain-ai/langgraph/tree/main/examples)
 - [Complex data extraction with function calling](https://langchain-ai.github.io/langgraph/tutorials/extraction/retries/)
 - [How to edit graph state](https://langchain-ai.github.io/langgraph/how-tos/human_in_the_loop/edit-graph-state/)
+- [Memory in the background](https://www.youtube.com/watch?v=R1jKQ1Jn5T4&ab_channel=LangChain)
+- [Building an agent with LangGraph](https://www.kaggle.com/code/markishere/day-3-building-an-agent-with-langgraph/)
 - [How to create tools in Langchain](https://python.langchain.com/docs/how_to/custom_tools/)
 - [Simple Serverless FastAPI with AWS Lambda](https://www.deadbear.io/simple-serverless-fastapi-with-aws-lambda/)
 

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/README.md

@@ -197,10 +197,14 @@ and includes:
 - [LangGraph documentation](https://langchain-ai.github.io/langgraph/concepts/low_level/#multiple-schemas)
 - [LangGraph Memory Concept](https://langchain-ai.github.io/langgraph/concepts/memory/)
 - [LangGraph Memory Persistence](https://langchain-ai.github.io/langgraph/concepts/persistence/#memory)
+- [LangGraph Memory Template](https://github.com/langchain-ai/memory-template)
+- [LangGraph Human in the Loop](https://langchain-ai.github.io/langgraph/how-tos/human_in_the_loop/wait-user-input/)
 - [LangGraph 101 - blueprints](https://github.com/langchain-ai/langgraph-101)
 - [LangGraph - Examples](https://github.com/langchain-ai/langgraph/tree/main/examples)
 - [Complex data extraction with function calling](https://langchain-ai.github.io/langgraph/tutorials/extraction/retries/)
 - [How to edit graph state](https://langchain-ai.github.io/langgraph/how-tos/human_in_the_loop/edit-graph-state/)
+- [Memory in the background](https://www.youtube.com/watch?v=R1jKQ1Jn5T4&ab_channel=LangChain)
+- [Building an agent with LangGraph](https://www.kaggle.com/code/markishere/day-3-building-an-agent-with-langgraph/)
 - [How to create tools in Langchain](https://python.langchain.com/docs/how_to/custom_tools/)
 - [Simple Serverless FastAPI with AWS Lambda](https://www.deadbear.io/simple-serverless-fastapi-with-aws-lambda/)
 

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/langgraph_agent_toolkit/agents/blueprints/react/agent.py

@@ -36,7 +36,7 @@ observability.push_prompt(
             content="messages",
         ),
     ],
-    create_new_version=False,
+    force_create_new_version=False,
 )
 
 prompt = ObservabilityChatPromptTemplate.from_observability_platform(

langgraph_agent_toolkit-0.5.0/langgraph_agent_toolkit/agents/components/utils.py

@@ -0,0 +1,40 @@
+from typing import TypeVar
+
+from langchain_core.messages.utils import trim_messages
+from langchain_core.runnables import RunnableConfig
+from langgraph.managed.is_last_step import RemainingSteps
+from langgraph.prebuilt.chat_agent_executor import AgentState, AgentStateWithStructuredResponse
+
+from langgraph_agent_toolkit.helper.constants import DEFAULT_MAX_MESSAGE_HISTORY_LENGTH
+
+
+T = TypeVar("T")
+
+
+class AgentStateWithRemainingSteps(AgentState):
+    remaining_steps: RemainingSteps
+
+
+class AgentStateWithStructuredResponseAndRemainingSteps(AgentStateWithStructuredResponse):
+    remaining_steps: RemainingSteps
+
+
+def pre_model_hook_standard(state: T, config: RunnableConfig):
+    _max_messages = config.get("configurable", {}).get("memory_saver_params", {}).get("k", None)
+
+    updated_messages = trim_messages(
+        state["messages"],
+        token_counter=len,
+        max_tokens=_max_messages or DEFAULT_MAX_MESSAGE_HISTORY_LENGTH,
+        strategy="last",
+        start_on="human",
+        end_on=("human", "tool"),
+        include_system=True,
+        allow_partial=False,
+    )
+
+    return {"llm_input_messages": updated_messages}
+
+
+def default_pre_model_hook(state: T, config: RunnableConfig) -> T:
+    return state

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.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[dict[str, Any]] = None,
+        model_parameter_values: Optional[Tuple[Tuple[str, Any], ...]] = None,  # Changed to tuple of tuples
         **kwargs: Any,
     ) -> ModelT:
         """Create and return a model instance.
@@ -97,7 +97,8 @@ class ModelFactory:
             model_name: The name of the model to use. If not provided, the default model name will be used.
             configurable_fields: The fields that are configurable. If not provided, the default fields will be used.
             config_prefix: The prefix to use for the configuration. If not provided, the default prefix will be used.
-            model_parameter_values: The values for the model parameters. If not provided, the default values will be used.
+            model_parameter_values: The values for the model parameters as a tuple of (key, value) pairs.
+                                    If not provided, the default values will be used.
             **kwargs: Additional keyword arguments to pass to the model.
 
         Returns:
@@ -108,8 +109,10 @@ class ModelFactory:
 
         """  # noqa: E501
         _configurable_fields = DEFAULT_CONFIGURABLE_FIELDS if configurable_fields is None else configurable_fields
-        _config_prefix = config_prefix or DEFAULT_CONFIG_PREFIX
-        _model_parameter_values = model_parameter_values or DEFAULT_MODEL_PARAMETER_VALUES
+        _config_prefix = DEFAULT_CONFIG_PREFIX if config_prefix is None else config_prefix
+        _model_parameter_values = (
+            DEFAULT_MODEL_PARAMETER_VALUES if model_parameter_values is None else dict(model_parameter_values)
+        )
 
         match model_provider:
             case ModelProvider.FAKE:
@@ -165,4 +168,4 @@ class ModelFactory:
         params.update(override_params)
 
         # Create and return the model
-        return cls.create(model_provider=provider, model_name=model_name, **params)
+        return cls.create(model_provider=provider, model_name=model_name, model_parameter_values=(), **params)

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/langgraph_agent_toolkit/core/observability/base.py

@@ -89,7 +89,7 @@ class BaseObservabilityPlatform(ABC):
     def _handle_existing_prompt(
         self,
         name: str,
-        create_new_version: bool = True,
+        force_create_new_version: bool = True,
         client: Any = None,
         client_pull_method: Optional[str] = None,
         client_delete_method: Optional[str] = None,
@@ -106,11 +106,11 @@ class BaseObservabilityPlatform(ABC):
         if not pull_method or not delete_method:
             return (existing_prompt, url)
 
-        if not create_new_version:
+        if not force_create_new_version:
             try:
                 existing_prompt = pull_method(name=name)
                 url = getattr(existing_prompt, "url", None)
-                logger.debug(f"Using existing prompt '{name}' as create_new_version is False")
+                logger.debug(f"Using existing prompt '{name}' as force_create_new_version is False")
             except Exception:
                 logger.debug(f"Existing prompt '{name}' not found, will create a new one")
         else:
@@ -275,14 +275,14 @@ class BaseObservabilityPlatform(ABC):
         name: str,
         prompt_template: PromptTemplateType,
         metadata: Optional[Dict[str, Any]] = None,
-        create_new_version: bool = True,
+        force_create_new_version: bool = True,
     ) -> None:
         self._prompts_dir.mkdir(exist_ok=True, parents=True)
 
         file_path = self._prompts_dir / f"{name}.jinja2"
         metadata_path = self._prompts_dir / f"{name}.metadata.joblib"
 
-        if create_new_version:
+        if force_create_new_version:
             if file_path.exists():
                 file_path.unlink()
             if metadata_path.exists():

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/langgraph_agent_toolkit/core/observability/empty.py

@@ -35,7 +35,7 @@ class EmptyObservability(BaseObservabilityPlatform):
         name: str,
         prompt_template: PromptTemplateType,
         metadata: Optional[Dict[str, Any]] = None,
-        create_new_version: bool = True,
+        force_create_new_version: bool = True,
     ) -> None:
         """Push a prompt using local storage.
 
@@ -43,10 +43,10 @@ class EmptyObservability(BaseObservabilityPlatform):
             name: Name of the prompt
             prompt_template: String template, list of message dicts, or prompt object
             metadata: Additional metadata for the prompt
-            create_new_version: If True, overwrite existing prompt with new version
+            force_create_new_version: If True, overwrite existing prompt with new version
 
         """
-        super().push_prompt(name, prompt_template, metadata, create_new_version)
+        super().push_prompt(name, prompt_template, metadata, force_create_new_version)
 
     def pull_prompt(
         self,

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/langgraph_agent_toolkit/core/observability/langfuse.py

@@ -1,3 +1,5 @@
+import hashlib
+import json
 from typing import Any, Dict, Literal, Optional, Tuple, Union
 
 from langfuse import Langfuse
@@ -32,44 +34,85 @@ class LangfuseObservability(BaseObservabilityPlatform):
             **kwargs,
         )
 
+    def _compute_prompt_hash(self, prompt_template: PromptTemplateType) -> str:
+        """Compute a hash of the prompt content to detect changes."""
+        if isinstance(prompt_template, str):
+            content_to_hash = prompt_template
+        elif isinstance(prompt_template, list):
+            content_to_hash = json.dumps(prompt_template, sort_keys=True)
+        else:
+            content_to_hash = str(prompt_template)
+
+        return hashlib.md5(content_to_hash.encode("utf-8")).hexdigest()
+
     @BaseObservabilityPlatform.requires_env_vars
     def push_prompt(
         self,
         name: str,
         prompt_template: PromptTemplateType,
         metadata: Optional[Dict[str, Any]] = None,
-        create_new_version: bool = True,
+        force_create_new_version: bool = True,
     ) -> None:
         langfuse = Langfuse()
         labels = metadata.get("labels", ["production"]) if metadata else ["production"]
 
+        # Generate hash for the current prompt
+        prompt_hash = self._compute_prompt_hash(prompt_template)
+
         # Handle existing prompt versions - custom implementation for Langfuse
         existing_prompt = None
-        if not create_new_version:
-            try:
-                existing_prompt = langfuse.get_prompt(name=name)
-                logger.debug(f"Using existing prompt '{name}' as create_new_version is False")
-            except Exception:
-                logger.debug(f"Existing prompt '{name}' not found, will create a new one")
+        content_changed = True
+
+        try:
+            existing_prompt = langfuse.get_prompt(name=name)
+
+            # Check if content has changed by comparing hashes
+            if existing_prompt.commit_message and existing_prompt.commit_message == prompt_hash:
+                content_changed = False
+                logger.debug(f"Prompt '{name}' content unchanged")
+            else:
+                logger.debug(f"Prompt '{name}' content changed from previous version")
+        except Exception:
+            logger.debug(f"Existing prompt '{name}' not found, will create a new one")
 
         prompt_obj = self._convert_to_chat_prompt(prompt_template)
         type_prompt = "text" if isinstance(prompt_template, str) else "chat"
 
-        if existing_prompt:
-            langfuse_prompt = existing_prompt
-        else:
+        create_new = (
+            # When force_create_new_version=True, always create a new version
+            force_create_new_version
+            or
+            # When no existing prompt, create a new one
+            existing_prompt is None
+            or
+            # When content has changed and we're not forcing to keep the old version
+            content_changed
+        )
+
+        if create_new:
             langfuse_prompt = langfuse.create_prompt(
                 name=name,
                 prompt=prompt_template,
                 labels=labels,
                 type=type_prompt,
+                commit_message=prompt_hash,  # Store hash in commit_message
             )
+            if existing_prompt is None:
+                logger.debug(f"Created new prompt '{name}' as it didn't exist before")
+            elif content_changed:
+                logger.debug(f"Created new prompt version '{name}' because content changed (hash: {prompt_hash})")
+            else:
+                logger.debug(f"Created new prompt version '{name}' because force_create_new_version=True")
+        else:
+            # Content unchanged and not forcing new version
+            langfuse_prompt = existing_prompt
+            logger.debug(f"Reusing existing prompt '{name}' as content is unchanged and force_create_new_version=False")
 
         full_metadata = metadata.copy() if metadata else {}
         full_metadata["langfuse_prompt"] = langfuse_prompt
         full_metadata["original_prompt"] = prompt_obj
 
-        super().push_prompt(name, prompt_template, full_metadata)
+        super().push_prompt(name, prompt_template, full_metadata, force_create_new_version)
 
     @BaseObservabilityPlatform.requires_env_vars
     def pull_prompt(

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/langgraph_agent_toolkit/core/observability/langsmith.py

@@ -54,7 +54,7 @@ class LangsmithObservability(BaseObservabilityPlatform):
         name: str,
         prompt_template: PromptTemplateType,
         metadata: Optional[Dict[str, Any]] = None,
-        create_new_version: bool = True,
+        force_create_new_version: bool = True,
     ) -> None:
         """Push a prompt to LangSmith."""
         client = LangsmithClient()
@@ -64,7 +64,11 @@ class LangsmithObservability(BaseObservabilityPlatform):
 
         # Handle existing prompt versions
         existing_prompt, existing_url = self._handle_existing_prompt(
-            name, create_new_version, client, client_pull_method="pull_prompt", client_delete_method="delete_prompt"
+            name,
+            force_create_new_version,
+            client,
+            client_pull_method="pull_prompt",
+            client_delete_method="delete_prompt",
         )
 
         url = None

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

@@ -1,3 +1,6 @@
+import os
+
+
 # This is the initial default agent name, but it may be overridden at runtime
 DEFAULT_AGENT = "react-agent"
 _CURRENT_DEFAULT_AGENT = DEFAULT_AGENT
@@ -13,7 +16,7 @@ def set_default_agent(agent_name):
     return _CURRENT_DEFAULT_AGENT
 
 
-DEFAULT_MAX_MESSAGE_HISTORY_LENGTH = 6 + 1  # N messages + 1 system message
+DEFAULT_MAX_MESSAGE_HISTORY_LENGTH = os.getenv("DEFAULT_MAX_MESSAGE_HISTORY_LENGTH", 18)
 DEFAULT_RECURSION_LIMIT = 25
 DEFAULT_CONFIG_PREFIX = "agent"
 DEFAULT_CONFIGURABLE_FIELDS = ("temperature", "max_tokens", "top_p", "streaming")

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

@@ -30,6 +30,20 @@ APP_TITLE = "Agent Service Toolkit"
 APP_ICON = "🧰"
 
 
+def create_welcome_message(agent: str) -> ChatMessage:
+    """Create a welcome message based on the current agent."""
+    match agent:
+        case "chatbot":
+            welcome_content = "Hello! I'm a simple chatbot. Ask me anything!"
+        case "interrupt-agent":
+            welcome_content = (
+                "Hello! I'm an interrupt agent. Tell me your birthday and I will predict your personality!"
+            )
+        case _:
+            welcome_content = "Hello! I'm an AI agent. Ask me anything!"
+    return ChatMessage(type="ai", content=welcome_content)
+
+
 async def main() -> None:
     st.set_page_config(
         page_title=APP_TITLE,
@@ -76,18 +90,7 @@ async def main() -> None:
             thread_id = str(uuid.uuid4())
             messages = []
             # Add welcome message to messages when creating a new thread
-            welcome_message = None
-            match agent_client.agent:
-                case "chatbot":
-                    welcome_content = "Hello! I'm a simple chatbot. Ask me anything!"
-                case "interrupt-agent":
-                    welcome_content = (
-                        "Hello! I'm an interrupt agent. Tell me your birthday and I will predict your personality!"
-                    )
-                case _:
-                    welcome_content = "Hello! I'm an AI agent. Ask me anything!"
-            welcome_message = ChatMessage(type="ai", content=welcome_content)
-            messages.append(welcome_message)
+            messages.append(create_welcome_message(agent_client.agent))
         else:
             try:
                 messages: List[ChatMessage] = agent_client.get_history(
@@ -111,6 +114,8 @@ async def main() -> None:
         if st.button(":material/chat: New Chat", use_container_width=True):
             st.session_state.messages = []
             st.session_state.thread_id = str(uuid.uuid4())
+            # Add welcome message to new chat
+            st.session_state.messages.append(create_welcome_message(agent_client.agent))
             st.rerun()
 
         with st.popover(":material/settings: Settings", use_container_width=True):

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "langgraph-agent-toolkit"
-version = "0.4.4"
+version = "0.5.0"
 description = "Full toolkit for running an AI agent service built with LangGraph, FastAPI and Streamlit"
 readme = "README.md"
 authors = [{ name = "Roman Kryvokhyzha", email = "kriwohizha@gmail.com" }, { name = "Joshua Carroll", email = "carroll.joshk@gmail.com" }]
@@ -43,6 +43,7 @@ dependencies = [
     "langgraph-supervisor ~= 0.0.21",
     "numpy ~= 1.26.4; python_version <= '3.12'",
     "numpy ~= 2.2.3; python_version >= '3.13'",
+    "onnxruntime ~= 1.21.1",
     "pandas ~= 2.2.3",
     "psycopg[binary,pool] ~= 3.2.4",
     "pyarrow >= 19.0.1",

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/01-invoke-proxy.py

@@ -1,14 +1,22 @@
 import os
 
+import rootutils
 from dotenv import find_dotenv, load_dotenv
+
+
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langchain.schema import HumanMessage
 from langchain_openai import ChatOpenAI
 from pydantic import BaseModel, Field
 
 
-load_dotenv(find_dotenv())
-
-
 class Response(BaseModel):
     answer: str = Field(..., description="The answer to the question.")
     similar_questions: list[str] = Field(..., description="A list of similar questions to the one asked. Limit: 3")

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/02-test-local-pm.py

@@ -1,11 +1,19 @@
 import rootutils
 from dotenv import find_dotenv, load_dotenv
 
+
+PATH_TO_ROOT = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langgraph_agent_toolkit.core.observability.empty import EmptyObservability
 
 
-load_dotenv(find_dotenv())
-custom_dir = rootutils.find_root(__file__, indicator=".project-root") / "data" / "prompts"
+custom_dir = PATH_TO_ROOT / "data" / "prompts"
 
 
 if __name__ == "__main__":

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/03-test-langfuse-pm.py

@@ -1,9 +1,16 @@
+import rootutils
 from dotenv import find_dotenv, load_dotenv
 
-from langgraph_agent_toolkit.core.observability.langfuse import LangfuseObservability
 
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
 
-load_dotenv(find_dotenv())
+from langgraph_agent_toolkit.core.observability.langfuse import LangfuseObservability
 
 
 if __name__ == "__main__":
@@ -22,6 +29,7 @@ Additional context:
     observability.push_prompt(
         name="joke-generator",
         prompt_template=joke_template,
+        force_create_new_version=True,
     )
     print("Prompt pushed to Langfuse")
 

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/04-test-langsmith-pm.py

@@ -1,9 +1,16 @@
+import rootutils
 from dotenv import find_dotenv, load_dotenv
 
-from langgraph_agent_toolkit.core.observability.langsmith import LangsmithObservability
 
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
 
-load_dotenv(find_dotenv())
+from langgraph_agent_toolkit.core.observability.langsmith import LangsmithObservability
 
 
 if __name__ == "__main__":

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/05-test-prompt-types.py

@@ -1,11 +1,19 @@
+import rootutils
 from dotenv import find_dotenv, load_dotenv
 
+
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langgraph_agent_toolkit.core.observability.factory import ObservabilityFactory
 from langgraph_agent_toolkit.core.observability.types import ChatMessageDict, ObservabilityBackend
 
 
-load_dotenv(find_dotenv())
-
 if __name__ == "__main__":
     # Create observability instance using the factory
     observability = ObservabilityFactory.create(ObservabilityBackend.EMPTY)

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/06-test-custom-chat-prompt-template.py

@@ -1,10 +1,20 @@
 import time
 from typing import List
 
+import rootutils
 from dotenv import find_dotenv, load_dotenv
+
+
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
-from langgraph.prebuilt import create_react_agent
 
 from langgraph_agent_toolkit.core.observability.factory import ObservabilityFactory
 from langgraph_agent_toolkit.core.observability.types import ChatMessageDict, ObservabilityBackend
@@ -25,9 +35,6 @@ def print_separator(title: str = None):
         print(f"\n{'=' * width}")
 
 
-# Load environment variables
-load_dotenv(find_dotenv())
-
 if __name__ == "__main__":
     print_separator("ObservabilityChatPromptTemplate Demo")
 

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/07-test-get-chat-history.py

@@ -1,14 +1,19 @@
-import sys
-
+import rootutils
 from dotenv import find_dotenv, load_dotenv
+
+
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langchain_core.messages import AnyMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.func import Pregel
 
-
-load_dotenv(find_dotenv(filename=".local.env"), override=True)
-sys.path.append(".")
-
 from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.agents.blueprints.react.agent import react_agent
 from langgraph_agent_toolkit.core.memory.factory import MemoryFactory

{langgraph_agent_toolkit-0.4.4 → langgraph_agent_toolkit-0.5.0}/scripts/python/08-test-update-chat-history.py

@@ -1,14 +1,19 @@
-import sys
-
+import rootutils
 from dotenv import find_dotenv, load_dotenv
+
+
+_ = rootutils.setup_root(
+    search_from=__file__,
+    indicator=".project-root",
+    pythonpath=True,
+    dotenv=False,
+)
+load_dotenv(find_dotenv(".local.env"), override=True)
+
 from langchain_core.messages import AnyMessage
 from langchain_core.runnables import RunnableConfig
 from langgraph.func import Pregel
 
-
-load_dotenv(find_dotenv(filename=".local.env"), override=True)
-sys.path.append(".")
-
 from langgraph_agent_toolkit.agents.agent import Agent
 from langgraph_agent_toolkit.agents.blueprints.react.agent import react_agent
 from langgraph_agent_toolkit.core.memory.factory import MemoryFactory