Tools Configuration

tools_config:
  - name: search_docs
    description: Search the document datastore for relevant information.
    step_config:
      type: SearchUnstructuredDataStep
      config:
        top_k: 50
        lexical_alpha: 0.1
        semantic_alpha: 0.9
        reranker: "ctxl-rerank-v2-instruct-multilingual-FP8"
        rerank_top_k: 12
        reranker_score_filter_threshold: 0.2

tools_config:
  - name: search_and_summarize
    description: Search documents and summarize the results.
    graph_config:
      version: 0.1
      inputs:
        query: str
      outputs:
        summary: str
      nodes:
        search:
          type: SearchUnstructuredDataStep
          config:
            top_k: 20
          input_mapping:
            query: __inputs__#query
        
        summarize:
          type: LanguageModelStep
          config:
            prompt_template: |
              Summarize these search results for the query "{query}":
              {retrievals}
            model_id: "vertex_ai/gemini-2.0-flash"
          input_mapping:
            query: __inputs__#query
            retrievals: search#retrievals
        
        __outputs__:
          type: output
          input_mapping:
            summary: summarize#response

tools_config:
  - name: search_docs
    description: |
      Search internal documentation. Use for questions about 
      company policies, procedures, and technical guides.
    step_config:
      type: SearchUnstructuredDataStep
      config:
        top_k: 50

  - name: search_web
    description: |
      Search the web for current information. Use for questions
      about external topics, news, or publicly available data.
    step_config:
      type: WebSearchStep
      config:
        model: "gemini-2.5-flash"

  - name: query_database
    description: |
      Query the structured database. Use for questions about
      metrics, statistics, or tabular data.
    step_config:
      type: QueryStructuredDatastoreStep

# Good: Specific about when to use
description: |
  Search the HR policy database. Use this tool for questions about:
  - Employee benefits and compensation
  - Time off and leave policies  
  - Performance review processes
  - Hiring and onboarding procedures

# Bad: Too vague
description: Search for HR stuff.

step_config:
  type: SearchUnstructuredDataStep
  config:
    top_k: 50
    lexical_alpha: 0.1
    semantic_alpha: 0.9
    reranker: "ctxl-rerank-v2-instruct-multilingual-FP8"
    rerank_top_k: 12
    reranker_score_filter_threshold: 0.2

tools_config:
  - name: crm_search
    description: Search customer records in the CRM system.
    step_config:
      type: MCPClientStep
      config:
        server_url: "https://mcp-server.example.com"
        tool_name: "search_customers"
        tool_args: '{"query": "$query"}'
        auth_headers:
          Authorization: "Bearer ${CRM_API_TOKEN}"