osbornep8 commited on
Commit
4ada1f6
·
verified ·
1 Parent(s): b90f5ac

Update app.py

Browse files

Added ANTHROPIC_API_KEY

Files changed (1) hide show
  1. app.py +39 -39
app.py CHANGED
@@ -1,39 +1,39 @@
1
- import yaml
2
- import os
3
- from smolagents import GradioUI, CodeAgent, LiteLLMModel
4
-
5
- # Get current directory path
6
- CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
-
8
- from tools.final_answer import FinalAnswerTool as FinalAnswer
9
-
10
-
11
-
12
- model = LiteLLMModel(
13
- max_tokens=5000,
14
- model_id='anthropic/claude-3-5-haiku-20241022',
15
- api_base=None,
16
- )
17
-
18
- final_answer = FinalAnswer()
19
-
20
-
21
- with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
22
- prompt_templates = yaml.safe_load(stream)
23
-
24
- agent = CodeAgent(
25
- model=model,
26
- tools=[],
27
- managed_agents=[],
28
- max_steps=20,
29
- verbosity_level=1,
30
- planning_interval=None,
31
- name=None,
32
- description=None,
33
- executor_type='local',
34
- executor_kwargs={},
35
- max_print_outputs_length=None,
36
- prompt_templates=prompt_templates
37
- )
38
- if __name__ == "__main__":
39
- GradioUI(agent).launch()
 
1
+ import yaml
2
+ import os
3
+ from smolagents import GradioUI, CodeAgent, LiteLLMModel
4
+
5
+ # Get current directory path
6
+ CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
+
8
+ from tools.final_answer import FinalAnswerTool as FinalAnswer
9
+
10
+
11
+
12
+ model = LiteLLMModel(
13
+ max_tokens=5000,
14
+ model_id='anthropic/claude-3-5-haiku-20241022',
15
+ api_key=os.environ['ANTHROPIC_API_KEY'],
16
+ )
17
+
18
+ final_answer = FinalAnswer()
19
+
20
+
21
+ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
22
+ prompt_templates = yaml.safe_load(stream)
23
+
24
+ agent = CodeAgent(
25
+ model=model,
26
+ tools=[],
27
+ managed_agents=[],
28
+ max_steps=20,
29
+ verbosity_level=1,
30
+ planning_interval=None,
31
+ name=None,
32
+ description=None,
33
+ executor_type='local',
34
+ executor_kwargs={},
35
+ max_print_outputs_length=None,
36
+ prompt_templates=prompt_templates
37
+ )
38
+ if __name__ == "__main__":
39
+ GradioUI(agent).launch()