langgraph-agent-toolkit 0.1.1__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.1 → 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="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
+# 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.1 → 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.1 → langgraph_agent_toolkit-0.2.0}/.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.1 → langgraph_agent_toolkit-0.2.0}/CHANGELOG.md

@@ -6,6 +6,48 @@ 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
+
+- `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.1 → langgraph_agent_toolkit-0.2.0}/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.1 → langgraph_agent_toolkit-0.2.0}/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.1 → langgraph_agent_toolkit-0.2.0}/PKG-INFO

@@ -1,8 +1,9 @@
 Metadata-Version: 2.4
 Name: langgraph-agent-toolkit
-Version: 0.1.1
+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
 Author-email: Roman Kryvokhyzha <kriwohizha@gmail.com>
 License-Expression: MIT
 License-File: LICENSE
@@ -73,10 +74,12 @@ Description-Content-Type: text/markdown
 [![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)
 [![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: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![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/) -->
 
@@ -100,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
@@ -116,7 +119,13 @@ 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
@@ -124,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
 
@@ -174,10 +184,29 @@ Features include:
    **🔑 LLM API Configuration**
 
    ```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
+   # 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**
@@ -334,13 +363,11 @@ 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:
 
@@ -409,6 +436,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.1 → langgraph_agent_toolkit-0.2.0}/README.md

@@ -3,10 +3,12 @@
 [![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)
 [![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: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
+[![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,13 @@ 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
@@ -54,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
 
@@ -104,10 +113,29 @@ Features include:
    **🔑 LLM API Configuration**
 
    ```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
+   # 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**
@@ -264,13 +292,11 @@ 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:
 
@@ -339,6 +365,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.1 → 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,11 +18,7 @@ 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.
-
+# list of credentials to be used by the models
 credential_list:
   - credential_name: default_azure_credential
     credential_values:
@@ -30,6 +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`
@@ -39,5 +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.1 → langgraph_agent_toolkit-0.2.0}/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