agno 2.4.7__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/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)