Spaces:
Running
Running
File size: 1,637 Bytes
763ec84 7e103cf 763ec84 7e103cf 763ec84 7e103cf 763ec84 7e103cf 763ec84 |
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 49 50 51 52 53 54 55 56 57 |
import os
from tools import (
get_area_lat_lon,
get_wave_forecast,
get_wind_forecast,
)
from any_agent.logging import logger
from any_agent.tools.web_browsing import search_tavily, search_web, visit_webpage
MODEL_OPTIONS = [
"openai/gpt-4.1-nano",
"openai/gpt-4.1-mini",
"openai/gpt-4o",
"gemini/gemini-2.0-flash-lite",
"gemini/gemini-2.0-flash",
]
DEFAULT_EVALUATION_MODEL = MODEL_OPTIONS[0]
DEFAULT_EVALUATION_CRITERIA = [
{
"criteria": "Check if the agent considered at least three surf spot options",
},
{
"criteria": "Check if the agent gathered wind forecasts for each surf spot being evaluated.",
},
{
"criteria": "Check if the agent gathered wave forecasts for each surf spot being evaluated.",
},
{
"criteria": "Check if the agent used any web search tools to explore which surf spots should be considered",
},
{
"criteria": "Check if the final answer contains any description about the weather (air temp, chance of rain, etc) at the chosen location",
},
{
"criteria": "Check if the final answer includes one of the surf spots evaluated by tools",
},
{
"criteria": "Check if the final answer includes information about some alternative surf spots if the user is not satisfied with the chosen one",
},
]
DEFAULT_TOOLS = [
get_wind_forecast,
get_wave_forecast,
get_area_lat_lon,
search_web,
visit_webpage,
]
if os.getenv("TAVILY_API_KEY"):
DEFAULT_TOOLS.append(search_tavily)
else:
logger.warning("TAVILY_API_KEY not set, skipping Tavily search tool")
|