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

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

@@ -48,10 +48,10 @@ POSTGRES_MAX_IDLE=5
 USE_FAKE_MODEL=false
 
 # Set a default model
-DEFAULT_MODEL="openai-compatible"
+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
-COMPATIBLE_MODEL=
-COMPATIBLE_API_KEY=
-COMPATIBLE_BASE_URL=http://litellm:4000/v1
+MODEL_NAME=
+MODEL_API_KEY=
+MODEL_BASE_URL=http://litellm:4000/v1

{langgraph_agent_toolkit-0.1.0 → langgraph_agent_toolkit-0.1.2}/.github/workflows/test.yml

@@ -3,8 +3,18 @@ name: Build and test
 on:
   push:
     branches: ["main"]
+    paths:
+      - "docker/**"
+      - "langgraph_agent_toolkit/**"
+      - "tests/**"
+      - "pyproject.toml"
   pull_request:
     branches: ["main"]
+    paths:
+      - "docker/**"
+      - "langgraph_agent_toolkit/**"
+      - "tests/**"
+      - "pyproject.toml"
   workflow_call:
 
 permissions:
@@ -41,9 +51,9 @@ jobs:
         run: |
           uv run pytest --cov=langgraph_agent_toolkit/ --cov-report=xml
         env:
-          COMPATIBLE_MODEL: ${{ secrets.COMPATIBLE_MODEL }}
-          COMPATIBLE_API_KEY: ${{ secrets.COMPATIBLE_API_KEY }}
-          COMPATIBLE_BASE_URL: ${{ secrets.COMPATIBLE_BASE_URL }}
+          MODEL_NAME: ${{ secrets.MODEL_NAME }}
+          MODEL_API_KEY: ${{ secrets.MODEL_API_KEY }}
+          MODEL_BASE_URL: ${{ secrets.MODEL_BASE_URL }}
 
       - name: Upload coverage reports to Codecov
         uses: codecov/codecov-action@v5

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

@@ -6,6 +6,37 @@ 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.1.2]
+
+### Added
+
+- `user_id` parameter
+- `store` creator to memory classes
+
+### Fixed
+
+- enhance error handling and testing in Streamlit app
+- add new chat button
+- variable names
+- type hints
+
+### Removed
+
+- print statements
+
+## [0.1.1]
+
+### Added
+
+- `get_default_agent` and `set_default_agent` functions
+
+### Fixed
+
+- Minor fixes
+- Refactoring
+- Update dependencies
+- Update README
+
 ## [0.1.0]
 
 ### Changed

{langgraph_agent_toolkit-0.1.0 → langgraph_agent_toolkit-0.1.2}/CONTRIBUTING.md

@@ -4,59 +4,60 @@ First off, thank you for considering contributing to `Langgraph Agent Toolkit`!
 
 ## Development Setup
 
-1.  Make sure you have Python 3.10+ installed
-2.  Install [uv](https://docs.astral.sh/uv/getting-started/installation/)
-    package manager
-3.  Fork the repository
-4.  Clone your fork
+1. Make sure you have Python 3.10+ installed
+2. Install [uv](https://docs.astral.sh/uv/getting-started/installation/) package
+   manager
+3. Fork the repository
+4. Clone your fork
 
-    ```bash
-    git clone https://github.com/YOUR-USERNAME/langgraph-agent-toolkit.git
-    cd langgraph-agent-toolkit
+   ```bash
+   git clone https://github.com/YOUR-USERNAME/langgraph-agent-toolkit.git
+   cd langgraph-agent-toolkit
 
-    # Add the upstream remote
-    git remote add upstream https://github.com/kryvokhyzha/langgraph-agent-toolkit.git
-    ```
+   # Add the upstream remote
+   git remote add upstream https://github.com/kryvokhyzha/langgraph-agent-toolkit.git
+   ```
 
-5.  Set up the development environment:
+5. Set up the development environment:
 
-    ```bash
-    uv sync
-    ```
+   ```bash
+   uv sync
+   ```
 
-    That's it! The `uv sync` command will automatically create and use a virtual
-    environment.
+   That's it! The `uv sync` command will automatically create and use a virtual
+   environment.
 
-6.  Install pre-commit hooks:
+6. Install pre-commit hooks:
 
-    ```bash
-    uv run pre-commit install
-    uv run pre-commit run
-    ```
+   ```bash
+   uv run pre-commit install
+   uv run pre-commit run
+   ```
 
-    Pre-commit hooks will automatically run checks (like ruff, formatting, etc.)
-    when you make a commit, ensuring your code follows our style guidelines.
+   Pre-commit hooks will automatically run checks (like ruff, formatting, etc.)
+   when you make a commit, ensuring your code follows our style guidelines.
 
 ### Running Commands
 
 You have two options for running commands:
 
-1.  **With the virtual environment activated**:
+1. **With the virtual environment activated**:
 
-    ```bash
-    source .venv/bin/activate  # On Windows: .venv\Scripts\activate
+   ```bash
+   source .venv/bin/activate  # On Windows: .venv\Scripts\activate
 
-    # Then run commands directly
-    pytest
-    pre-commit run --all-files
-    ```
+   # Then run commands directly
+   pytest
+   pre-commit run --all-files
+   ```
 
-2.  **Without activating the virtual environment**:
-    ```bash
-    # Use uv run prefix for all commands
-    uv run pytest
-    uv run pre-commit run --all-files
-    ```
+2. **Without activating the virtual environment**:
+
+   ```bash
+   # Use uv run prefix for all commands
+   uv run pytest
+   uv run pre-commit run --all-files
+   ```
 
 Both approaches work - use whichever is more convenient for you.
 
@@ -69,26 +70,28 @@ Both approaches work - use whichever is more convenient for you.
 
 When adding new dependencies to the library:
 
-1.  **Runtime dependencies** - packages needed to run the application:
+1. **Runtime dependencies** - packages needed to run the application:
+
+   ```bash
+   uv add new-package
+   ```
 
-    ```bash
-    uv add new-package
-    ```
+2. **Development dependencies** - packages needed for development, testing, or
+   CI:
 
-2.  **Development dependencies** - packages needed for development, testing, or
-    CI:
-    ```bash
-    uv add --group dev new-package
-    ```
+   ```bash
+   uv add --group dev new-package
+   ```
 
 After adding dependencies, make sure to:
 
-1.  Test that everything works with the new package
-2.  Commit both `pyproject.toml` and `uv.lock` files:
-    ```bash
-    git add pyproject.toml uv.lock
-    git commit -m "Add new-package dependency"
-    ```
+1. Test that everything works with the new package
+2. Commit both `pyproject.toml` and `uv.lock` files:
+
+   ```bash
+   git add pyproject.toml uv.lock
+   git commit -m "Add new-package dependency"
+   ```
 
 ## Development Process
 

{langgraph_agent_toolkit-0.1.0 → langgraph_agent_toolkit-0.1.2}/Makefile

@@ -35,7 +35,7 @@ pre_commit_rm_hooks:
 	pre-commit --uninstall-hooks
 
 push_new_tag:
-	sh .github/release/tag_from_pyproject.sh
+	sh .github/scripts/tag_from_pyproject.sh
 
 run_langgraph:
 	uv run langgraph dev

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

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.1.0
+Version: 0.1.2
 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
 Author-email: Roman Kryvokhyzha <kriwohizha@gmail.com>
 License-Expression: MIT
 License-File: LICENSE
@@ -36,7 +37,6 @@ Requires-Dist: langgraph-cli[inmem]~=0.2.3
 Requires-Dist: langgraph-supervisor~=0.0.16
 Requires-Dist: langgraph~=0.3.30
 Requires-Dist: langsmith~=0.3.31
-Requires-Dist: litellm~=1.66.1
 Requires-Dist: loguru~=0.7.3
 Requires-Dist: multidict>6.3.2
 Requires-Dist: numpy~=1.26.4; python_version <= '3.12'
@@ -51,7 +51,6 @@ Requires-Dist: python-dotenv~=1.1.0
 Requires-Dist: rootutils>=1.0.7
 Requires-Dist: setuptools>=75.6.0
 Requires-Dist: streamlit~=1.44.1
-Requires-Dist: tiktoken>=0.8.0
 Requires-Dist: uvicorn~=0.34.0
 Requires-Dist: watchdog~=6.0.0
 Provides-Extra: aws
@@ -72,13 +71,15 @@ Description-Content-Type: text/markdown
 
 # 🧰 Langgraph Agent Toolkit
 
-<!-- [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) -->
-
 [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/)
+[![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/)
+[![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
 [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit)
 [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml)
-[![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)
+
+<!-- [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE) -->
 
 <!-- [![Streamlit App](https://static.streamlit.io/badges/streamlit_badge_black_red.svg)](https://langgraph-agent-toolkit.streamlit.app/) -->
 
@@ -102,7 +103,7 @@ Features include:
 
 1. Create a `.env` file based on [`.env.example`](./.env.example)
 
-2. **Option 1: Run with Python**
+2. **Option 1: Run with Python from source**
 
    ```sh
    # Install dependencies
@@ -118,7 +119,14 @@ Features include:
    streamlit run langgraph_agent_toolkit/streamlit_app.py
    ```
 
-3. **Option 2: Run with Docker**
+3. **Option 2: Run with Python from PyPi repository**
+
+   ```sh
+   pip install langgraph-agent-toolkit
+   ```
+
+4. **Option 3: Run with Docker**
+
    ```sh
    docker compose watch
    ```
@@ -176,9 +184,9 @@ Features include:
 
    ```env
    # LiteLLM configuration
-   COMPATIBLE_MODEL="gpt4o"                        # LiteLLM model name
-   COMPATIBLE_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
-   COMPATIBLE_API_KEY=sk-any-string                # LiteLLM master key
+   MODEL_NAME="gpt4o"                         # LiteLLM model name
+   MODEL_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
+   MODEL_API_KEY=sk-any-string                # LiteLLM master key
    ```
 
    **🗄️ Database Configuration**
@@ -335,15 +343,14 @@ enables live reloading:
 
 3. Access endpoints:
 
-   - 🖥️ Streamlit app: `http://localhost:8501`
+   - 🖥️ Streamlit app: `http://0.0.0.0:8501`
    - 🔌 Agent API: `http://0.0.0.0:8080`
-   - 📚 API docs: `http://0.0.0.0:8080/redoc`
-   - 📊 Langfuse dashboard: `http://localhost:3000`
-   - 📦 MinIO console: `http://localhost:9091`
-   - 🤖 LiteLLM API: `http://localhost:4000`
-   - 📈 ClickHouse HTTP: `http://localhost:8123`
+   - 📚 API docs: `http://0.0.0.0:8080/docs`
+   - 📊 Langfuse dashboard: `http://0.0.0.0:3000`
+   - 🤖 LiteLLM API: `http://0.0.0.0:4000`
 
 4. Stop services:
+
    ```sh
    docker compose down
    ```
@@ -409,6 +416,8 @@ The project works with
 ## 📚 Useful Resources
 
 - [LangGraph documentation](https://langchain-ai.github.io/langgraph/concepts/low_level/#multiple-schemas)
+- [LangGraph Memory Concept](https://langchain-ai.github.io/langgraph/concepts/memory/)
+- [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/)
 
 ## 👥 Development and Contributing

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

@@ -1,12 +1,14 @@
 # 🧰 Langgraph Agent Toolkit
 
-<!-- [![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/) -->
-
 [![build status](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml/badge.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/actions/workflows/test.yml)
-[![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![PyPI version](https://img.shields.io/pypi/v/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/)
+[![PyPI Downloads](https://img.shields.io/pypi/dm/langgraph-agent-toolkit.svg)](https://pypi.org/project/langgraph-agent-toolkit/)
+[![Code style: ruff](https://img.shields.io/badge/code%20style-ruff-000000.svg)](https://github.com/astral-sh/ruff)
 [![codecov](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit/graph/badge.svg?token=OHSACTNSWZ)](https://codecov.io/gh/kryvokhyzha/langgraph-agent-toolkit)
 [![Python Version](https://img.shields.io/python/required-version-toml?tomlFilePath=https%3A%2F%2Fraw.githubusercontent.com%2Fkryvokhyzha%2Flanggraph-agent-toolkit%2Frefs%2Fheads%2Fmain%2Fpyproject.toml)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/pyproject.toml)
-[![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE)
+
+<!-- [![GitHub License](https://img.shields.io/github/license/kryvokhyzha/langgraph-agent-toolkit)](https://github.com/kryvokhyzha/langgraph-agent-toolkit/blob/main/LICENSE) -->
 
 <!-- [![Streamlit App](https://static.streamlit.io/badges/streamlit_badge_black_red.svg)](https://langgraph-agent-toolkit.streamlit.app/) -->
 
@@ -30,7 +32,7 @@ Features include:
 
 1. Create a `.env` file based on [`.env.example`](./.env.example)
 
-2. **Option 1: Run with Python**
+2. **Option 1: Run with Python from source**
 
    ```sh
    # Install dependencies
@@ -46,7 +48,14 @@ Features include:
    streamlit run langgraph_agent_toolkit/streamlit_app.py
    ```
 
-3. **Option 2: Run with Docker**
+3. **Option 2: Run with Python from PyPi repository**
+
+   ```sh
+   pip install langgraph-agent-toolkit
+   ```
+
+4. **Option 3: Run with Docker**
+
    ```sh
    docker compose watch
    ```
@@ -104,9 +113,9 @@ Features include:
 
    ```env
    # LiteLLM configuration
-   COMPATIBLE_MODEL="gpt4o"                        # LiteLLM model name
-   COMPATIBLE_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
-   COMPATIBLE_API_KEY=sk-any-string                # LiteLLM master key
+   MODEL_NAME="gpt4o"                         # LiteLLM model name
+   MODEL_BASE_URL=http://litellm:4000/v1      # LiteLLM endpoint
+   MODEL_API_KEY=sk-any-string                # LiteLLM master key
    ```
 
    **🗄️ Database Configuration**
@@ -263,15 +272,14 @@ enables live reloading:
 
 3. Access endpoints:
 
-   - 🖥️ Streamlit app: `http://localhost:8501`
+   - 🖥️ Streamlit app: `http://0.0.0.0:8501`
    - 🔌 Agent API: `http://0.0.0.0:8080`
-   - 📚 API docs: `http://0.0.0.0:8080/redoc`
-   - 📊 Langfuse dashboard: `http://localhost:3000`
-   - 📦 MinIO console: `http://localhost:9091`
-   - 🤖 LiteLLM API: `http://localhost:4000`
-   - 📈 ClickHouse HTTP: `http://localhost:8123`
+   - 📚 API docs: `http://0.0.0.0:8080/docs`
+   - 📊 Langfuse dashboard: `http://0.0.0.0:3000`
+   - 🤖 LiteLLM API: `http://0.0.0.0:4000`
 
 4. Stop services:
+
    ```sh
    docker compose down
    ```
@@ -337,6 +345,8 @@ The project works with
 ## 📚 Useful Resources
 
 - [LangGraph documentation](https://langchain-ai.github.io/langgraph/concepts/low_level/#multiple-schemas)
+- [LangGraph Memory Concept](https://langchain-ai.github.io/langgraph/concepts/memory/)
+- [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/)
 
 ## 👥 Development and Contributing

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

@@ -16,11 +16,6 @@ model_list:
     model_info:
       supported_environments: ["development", "production", "staging"]
 
-litellm_settings:
-  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.
-
 credential_list:
   - credential_name: default_azure_credential
     credential_values:
@@ -30,6 +25,12 @@ credential_list:
     credential_info:
       description: "Production credentials for EU region"
 
+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.
+
 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`
@@ -41,3 +42,6 @@ router_settings: # router_settings are optional
 
 general_settings:
   master_key: os.environ/LITELLM_MASTER_KEY
+  database_connection_pool_limit: 1000
+  store_model_in_db: True
+  store_prompts_in_spend_logs: True

{langgraph_agent_toolkit-0.1.0 → langgraph_agent_toolkit-0.1.2}/langgraph_agent_toolkit/agents/agent.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass
 
 from langgraph.func import Pregel
-from langgraph.graph.state import CompiledStateGraph
 
 from langgraph_agent_toolkit.core.observability.base import BaseObservabilityPlatform
 
@@ -10,5 +9,5 @@ from langgraph_agent_toolkit.core.observability.base import BaseObservabilityPla
 class Agent:
     name: str
     description: str
-    graph: CompiledStateGraph | Pregel
+    graph: Pregel
     observability: BaseObservabilityPlatform | None = None