intentkit 0.6.7.dev8__tar.gz → 0.6.7.dev9__tar.gz

{intentkit-0.6.7.dev8 → intentkit-0.6.7.dev9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: intentkit
-Version: 0.6.7.dev8
+Version: 0.6.7.dev9
 Summary: Intent-based AI Agent Platform - Core Package
 Project-URL: Homepage, https://github.com/crestal-network/intentkit
 Project-URL: Repository, https://github.com/crestal-network/intentkit

{intentkit-0.6.7.dev8 → intentkit-0.6.7.dev9}/__init__.py

@@ -3,7 +3,7 @@
 A powerful platform for building AI agents with blockchain and cryptocurrency capabilities.
 """
 
-__version__ = "0.6.7-dev8"
+__version__ = "0.6.7-dev9"
 __author__ = "hyacinthus"
 __email__ = "hyacinthus@gmail.com"
 

{intentkit-0.6.7.dev8 → intentkit-0.6.7.dev9}/config/config.py

@@ -59,6 +59,7 @@ class Config:
             }
         # ==== this part can be load from env or aws secrets manager
         self.db["auto_migrate"] = self.load("DB_AUTO_MIGRATE", "true") == "true"
+        self.db["pool_size"] = int(self.load("DB_POOL_SIZE", "3"))
         self.debug = self.load("DEBUG") == "true"
         self.debug_checkpoint = (
             self.load("DEBUG_CHECKPOINT", "false") == "true"

{intentkit-0.6.7.dev8 → intentkit-0.6.7.dev9}/models/db.py

@@ -23,6 +23,9 @@ async def init_db(
     auto_migrate: Annotated[
         bool, Field(default=True, description="Whether to run migrations automatically")
     ],
+    pool_size: Annotated[
+        int, Field(default=3, description="Database connection pool size")
+    ] = 3,
 ) -> None:
     """Initialize the database and handle schema updates.
 
@@ -33,6 +36,7 @@ async def init_db(
         dbname: Database name
         port: Database port (default: 5432)
         auto_migrate: Whether to run migrations automatically (default: True)
+        pool_size: Database connection pool size (default: 3)
     """
     global engine, _langgraph_checkpointer
     # Initialize psycopg pool and AsyncPostgresSaver if not already initialized
@@ -43,8 +47,8 @@ async def init_db(
             )
             pool = AsyncConnectionPool(
                 conninfo=conn_string,
-                min_size=3,
-                max_size=20,
+                min_size=pool_size,
+                max_size=pool_size * 2,
                 timeout=60,
                 max_idle=30 * 60,
             )
@@ -60,8 +64,8 @@ async def init_db(
         if host:
             engine = create_async_engine(
                 f"postgresql+asyncpg://{username}:{quote_plus(password)}@{host}:{port}/{dbname}",
-                pool_size=20,  # Increase pool size
-                max_overflow=30,  # Increase max overflow
+                pool_size=pool_size,
+                max_overflow=pool_size * 2,  # Set overflow to 2x pool size
                 pool_timeout=60,  # Increase timeout
                 pool_pre_ping=True,  # Enable connection health checks
                 pool_recycle=3600,  # Recycle connections after 1 hour

{intentkit-0.6.7.dev8 → intentkit-0.6.7.dev9}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "intentkit"
-version = "0.6.7-dev8"
+version = "0.6.7-dev9"
 description = "Intent-based AI Agent Platform - Core Package"
 authors = [{ name = "hyacinthus", email = "hyacinthus@gmail.com" }]
 requires-python = ">=3.12"