fix: address final reviewer feedback

- Add error handling for agents_update CLI command
- Remove unnecessary SessionInfo ImportError fallback for consistent dict shapes
- Fix ComputeConfig parameter passing in _install_via_compute

Co-authored-by: Mervin Praison <MervinPraison@users.noreply.github.com>

Author: praisonai-triage-agent[bot]

src/praisonai/praisonai/cli/commands/managed.py

@@ -419,8 +419,12 @@ def agents_update(
     if not kwargs:
         typer.echo("Nothing to update. Pass --name, --system, or --model.")
         raise typer.Exit(0)
-    updated = client.beta.agents.update(agent_id, **kwargs)
-    typer.echo(f"Updated agent: {updated.id} (v{getattr(updated,'version','')})")
+    try:
+        updated = client.beta.agents.update(agent_id, **kwargs)
+        typer.echo(f"Updated agent: {updated.id} (v{getattr(updated,'version','')})")
+    except Exception as e:
+        typer.echo(f"Error updating agent {agent_id}: {e}", err=True)
+        raise typer.Exit(1)
 
 
 @agents_app.command("delete")

src/praisonai/praisonai/integrations/managed_agents.py

@@ -581,23 +581,13 @@ def retrieve_session(self) -> Dict[str, Any]:
             }
 
         # Use unified SessionInfo schema for consistency with Local backend
-        try:
-            from praisonaiagents.managed import SessionInfo
-            session_info = SessionInfo(
-                id=getattr(sess, "id", self._session_id),
-                status=getattr(sess, "status", None),
-                usage=usage_dict
-            )
-            return session_info.to_dict()
-        except ImportError:
-            # Fallback to old format if SessionInfo not available
-            result: Dict[str, Any] = {
-                "id": getattr(sess, "id", self._session_id),
-                "status": getattr(sess, "status", None),
-            }
-            if usage_dict:
-                result["usage"] = usage_dict
-            return result
+        from praisonaiagents.managed import SessionInfo
+        session_info = SessionInfo(
+            id=getattr(sess, "id", self._session_id),
+            status=getattr(sess, "status", None),
+            usage=usage_dict
+        )
+        return session_info.to_dict()
 
     # ------------------------------------------------------------------
     # list_sessions — ManagedBackendProtocol

src/praisonai/praisonai/integrations/managed_local.py

@@ -536,13 +536,11 @@ async def _install_via_compute(self, pip_pkgs: List[str]) -> None:
         """Install packages via compute provider."""
         if not self._compute_instance_id:
             # Provision compute instance if needed
-            from praisonaiagents.managed.protocols import ComputeConfig
-            config = ComputeConfig(
+            await self.provision_compute(
                 packages={"pip": pip_pkgs},
                 image="python:3.12-slim",
                 working_dir=self._cfg.get("working_dir", "/workspace")
             )
-            await self.provision_compute(config=config)
         else:
             # Install on existing instance
             import shlex