agno 2.4.6__py3-none-any.whl → 2.4.8__py3-none-any.whl

agno/agent/agent.py

@@ -11493,7 +11493,7 @@ class Agent:
 
         return get_tool_call_history
 
-    def _update_session_state_tool(self, session_state, session_state_updates: dict) -> str:
+    def _update_session_state_tool(self, run_context: RunContext, session_state_updates: dict) -> str:
         """
         Update the shared session state.  Provide any updates as a dictionary of key-value pairs.
         Example:
@@ -11502,6 +11502,10 @@ class Agent:
         Args:
             session_state_updates (dict): The updates to apply to the shared session state. Should be a dictionary of key-value pairs.
         """
+
+        if run_context.session_state is None:
+            run_context.session_state = {}
+        session_state = run_context.session_state
         for key, value in session_state_updates.items():
             session_state[key] = value
 

agno/db/base.py

@@ -576,6 +576,7 @@ class BaseDb(ABC):
         component_type: Optional[ComponentType] = None,
         name: Optional[str] = None,
         description: Optional[str] = None,
+        current_version: Optional[int] = None,
         metadata: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         """Create or update a component.
@@ -585,6 +586,7 @@ class BaseDb(ABC):
             component_type: Type (agent|team|workflow). Required for create, optional for update.
             name: Display name.
             description: Optional description.
+            current_version: Optional current version.
             metadata: Optional metadata dict.
 
         Returns:

agno/db/postgres/postgres.py

@@ -3241,6 +3241,7 @@ class PostgresDb(BaseDb):
         component_type: Optional[ComponentType] = None,
         name: Optional[str] = None,
         description: Optional[str] = None,
+        current_version: Optional[int] = None,
         metadata: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         """Create or update a component.
@@ -3250,6 +3251,7 @@ class PostgresDb(BaseDb):
             component_type: Type (agent|team|workflow). Required for create, optional for update.
             name: Display name.
             description: Optional description.
+            current_version: Optional current version.
             metadata: Optional metadata dict.
 
         Returns:
@@ -3317,6 +3319,8 @@ class PostgresDb(BaseDb):
                         updates["name"] = name
                     if description is not None:
                         updates["description"] = description
+                    if current_version is not None:
+                        updates["current_version"] = current_version
                     if metadata is not None:
                         updates["metadata"] = metadata
 
@@ -3693,8 +3697,8 @@ class PostgresDb(BaseDb):
             raise ValueError(f"Invalid stage: {stage}")
 
         try:
-            configs_table = self._get_table(table_type="component_configs", create_table_if_not_found=True)
             components_table = self._get_table(table_type="components")
+            configs_table = self._get_table(table_type="component_configs", create_table_if_not_found=True)
             links_table = self._get_table(table_type="component_links", create_table_if_not_found=True)
 
             if components_table is None:
@@ -3881,10 +3885,6 @@ class PostgresDb(BaseDb):
                 if config_row is None:
                     return False
 
-                # Cannot delete published configs
-                if config_row == "published":
-                    raise ValueError(f"Cannot delete published config {component_id} v{version}")
-
                 # Check if it's current version
                 current = sess.execute(
                     select(components_table.c.current_version).where(components_table.c.component_id == component_id)

agno/db/singlestore/singlestore.py

@@ -173,8 +173,7 @@ class SingleStoreDb(BaseDb):
                     column_kwargs["unique"] = True
                 columns.append(Column(*column_args, **column_kwargs))
 
-            # Create the table object without constraints to avoid autoload issues
-            table = Table(table_name, self.metadata, *columns, schema=self.db_schema)
+            table = Table(table_name, self.metadata, *columns, schema=self.db_schema, extend_existing=True)
 
             return table
 
@@ -240,8 +239,7 @@ class SingleStoreDb(BaseDb):
 
                 columns.append(Column(*column_args, **column_kwargs))
 
-            # Create the table object
-            table = Table(table_name, self.metadata, *columns, schema=self.db_schema)
+            table = Table(table_name, self.metadata, *columns, schema=self.db_schema, extend_existing=True)
 
             # Add multi-column unique constraints with table-specific names
             for constraint in schema_unique_constraints:
@@ -396,7 +394,8 @@ class SingleStoreDb(BaseDb):
 
         if table_type == "spans":
             # Ensure traces table exists first (for foreign key)
-            self._get_table(table_type="traces", create_table_if_not_found=create_table_if_not_found)
+            if create_table_if_not_found:
+                self._get_table(table_type="traces", create_table_if_not_found=True)
             self.spans_table = self._get_or_create_table(
                 table_name=self.span_table_name,
                 table_type="spans",

agno/db/sqlite/sqlite.py

@@ -3139,6 +3139,7 @@ class SqliteDb(BaseDb):
         component_type: Optional[ComponentType] = None,
         name: Optional[str] = None,
         description: Optional[str] = None,
+        current_version: Optional[int] = None,
         metadata: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         """Create or update a component.
@@ -3148,6 +3149,7 @@ class SqliteDb(BaseDb):
             component_type: Type (agent|team|workflow). Required for create, optional for update.
             name: Display name.
             description: Optional description.
+            current_version: Optional current version.
             metadata: Optional metadata dict.
 
         Returns:
@@ -3213,6 +3215,8 @@ class SqliteDb(BaseDb):
                         updates["name"] = name
                     if description is not None:
                         updates["description"] = description
+                    if current_version is not None:
+                        updates["current_version"] = current_version
                     if metadata is not None:
                         updates["metadata"] = metadata
 
@@ -3759,10 +3763,6 @@ class SqliteDb(BaseDb):
                 if config_row is None:
                     return False
 
-                # Cannot delete published configs
-                if config_row.stage == "published":
-                    raise ValueError(f"Cannot delete published config {component_id} v{version}")
-
                 # Check if it's current version
                 current = sess.execute(
                     select(components_table.c.current_version).where(components_table.c.component_id == component_id)