Aditya0619 commited on
Commit
8b235fb
·
verified ·
1 Parent(s): 19d65d2

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +42 -0
config.py ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Configuration file for CrewAI Agent
2
+
3
+ # LLM Provider Configuration
4
+ DEFAULT_PROVIDER = "google" # Options: "google", "huggingface"
5
+
6
+ # Model Configuration per Provider
7
+ MODELS = {
8
+ "google": "gemini-2.0-flash",
9
+ "huggingface": "microsoft/DialoGPT-medium"
10
+ }
11
+
12
+ # Search Configuration
13
+ MAX_SEARCH_RESULTS = {
14
+ "wikipedia": 2,
15
+ "web": 3,
16
+ "arxiv": 3
17
+ }
18
+
19
+ # Agent Configuration
20
+ AGENT_ROLES = {
21
+ "research": {
22
+ "role": "Research Specialist",
23
+ "goal": "Gather comprehensive and accurate information from multiple sources",
24
+ "tools": ["wikipedia_search", "web_search", "arxiv_search"]
25
+ },
26
+ "calculation": {
27
+ "role": "Mathematical Analyst",
28
+ "goal": "Perform accurate mathematical calculations and analysis",
29
+ "tools": ["calculator"]
30
+ },
31
+ "general": {
32
+ "role": "General Assistant",
33
+ "goal": "Provide comprehensive answers by coordinating with specialized agents",
34
+ "tools": "all"
35
+ }
36
+ }
37
+
38
+ # Task Assignment Keywords
39
+ TASK_KEYWORDS = {
40
+ "calculation": ["calculate", "compute", "add", "subtract", "multiply", "divide", "math", "equation", "solve"],
41
+ "research": ["search", "find", "research", "information", "what is", "who is", "when", "where", "how", "explain"]
42
+ }