Amalia commited on
Commit
d393964
·
1 Parent(s): 8a4e9dd

add prompt from gaia

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -93,10 +93,6 @@ def wikipedia_search(query: str, load_max_docs: int=3) -> str:
93
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
94
  class BasicAgent:
95
  def __init__(self):
96
-
97
- model = LiteLLMModel(model_id=config.CLAUDE_3_5_SONNET_MODEL_ID, api_key=ANTHROPIC_API_KEY,
98
- temperature=DEFAULT_TEMPERATURE,
99
- max_retries=3)
100
 
101
  # System prompt recommended by GAIA benchmark instruction https://huggingface.co/spaces/gaia-benchmark/leaderboard
102
  # Emitting the part of final answer, as instructed by the HF course
@@ -135,6 +131,7 @@ class BasicAgent:
135
  # planning_interval=DEFAULT_PLANNING_INTERVAL,
136
  # step_callbacks=[generation_error_interrupter]
137
  )
 
138
 
139
  print("BasicAgent initialized.")
140
  def __call__(self, question: str) -> str:
 
93
  # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
94
  class BasicAgent:
95
  def __init__(self):
 
 
 
 
96
 
97
  # System prompt recommended by GAIA benchmark instruction https://huggingface.co/spaces/gaia-benchmark/leaderboard
98
  # Emitting the part of final answer, as instructed by the HF course
 
131
  # planning_interval=DEFAULT_PLANNING_INTERVAL,
132
  # step_callbacks=[generation_error_interrupter]
133
  )
134
+ self.agent.prompt_templates["system_prompt"] = self.agent.prompt_templates["system_prompt"] + system_prompt
135
 
136
  print("BasicAgent initialized.")
137
  def __call__(self, question: str) -> str: