PraisonAI 4.6.34__tar.gz → 4.6.35__tar.gz

{praisonai-4.6.34 → praisonai-4.6.35}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.34
+Version: 4.6.35
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,7 +11,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.34
+Requires-Dist: praisonaiagents>=1.6.35
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<=1.82.6,>=1.81.0
 Requires-Dist: PyYAML>=6.0

{praisonai-4.6.34 → praisonai-4.6.35}/PraisonAI.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: PraisonAI
-Version: 4.6.34
+Version: 4.6.35
 Summary: PraisonAI is an AI Agents Framework with Self Reflection. PraisonAI application combines PraisonAI Agents, AutoGen, and CrewAI into a low-code solution for building and managing multi-agent LLM systems, focusing on simplicity, customisation, and efficient human-agent collaboration.
 Author: Mervin Praison
 License: MIT
@@ -11,7 +11,7 @@ Description-Content-Type: text/markdown
 Requires-Dist: rich>=13.7
 Requires-Dist: markdown>=3.5
 Requires-Dist: pyparsing>=3.0.0
-Requires-Dist: praisonaiagents>=1.6.34
+Requires-Dist: praisonaiagents>=1.6.35
 Requires-Dist: python-dotenv>=0.19.0
 Requires-Dist: litellm<=1.82.6,>=1.81.0
 Requires-Dist: PyYAML>=6.0

{praisonai-4.6.34 → praisonai-4.6.35}/PraisonAI.egg-info/SOURCES.txt

@@ -584,6 +584,7 @@ praisonai/persistence/__init__.py
 praisonai/persistence/config.py
 praisonai/persistence/factory.py
 praisonai/persistence/orchestrator.py
+praisonai/persistence/registry.py
 praisonai/persistence/conversation/__init__.py
 praisonai/persistence/conversation/async_mysql.py
 praisonai/persistence/conversation/async_postgres.py

{praisonai-4.6.34 → praisonai-4.6.35}/PraisonAI.egg-info/requires.txt

@@ -1,7 +1,7 @@
 rich>=13.7
 markdown>=3.5
 pyparsing>=3.0.0
-praisonaiagents>=1.6.34
+praisonaiagents>=1.6.35
 python-dotenv>=0.19.0
 litellm<=1.82.6,>=1.81.0
 PyYAML>=6.0

{praisonai-4.6.34 → praisonai-4.6.35}/praisonai/deploy.py

@@ -57,7 +57,7 @@ class CloudDeployer:
             file.write("FROM python:3.11-slim\n")
             file.write("WORKDIR /app\n")
             file.write("COPY . .\n")
-            file.write("RUN pip install flask praisonai==4.6.34 gunicorn markdown\n")
+            file.write("RUN pip install flask praisonai==4.6.35 gunicorn markdown\n")
             file.write("EXPOSE 8080\n")
             file.write('CMD ["gunicorn", "-b", "0.0.0.0:8080", "api:app"]\n')
             

{praisonai-4.6.34 → praisonai-4.6.35}/praisonai/llm/registry.py

@@ -21,6 +21,7 @@ Example:
 """
 
 from typing import Any, Callable, Dict, List, Optional, Type, Union
+import threading
 
 
 # Type aliases
@@ -35,12 +36,33 @@ class LLMProviderRegistry:
     
     Manages registration and resolution of LLM providers by name.
     Supports lazy loading, aliases, and isolated instances.
+    Thread-safe for concurrent operations.
     """
     
+    _instance: Optional["LLMProviderRegistry"] = None
+    _instance_lock = threading.Lock()
+    
     def __init__(self):
         """Initialize an empty registry."""
         self._providers: Dict[str, ProviderType] = {}
         self._aliases: Dict[str, str] = {}  # alias -> canonical name
+        self._lock = threading.RLock()  # RLock for re-entrant calls
+        # Register built-in providers during initialization
+        _register_builtin_providers(self)
+    
+    @classmethod
+    def get_instance(cls) -> "LLMProviderRegistry":
+        """
+        Get the singleton registry instance.
+        
+        Returns:
+            LLMProviderRegistry: The singleton registry instance
+        """
+        if cls._instance is None:
+            with cls._instance_lock:
+                if cls._instance is None:
+                    cls._instance = cls()
+        return cls._instance
     
     def register(
         self,
@@ -62,35 +84,48 @@ class LLMProviderRegistry:
         Raises:
             ValueError: If name is already registered (unless override=True)
         """
-        normalized_name = name.lower()
-        
-        # Check for existing registration
-        if normalized_name in self._providers and not override:
-            raise ValueError(
-                f"Provider '{name}' is already registered. "
-                f"Use override=True to replace it."
-            )
-        
-        self._providers[normalized_name] = provider
-        
-        # Register aliases
-        if aliases:
-            for alias in aliases:
-                normalized_alias = alias.lower()
-                # Check collision with existing provider name
-                if normalized_alias in self._providers and not override:
-                    raise ValueError(
-                        f"Alias '{alias}' conflicts with existing provider name. "
-                        f"Use override=True to replace it."
-                    )
-                # Check collision with existing alias
-                if normalized_alias in self._aliases and not override:
-                    existing_target = self._aliases[normalized_alias]
-                    raise ValueError(
-                        f"Alias '{alias}' is already registered (points to '{existing_target}'). "
-                        f"Use override=True to replace it."
-                    )
-                self._aliases[normalized_alias] = normalized_name
+        with self._lock:
+            normalized_name = name.lower()
+            
+            # Check for existing registration
+            if normalized_name in self._providers and not override:
+                raise ValueError(
+                    f"Provider '{name}' is already registered. "
+                    f"Use override=True to replace it."
+                )
+            
+            # Check if this canonical name conflicts with existing alias
+            if normalized_name in self._aliases and not override:
+                existing_target = self._aliases[normalized_name] 
+                raise ValueError(
+                    f"Cannot register provider '{name}' - it conflicts with existing alias "
+                    f"(currently points to '{existing_target}'). Use override=True to replace it."
+                )
+            
+            # If override=True, clean up any existing alias before registering canonical name
+            if override and normalized_name in self._aliases:
+                del self._aliases[normalized_name]
+            
+            self._providers[normalized_name] = provider
+            
+            # Register aliases
+            if aliases:
+                for alias in aliases:
+                    normalized_alias = alias.lower()
+                    # Check collision with existing provider name
+                    if normalized_alias in self._providers and not override:
+                        raise ValueError(
+                            f"Alias '{alias}' conflicts with existing provider name. "
+                            f"Use override=True to replace it."
+                        )
+                    # Check collision with existing alias
+                    if normalized_alias in self._aliases and not override:
+                        existing_target = self._aliases[normalized_alias]
+                        raise ValueError(
+                            f"Alias '{alias}' is already registered (points to '{existing_target}'). "
+                            f"Use override=True to replace it."
+                        )
+                    self._aliases[normalized_alias] = normalized_name
     
     def unregister(self, name: str) -> bool:
         """
@@ -102,27 +137,28 @@ class LLMProviderRegistry:
         Returns:
             True if provider was unregistered, False if not found
         """
-        normalized_name = name.lower()
-        
-        # Check if it's an alias
-        if normalized_name in self._aliases:
-            del self._aliases[normalized_name]
-            return True
-        
-        # Check if it's a canonical name
-        if normalized_name in self._providers:
-            # Remove all aliases pointing to this provider
-            aliases_to_remove = [
-                alias for alias, canonical in self._aliases.items()
-                if canonical == normalized_name
-            ]
-            for alias in aliases_to_remove:
-                del self._aliases[alias]
+        with self._lock:
+            normalized_name = name.lower()
             
-            del self._providers[normalized_name]
-            return True
-        
-        return False
+            # Check if it's an alias
+            if normalized_name in self._aliases:
+                del self._aliases[normalized_name]
+                return True
+            
+            # Check if it's a canonical name
+            if normalized_name in self._providers:
+                # Remove all aliases pointing to this provider
+                aliases_to_remove = [
+                    alias for alias, canonical in self._aliases.items()
+                    if canonical == normalized_name
+                ]
+                for alias in aliases_to_remove:
+                    del self._aliases[alias]
+                
+                del self._providers[normalized_name]
+                return True
+            
+            return False
     
     def has(self, name: str) -> bool:
         """
@@ -134,8 +170,9 @@ class LLMProviderRegistry:
         Returns:
             True if provider is registered
         """
-        normalized_name = name.lower()
-        return normalized_name in self._providers or normalized_name in self._aliases
+        with self._lock:
+            normalized_name = name.lower()
+            return normalized_name in self._providers or normalized_name in self._aliases
     
     def list(self) -> List[str]:
         """
@@ -144,7 +181,8 @@ class LLMProviderRegistry:
         Returns:
             List of provider names
         """
-        return list(self._providers.keys())
+        with self._lock:
+            return list(self._providers.keys())
     
     def list_all(self) -> List[str]:
         """
@@ -153,7 +191,8 @@ class LLMProviderRegistry:
         Returns:
             List of all registered names and aliases
         """
-        return list(self._providers.keys()) + list(self._aliases.keys())
+        with self._lock:
+            return list(self._providers.keys()) + list(self._aliases.keys())
     
     def resolve(
         self,
@@ -175,21 +214,22 @@ class LLMProviderRegistry:
         Raises:
             ValueError: If provider not found
         """
-        normalized_name = name.lower()
-        
-        # Resolve alias to canonical name
-        canonical_name = self._aliases.get(normalized_name, normalized_name)
-        
-        provider = self._providers.get(canonical_name)
-        if provider is None:
-            available = self.list()
-            raise ValueError(
-                f"Unknown provider: '{name}'. "
-                f"Available providers: {', '.join(available) if available else 'none'}. "
-                f"Register a custom provider with register_llm_provider('{name}', YourProviderClass)."
-            )
+        with self._lock:
+            normalized_name = name.lower()
+            
+            # Resolve alias to canonical name
+            canonical_name = self._aliases.get(normalized_name, normalized_name)
+            
+            provider = self._providers.get(canonical_name)
+            if provider is None:
+                available = list(self._providers.keys())  # Don't call self.list() to avoid double-locking
+                raise ValueError(
+                    f"Unknown provider: '{name}'. "
+                    f"Available providers: {', '.join(available) if available else 'none'}. "
+                    f"Register a custom provider with register_llm_provider('{name}', YourProviderClass)."
+                )
         
-        # Create instance
+        # Create instance outside the lock
         return provider(model_id, config)
     
     def get(self, name: str) -> Optional[ProviderType]:
@@ -202,30 +242,24 @@ class LLMProviderRegistry:
         Returns:
             Provider class/factory or None
         """
-        normalized_name = name.lower()
-        canonical_name = self._aliases.get(normalized_name, normalized_name)
-        return self._providers.get(canonical_name)
+        with self._lock:
+            normalized_name = name.lower()
+            canonical_name = self._aliases.get(normalized_name, normalized_name)
+            return self._providers.get(canonical_name)
 
 
 # ============================================================================
 # Default Registry Singleton
 # ============================================================================
 
-_default_registry: Optional[LLMProviderRegistry] = None
-
-
 def get_default_llm_registry() -> LLMProviderRegistry:
     """
     Get the default global LLM provider registry.
     
     This is the registry used by create_llm_provider() when no custom registry
-    is specified.
+    is specified. Uses the thread-safe singleton pattern.
     """
-    global _default_registry
-    if _default_registry is None:
-        _default_registry = LLMProviderRegistry()
-        _register_builtin_providers(_default_registry)
-    return _default_registry
+    return LLMProviderRegistry.get_instance()
 
 
 def _register_builtin_providers(registry: LLMProviderRegistry) -> None:
@@ -378,5 +412,5 @@ def create_llm_provider(
 
 def _reset_default_registry() -> None:
     """Reset the default registry (mainly for testing)."""
-    global _default_registry
-    _default_registry = None
+    with LLMProviderRegistry._instance_lock:
+        LLMProviderRegistry._instance = None

{praisonai-4.6.34 → praisonai-4.6.35}/praisonai/persistence/conversation/async_mysql.py

@@ -41,7 +41,7 @@ class AsyncMySQLConversationStore(ConversationStore):
         database: str = "praisonai",
         user: str = "root",
         password: str = "",
-        table_prefix: str = "praisonai_",
+        table_prefix: str = "praison_",
         pool_size: int = 10,
     ):
         """
@@ -122,6 +122,7 @@ class AsyncMySQLConversationStore(ConversationStore):
                         user_id VARCHAR(255),
                         agent_id VARCHAR(255),
                         name VARCHAR(255),
+                        state JSON,
                         metadata JSON,
                         created_at DOUBLE,
                         updated_at DOUBLE,
@@ -136,6 +137,8 @@ class AsyncMySQLConversationStore(ConversationStore):
                         session_id VARCHAR(255),
                         role VARCHAR(50),
                         content TEXT,
+                        tool_calls JSON,
+                        tool_call_id VARCHAR(255),
                         metadata JSON,
                         created_at DOUBLE,
                         INDEX idx_session_id (session_id),
@@ -152,9 +155,10 @@ class AsyncMySQLConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             async with conn.cursor() as cur:
                 await cur.execute(f"""
-                    INSERT INTO {table} (session_id, user_id, agent_id, name, metadata, created_at, updated_at)
-                    VALUES (%s, %s, %s, %s, %s, %s, %s)
+                    INSERT INTO {table} (session_id, user_id, agent_id, name, state, metadata, created_at, updated_at)
+                    VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
                 """, (session.session_id, session.user_id, session.agent_id, session.name,
+                      json.dumps(session.state) if session.state else None,
                       json.dumps(session.metadata) if session.metadata else None,
                       session.created_at, session.updated_at))
         
@@ -173,7 +177,7 @@ class AsyncMySQLConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             async with conn.cursor() as cur:
                 await cur.execute(f"""
-                    SELECT session_id, user_id, agent_id, name, metadata, created_at, updated_at
+                    SELECT session_id, user_id, agent_id, name, state, metadata, created_at, updated_at
                     FROM {table} WHERE session_id = %s
                 """, (session_id,))
                 row = await cur.fetchone()
@@ -184,9 +188,10 @@ class AsyncMySQLConversationStore(ConversationStore):
                 user_id=row[1],
                 agent_id=row[2],
                 name=row[3],
-                metadata=json.loads(row[4]) if row[4] else None,
-                created_at=row[5],
-                updated_at=row[6]
+                state=json.loads(row[4]) if row[4] else None,
+                metadata=json.loads(row[5]) if row[5] else None,
+                created_at=row[6],
+                updated_at=row[7]
             )
         return None
     
@@ -206,9 +211,11 @@ class AsyncMySQLConversationStore(ConversationStore):
             async with conn.cursor() as cur:
                 await cur.execute(f"""
                     UPDATE {table} 
-                    SET name = %s, metadata = %s, updated_at = %s
+                    SET name = %s, state = %s, metadata = %s, updated_at = %s
                     WHERE session_id = %s
-                """, (session.name, json.dumps(session.metadata) if session.metadata else None,
+                """, (session.name,
+                      json.dumps(session.state) if session.state else None,
+                      json.dumps(session.metadata) if session.metadata else None,
                       session.updated_at, session.session_id))
         
         return session
@@ -263,7 +270,7 @@ class AsyncMySQLConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             async with conn.cursor() as cur:
                 await cur.execute(f"""
-                    SELECT session_id, user_id, agent_id, name, metadata, created_at, updated_at
+                    SELECT session_id, user_id, agent_id, name, state, metadata, created_at, updated_at
                     FROM {table} {where_clause}
                     ORDER BY updated_at DESC
                     LIMIT %s OFFSET %s
@@ -276,9 +283,10 @@ class AsyncMySQLConversationStore(ConversationStore):
                 user_id=row[1],
                 agent_id=row[2],
                 name=row[3],
-                metadata=json.loads(row[4]) if row[4] else None,
-                created_at=row[5],
-                updated_at=row[6]
+                state=json.loads(row[4]) if row[4] else None,
+                metadata=json.loads(row[5]) if row[5] else None,
+                created_at=row[6],
+                updated_at=row[7]
             )
             for row in rows
         ]
@@ -304,9 +312,11 @@ class AsyncMySQLConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             async with conn.cursor() as cur:
                 await cur.execute(f"""
-                    INSERT INTO {table} (id, session_id, role, content, metadata, created_at)
-                    VALUES (%s, %s, %s, %s, %s, %s)
+                    INSERT INTO {table} (id, session_id, role, content, tool_calls, tool_call_id, metadata, created_at)
+                    VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
                 """, (message.id, session_id, message.role, message.content,
+                      json.dumps(message.tool_calls) if message.tool_calls else None,
+                      message.tool_call_id,
                       json.dumps(message.metadata) if message.metadata else None,
                       message.created_at))
         
@@ -347,7 +357,7 @@ class AsyncMySQLConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             async with conn.cursor() as cur:
                 await cur.execute(f"""
-                    SELECT id, session_id, role, content, metadata, created_at
+                    SELECT id, session_id, role, content, tool_calls, tool_call_id, metadata, created_at
                     FROM {table} {where_clause}
                     ORDER BY created_at ASC {limit_clause}
                 """, params)
@@ -359,8 +369,10 @@ class AsyncMySQLConversationStore(ConversationStore):
                 session_id=row[1],
                 role=row[2],
                 content=row[3],
-                metadata=json.loads(row[4]) if row[4] else None,
-                created_at=row[5]
+                tool_calls=json.loads(row[4]) if row[4] else None,
+                tool_call_id=row[5],
+                metadata=json.loads(row[6]) if row[6] else None,
+                created_at=row[7]
             )
             for row in rows
         ]

{praisonai-4.6.34 → praisonai-4.6.35}/praisonai/persistence/conversation/async_postgres.py

@@ -39,7 +39,7 @@ class AsyncPostgresConversationStore(ConversationStore):
         database: str = "praisonai",
         user: str = "postgres",
         password: str = "",
-        table_prefix: str = "praisonai_",
+        table_prefix: str = "praison_",
         pool_size: int = 10,
     ):
         """
@@ -116,6 +116,7 @@ class AsyncPostgresConversationStore(ConversationStore):
                     user_id VARCHAR(255),
                     agent_id VARCHAR(255),
                     name VARCHAR(255),
+                    state JSONB,
                     metadata JSONB,
                     created_at DOUBLE PRECISION,
                     updated_at DOUBLE PRECISION
@@ -128,6 +129,8 @@ class AsyncPostgresConversationStore(ConversationStore):
                     session_id VARCHAR(255) REFERENCES {sessions_table}(session_id) ON DELETE CASCADE,
                     role VARCHAR(50),
                     content TEXT,
+                    tool_calls JSONB,
+                    tool_call_id VARCHAR(255),
                     metadata JSONB,
                     created_at DOUBLE PRECISION
                 )
@@ -146,9 +149,10 @@ class AsyncPostgresConversationStore(ConversationStore):
         table = f"{self.table_prefix}sessions"
         async with self._pool.acquire() as conn:
             await conn.execute(f"""
-                INSERT INTO {table} (session_id, user_id, agent_id, name, metadata, created_at, updated_at)
-                VALUES ($1, $2, $3, $4, $5, $6, $7)
+                INSERT INTO {table} (session_id, user_id, agent_id, name, state, metadata, created_at, updated_at)
+                VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
             """, session.session_id, session.user_id, session.agent_id, session.name,
+                json.dumps(session.state) if session.state else None,
                 json.dumps(session.metadata) if session.metadata else None,
                 session.created_at, session.updated_at)
         
@@ -175,6 +179,7 @@ class AsyncPostgresConversationStore(ConversationStore):
                 user_id=row['user_id'],
                 agent_id=row['agent_id'],
                 name=row['name'],
+                state=json.loads(row['state']) if row['state'] else None,
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at'],
                 updated_at=row['updated_at']
@@ -196,9 +201,10 @@ class AsyncPostgresConversationStore(ConversationStore):
         async with self._pool.acquire() as conn:
             await conn.execute(f"""
                 UPDATE {table} 
-                SET name = $2, metadata = $3, updated_at = $4
+                SET name = $2, state = $3, metadata = $4, updated_at = $5
                 WHERE session_id = $1
             """, session.session_id, session.name,
+                json.dumps(session.state) if session.state else None,
                 json.dumps(session.metadata) if session.metadata else None,
                 session.updated_at)
         
@@ -267,6 +273,7 @@ class AsyncPostgresConversationStore(ConversationStore):
                 user_id=row['user_id'],
                 agent_id=row['agent_id'],
                 name=row['name'],
+                state=json.loads(row['state']) if row['state'] else None,
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at'],
                 updated_at=row['updated_at']
@@ -294,9 +301,11 @@ class AsyncPostgresConversationStore(ConversationStore):
         
         async with self._pool.acquire() as conn:
             await conn.execute(f"""
-                INSERT INTO {table} (id, session_id, role, content, metadata, created_at)
-                VALUES ($1, $2, $3, $4, $5, $6)
+                INSERT INTO {table} (id, session_id, role, content, tool_calls, tool_call_id, metadata, created_at)
+                VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
             """, message.id, session_id, message.role, message.content,
+                json.dumps(message.tool_calls) if message.tool_calls else None,
+                message.tool_call_id,
                 json.dumps(message.metadata) if message.metadata else None,
                 message.created_at)
         
@@ -349,6 +358,8 @@ class AsyncPostgresConversationStore(ConversationStore):
                 session_id=row['session_id'],
                 role=row['role'],
                 content=row['content'],
+                tool_calls=json.loads(row['tool_calls']) if row['tool_calls'] else None,
+                tool_call_id=row['tool_call_id'],
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at']
             )

{praisonai-4.6.34 → praisonai-4.6.35}/praisonai/persistence/conversation/async_sqlite.py

@@ -31,7 +31,7 @@ class AsyncSQLiteConversationStore(ConversationStore):
     def __init__(
         self,
         path: str = "praisonai_conversations.db",
-        table_prefix: str = "praisonai_",
+        table_prefix: str = "praison_",
     ):
         """
         Initialize async SQLite store.
@@ -76,6 +76,7 @@ class AsyncSQLiteConversationStore(ConversationStore):
                 user_id TEXT,
                 agent_id TEXT,
                 name TEXT,
+                state TEXT,
                 metadata TEXT,
                 created_at REAL,
                 updated_at REAL
@@ -88,6 +89,8 @@ class AsyncSQLiteConversationStore(ConversationStore):
                 session_id TEXT REFERENCES {sessions_table}(session_id) ON DELETE CASCADE,
                 role TEXT,
                 content TEXT,
+                tool_calls TEXT,
+                tool_call_id TEXT,
                 metadata TEXT,
                 created_at REAL
             )
@@ -107,9 +110,10 @@ class AsyncSQLiteConversationStore(ConversationStore):
         
         table = f"{self.table_prefix}sessions"
         await self._conn.execute(f"""
-            INSERT INTO {table} (session_id, user_id, agent_id, name, metadata, created_at, updated_at)
-            VALUES (?, ?, ?, ?, ?, ?, ?)
+            INSERT INTO {table} (session_id, user_id, agent_id, name, state, metadata, created_at, updated_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
         """, (session.session_id, session.user_id, session.agent_id, session.name,
+              json.dumps(session.state) if session.state else None,
               json.dumps(session.metadata) if session.metadata else None,
               session.created_at, session.updated_at))
         await self._conn.commit()
@@ -139,6 +143,7 @@ class AsyncSQLiteConversationStore(ConversationStore):
                 user_id=row['user_id'],
                 agent_id=row['agent_id'],
                 name=row['name'],
+                state=json.loads(row['state']) if row['state'] else None,
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at'],
                 updated_at=row['updated_at']
@@ -161,9 +166,11 @@ class AsyncSQLiteConversationStore(ConversationStore):
         
         await self._conn.execute(f"""
             UPDATE {table} 
-            SET name = ?, metadata = ?, updated_at = ?
+            SET name = ?, state = ?, metadata = ?, updated_at = ?
             WHERE session_id = ?
-        """, (session.name, json.dumps(session.metadata) if session.metadata else None,
+        """, (session.name,
+              json.dumps(session.state) if session.state else None,
+              json.dumps(session.metadata) if session.metadata else None,
               session.updated_at, session.session_id))
         await self._conn.commit()
         
@@ -233,6 +240,7 @@ class AsyncSQLiteConversationStore(ConversationStore):
                 user_id=row['user_id'],
                 agent_id=row['agent_id'],
                 name=row['name'],
+                state=json.loads(row['state']) if row['state'] else None,
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at'],
                 updated_at=row['updated_at']
@@ -261,9 +269,11 @@ class AsyncSQLiteConversationStore(ConversationStore):
         table = f"{self.table_prefix}messages"
         
         await self._conn.execute(f"""
-            INSERT INTO {table} (id, session_id, role, content, metadata, created_at)
-            VALUES (?, ?, ?, ?, ?, ?)
+            INSERT INTO {table} (id, session_id, role, content, tool_calls, tool_call_id, metadata, created_at)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
         """, (message.id, session_id, message.role, message.content,
+              json.dumps(message.tool_calls) if message.tool_calls else None,
+              message.tool_call_id,
               json.dumps(message.metadata) if message.metadata else None,
               message.created_at))
         await self._conn.commit()
@@ -316,6 +326,8 @@ class AsyncSQLiteConversationStore(ConversationStore):
                 session_id=row['session_id'],
                 role=row['role'],
                 content=row['content'],
+                tool_calls=json.loads(row['tool_calls']) if row['tool_calls'] else None,
+                tool_call_id=row['tool_call_id'],
                 metadata=json.loads(row['metadata']) if row['metadata'] else None,
                 created_at=row['created_at']
             )