Gym/resources_servers/finance_sec_search/configs/finance_sec_search.yaml at main · hackIDLE/Gym · GitHub

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
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
# Finance SEC search: SEC filing search with optional web search (via tavily_api_key in env.yaml)
finance_sec_search_resources_server:
  resources_servers:
    finance_sec_search:
      entrypoint: app.py
      domain: agent
      verified: false
      description: SEC EDGAR filing search for financial analysis questions
      value: Enable LLMs to search and analyze SEC filings
      cache_dir: cache
      retrieval_model_server:
        type: responses_api_models
        name: policy_model
      judge_model_server:
        type: responses_api_models
        name: search_judge_model
      judge_responses_create_params:
        input: []
      # Judge prompt: loaded from this file at startup. To override inline (e.g. from
      # NVFlow), set judge_prompt_template directly — it takes priority over the file.
      judge_prompt_template_fpath: prompt_templates/finance_sec_search_judge.yaml
      # Retrieval model parameters (adjust when swapping between thinking/instruct models)
      retrieval_max_output_tokens: 8192
      retrieval_model_context_length: 131072

# Judge model server (OpenAI or local — configured via env.yaml)
search_judge_model:
  responses_api_models:
    openai_model:
      entrypoint: app.py
      openai_base_url: ${search_judge_model_base_url}
      openai_api_key: ${search_judge_model_api_key}
      openai_model: ${search_judge_model_name}

finance_agent:
  responses_api_agents:
    simple_agent:
      entrypoint: app.py
      resources_server:
        type: resources_servers
        name: finance_sec_search_resources_server
      model_server:
        type: responses_api_models
        name: policy_model
      datasets:
      - name: example
        type: example
        jsonl_fpath: resources_servers/finance_sec_search/data/example.jsonl