# ui/agent/graph/nodes/config.py
from __future__ import annotations

from ...tool_schemas import (
    crawl_web_site,
    get_country_profile,
    scrape_web_page,
    search_immigration_info,
)

RESEARCH_TOOL_SCHEMAS = [
    get_country_profile,
    search_immigration_info,
    scrape_web_page,
    crawl_web_site,
]

MAX_TODOS = 4
MAX_RESEARCH_ROUNDS = 4
TOOL_RESULT_LIMIT = 4000
FINDING_SUMMARY_LIMIT = 2500
RESEARCHER_FINAL_MAX_TOKENS = FINDING_SUMMARY_LIMIT
PLANNER_MAX_TOKENS = 900
PLANNER_TEMPERATURE = 0.2
CONSOLIDATOR_MAX_TOKENS = 3000

RESEARCH_TOOL_NAMES = {
    schema["function"]["name"] for schema in RESEARCH_TOOL_SCHEMAS
}