Method __call__

src/tool/default_tools/leetcode.py:33–90

Fetch LeetCode problem information by slug or problem ID. Args: slug (str, optional): The problem slug (e.g., 'two-sum'). Defaults to None. problem_id (int, optional): The problem ID (e.g., 1). Defaults to None. Returns: ToolResp

(self, slug: Optional[str] = None, problem_id: Optional[int] = None, **kwargs)

        super().__init__(require_grad=require_grad, **kwargs)

    async def __call__(self, slug: Optional[str] = None, problem_id: Optional[int] = None, **kwargs) -> ToolResponse:
        """
        Fetch LeetCode problem information by slug or problem ID.

        Args:
            slug (str, optional): The problem slug (e.g., 'two-sum'). Defaults to None.
            problem_id (int, optional): The problem ID (e.g., 1). Defaults to None.
        
        Returns:
            ToolResponse: The response containing problem information.
        """
        try:
            if not slug and not problem_id:
                return ToolResponse(
                    success=False,
                    message="Either 'slug' or 'problem_id' must be provided",
                )
            
            # If only problem_id is provided, we need to get the slug first
            if problem_id and not slug:
                slug = await self._get_slug_by_id(problem_id)
                if not slug:
                    return ToolResponse(
                        success=False,
                        message=f"Problem with ID {problem_id} not found"
                    )
            
            # Fetch problem details using GraphQL
            problem_data = await self._fetch_problem_details(slug)
            
            if not problem_data:
                return ToolResponse(
                    success=False,
                    message=f"Failed to fetch problem details for slug: {slug}"
                )
            
            # Format the response
            formatted_response = self._format_problem_response(problem_data)
            
            return ToolResponse(
                success=True,
                message=formatted_response,
                extra=ToolExtra(
                    file_path=None,
                    data={
                        "slug": slug,
                        "problem_id": problem_id,
                        "problem_data": problem_data
                    }
                )
            )
            
        except Exception as e:
            logger.error(f"Error fetching LeetCode problem: {e}")
            return ToolResponse(
                success=False,
                message=f"Error fetching LeetCode problem: {str(e)}"
            )