File size: 1,604 Bytes
b4e2809
 
 
e1896bb
 
 
8b358c4
10c462d
8b358c4
 
e1896bb
 
8b358c4
e1896bb
10c462d
e1896bb
b4e2809
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""Configuration constants for the GAIA agent project."""

from smolagents import OpenAIServerModel, InferenceClientModel

# pylint: disable=line-too-long

# Which questions to answer
QUESTIONS = [1,3,5,8,9,11,13,17,18,20]

# GAIA benchmark scoring API
DEFAULT_API_URL = 'https://agents-course-unit4-scoring.hf.space'

# Additional instructions for agent. See here: https://huggingface.co/spaces/gaia-benchmark/leaderboard
INSTRUCTIONS = """
You are a general AI assistant. I will ask you a question. Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string. Submit the final answer via the final_answer tool.
"""

# Agent model definitions
MANAGER_MODEL = InferenceClientModel(
    "deepseek-ai/DeepSeek-V3",
    provider="together",
    max_tokens=64000
)

WORKER_MODEL = InferenceClientModel(
    "deepseek-ai/DeepSeek-V3",
    provider="together",
    max_tokens=64000
)

CHECK_MODEL = InferenceClientModel(
    "deepseek-ai/DeepSeek-V3",
    provider="together",
    max_tokens=64000
)

MODEL = OpenAIServerModel(
    model_id="gpt-4.1",
    max_tokens=8000
)

TOKEN_LIMITER = 5000
STEP_WAIT = 60