Agent Tool Definition

class SemanticSearchToolSchema(ToolSchema, name="find_relevant_files"):
    """Use this tool to search for files that are most relevant for a given task or question.
    The conversation history will be passed in with your query.

    - Call this tool ONE TIME, including all of your tasks/questions.
    - DO NOT call it multiple times before ending your turn.
    - Prefer this as the first step to explore or plan edits within a code repository.
    - Prefer this over using a bash command like `grep` or `find`.
    - Data files may be excluded from the search.

    The response will be a list of file paths and contents relative to the repository root, ordered from most relevant to least relevant. The format will look like this:

    path/to/file1
    ```
    file1 content...
    ```

    path/to/file2
    ```
    file2 content...
    ```
    """

    query: str = Field(
        ...,
        description="""Natural language description of what you are looking for in the repository.
        You can describe a particular change you want to make, a feature you want to implement, bug you are trying to fix, information you are searching for, etc.""",
    )

class SemanticSearchTool(Tool[SemanticSearchToolSchema], schema=SemanticSearchToolSchema):

    # Set a token limit for files to surface, and filter out low relevance files
    score_threshold: float = 0.5
    token_limit: int = 30_000

    def execute_impl(
        self,
        tool_input: SemanticSearchToolSchema,
        context: Context,
    ) -> str:
        try:
            client = RelaceClient()
            # Get list of files from repo, ignoring config/data/binary/generated files
            files = [
                {
                    "filename": str(path),
                    "content": path.read_text()
                }
                for path in context.repo.list_tracked_files(
                    ignore_tracked=(
                        # Ignore non-code files like:
                        # - Config files (yaml, env)
                        # - Lock files (package-lock.json)
                        # - Data files (csv, json)
                        # - Binary files (images)
                        # - Build output
                    )
                )
            ]

            # Call reranker with files, conversation context, and query
            ranked = client.rank_files(
                codebase=files,
                query=(
                    f"<conversation>{context.history.get_summary()}</conversation>\n"
                    f"<query>{tool_input.query}</query>"
                ),
                token_limit=self.token_limit,
                threshold=self.score_threshold,
            )

            # Construct response
            return "\n\n".join(
                f"{file_path.relative_to(context.repo.root_path)}\n"
                f"```\n{file_path.read_text()}\n```"
                for file_path in ranked
            )
        except Exception as e:
            raise ToolError(f"Error finding files: {str(e)}") from e