jatinror commited on
Commit
096395b
·
verified ·
1 Parent(s): 8700a10

Upload alfred.py

Browse files
Files changed (1) hide show
  1. alfred.py +36 -0
alfred.py ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dotenv import load_dotenv
3
+
4
+ from smolagents import (
5
+ CodeAgent,
6
+ OpenAIServerModel,
7
+ DuckDuckGoSearchTool,
8
+ )
9
+
10
+ from tools.rag_tool import search_guest_story
11
+ from tools.weather_tool import weather_info_tool
12
+ from tools.hub_stats_tool import hub_stats_tool
13
+
14
+ # Load environment variables
15
+ load_dotenv()
16
+
17
+ # Create Groq model
18
+ model = OpenAIServerModel(
19
+ model_id="llama-3.3-70b-versatile",
20
+ api_key=os.getenv("GROQ_API_KEY"),
21
+ api_base="https://api.groq.com/openai/v1",
22
+ )
23
+
24
+ # Create DuckDuckGo search tool
25
+ search_tool = DuckDuckGoSearchTool()
26
+
27
+ # Create Alfred Agent
28
+ agent = CodeAgent(
29
+ tools=[
30
+ search_guest_story,
31
+ search_tool,
32
+ weather_info_tool,
33
+ hub_stats_tool,
34
+ ],
35
+ model=model,
36
+ )