added get setup info tool

autogpt_platform/backend/backend/server/v2/chat/tools/get_required_setup_info.py

@@ -0,0 +1,133 @@
+"""Tool for getting required setup information for an agent."""
+
+import logging
+from typing import Any
+
+from backend.integrations.creds_manager import IntegrationCredentialsManager
+from backend.server.v2.chat.tools.base import BaseTool
+from backend.server.v2.chat.tools.get_agent_details import GetAgentDetailsTool
+from backend.server.v2.chat.tools.models import (
+    AgentDetailsResponse,
+    ErrorResponse,
+    SetupInfo,
+    SetupRequirementsResponse,
+    ToolResponseBase,
+    UserReadiness,
+)
+
+logger = logging.getLogger(__name__)
+
+
+class GetRequiredSetupInfoTool(BaseTool):
+    """Tool for getting required setup information including credentials and inputs."""
+
+    @property
+    def name(self) -> str:
+        return "get_required_setup_info"
+
+    @property
+    def description(self) -> str:
+        return """Check if an agent can be set up with the provided input data and credentials.
+        Call this AFTER get_agent_details to validate that you have all required inputs.
+        Pass the input dictionary you plan to use with run_agent or setup_agent to verify it's complete."""
+
+    @property
+    def parameters(self) -> dict[str, Any]:
+        return {
+            "type": "object",
+            "properties": {
+                "agent_id": {
+                    "type": "string",
+                    "description": "The marketplace agent slug (e.g., 'username/agent-name' or just 'agent-name' to search)",
+                },
+                "inputs": {
+                    "type": "object",
+                    "description": "The input dictionary you plan to provide. Should contain ALL required inputs from get_agent_details",
+                    "additionalProperties": True,
+                },
+            },
+            "required": ["agent_id"],
+        }
+
+    @property
+    def requires_auth(self) -> bool:
+        """This tool requires authentication."""
+        return True
+
+    async def _execute(
+        self,
+        user_id: str | None,
+        session_id: str,
+        **kwargs,
+    ) -> ToolResponseBase:
+        """
+        Retrieve and validate the required setup information for running or configuring an agent.
+
+        This checks all required credentials and input fields based on the agent details,
+        and verifies user readiness to run the agent based on provided inputs and available credentials.
+
+        Args:
+            user_id: The authenticated user's ID (must not be None; authentication required).
+            session_id: The chat session ID.
+            agent_id: The agent's marketplace slug (e.g. 'username/agent-name'). Also accepts Graph ID.
+            agent_version: (Optional) Specific agent/graph version (if applicable).
+
+        Returns:
+            SetupRequirementsResponse containing:
+                - agent and graph info,
+                - credential and input requirements,
+                - user readiness and missing credentials/fields,
+                - setup instructions.
+        """
+        ad = await GetAgentDetailsTool().execute(user_id, session_id, **kwargs)
+
+        assert (
+            user_id is not None
+        ), "GetRequiredSetupInfoTool - This should never happen user_id is None when auth is required"
+
+        if ad.result is None:
+            return ErrorResponse(
+                message="Failed to get agent details",
+                session_id=session_id,
+            )
+
+        agent_details = AgentDetailsResponse.model_validate(ad.result)
+
+        if isinstance(agent_details, ErrorResponse):
+            return agent_details
+
+        available_creds = await IntegrationCredentialsManager().store.get_all_creds(
+            user_id
+        )
+        required_credentials = []
+
+        for c in agent_details.agent.credentials:
+            if c.id not in available_creds:
+                required_credentials.append(c)
+
+        required_fields = set(agent_details.agent.inputs.get("required", []))
+        provided_inputs = kwargs.get("inputs", {})
+        missing_inputs = required_fields - set(provided_inputs.keys())
+
+        missing_credentials = {c.id: c.model_dump() for c in required_credentials}
+
+        user_readiness = UserReadiness(
+            has_all_credentials=len(required_credentials) == 0,
+            missing_credentials=missing_credentials,
+            ready_to_run=len(missing_inputs) == 0 and len(required_credentials) == 0,
+        )
+        requirements = {
+            "credentials": agent_details.agent.credentials,
+            "inputs": agent_details.agent.inputs,
+            "execution_modes": agent_details.agent.execution_options.model_dump(),
+        }
+        return SetupRequirementsResponse(
+            message="Agent details retrieved successfully",
+            session_id=session_id,
+            setup_info=SetupInfo(
+                agent_id=agent_details.agent.id,
+                agent_name=agent_details.agent.name,
+                user_readiness=user_readiness,
+                requirements=requirements,
+            ),
+        )

autogpt_platform/backend/backend/server/v2/chat/tools/models.py

@@ -177,7 +177,6 @@ class SetupInfo(BaseModel):
 
     agent_id: str
     agent_name: str
-    version: int
     requirements: dict[str, list[Any]] = Field(
         default_factory=lambda: {
             "credentials": [],