File size: 938 Bytes
7e2f74d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from tools.base_tool import BaseTool
from services.repositoryMemory import memory_service
from typing import Any

class ApiLookupTool(BaseTool):
    @property
    def name(self) -> str:
        return "api_lookup"

    @property
    def description(self) -> str:
        return "Queries project HTTP routes, api endpoints, and controllers. Inputs: repo_id (str)."

    def execute(self, **kwargs) -> Any:
        repo_id = kwargs.get("repo_id")
        if not repo_id:
            return {"error": "Missing required parameter: repo_id."}
            
        data = memory_service.retrieve(repo_id)
        if not data or "profile" not in data:
            return {"error": f"No profile found for repository {repo_id}."}
            
        profile = data["profile"]
        return {
            "api_endpoints": profile.get("api_endpoints", []),
            "authentication_methods": profile.get("authentication_methods", [])
        }