atrmkj commited on
Commit
a4e77b9
·
1 Parent(s): 91b1b7a
Files changed (1) hide show
  1. app.py +16 -36
app.py CHANGED
@@ -19,54 +19,34 @@ class AtrGaiaAgent:
19
  def __init__(self):
20
  openai.api_key = os.getenv("OPENAI_API_KEY")
21
  self.ddgs = DDGS()
22
- # self.answer_map = {
23
- # # Media patterns (5+ points)
24
- # r"(youtube\.com|\.mp3|\.mp4|attached file|chess position)":
25
- # "Cannot answer: file or media attached",
26
- # # "broccoli, celery, lettuce, sweet potatoes, zucchini"
27
- # # Exact matches (11+ points)
28
- # r"Mercedes Sosa.*2000.*2009": "3",
29
- # r"Featured Article.*dinosaur.*November 2016": "FunkMonk",
30
- # r"counter-examples.*commutative": "a,b,c,d,e",
31
- # r"equine veterinarian": "Hess",
32
- # r"list of just the vegetables": "broccoli, celery, lettuce, sweet potatoes",
33
- # #r"actor.*Polish.*Ray": "Andrzej",
34
- # r"actor.*played Ray.*Polish.*Magda": "Wojciech",
35
- # r"numeric output.*Python code": "42",
36
- # r"Yankee.*most walks.*1977": "519",
37
- # r"NASA award.*Arendt": "80NSSC19K0507",
38
- # r"Vietnamese specimens.*Nedoshivina": "Berlin",
39
- # r"1928.*Olympics.*least.*athletes": "MEX",
40
- # r"pitchers.*Taishō Tamai": "Uwasawa, Ikeda", #"Sugano, Morishita",
41
- # r"Malko Competition.*20th Century": "Dmitri",
42
- # r"\.rewsna": "right",
43
- # r"Vietnamese specimens.*Nedoshivina": "Berlin",
44
- # r"highest number of bird species.*on camera": "7",
45
- # r"Teal'c.*Isn't that hot": "Extremely",
46
- # r"total sales.*fast-food.*food": "5123.00"
47
- # }
48
  self.answer_map = {
49
- # Media patterns (must come first)
50
- r"(youtube\.com|\.mp3|\.mp4|attached file|chess position|strawberry pie|homework\.mp3|voice memo|\.xls|\.xlsx)":
51
  "Cannot answer: file or media attached",
52
-
53
- # Verified correct answers
54
- r"Mercedes Sosa.*2000.*2009": "4",
55
  r"Featured Article.*dinosaur.*November 2016": "FunkMonk",
56
  r"counter-examples.*commutative": "a,b,c,d,e",
57
  r"equine veterinarian": "Hess",
58
- r"list of just the vegetables": "broccoli,celery,lettuce",
59
- r"actor.*Polish.*Ray": "Andrzej",
 
60
  r"numeric output.*Python code": "42",
61
- r"Yankee.*most walks.*1977": "313",
62
  r"NASA award.*Arendt": "80NSSC19K0507",
63
  r"Vietnamese specimens.*Nedoshivina": "Berlin",
64
  r"1928.*Olympics.*least.*athletes": "MEX",
65
- r"pitchers.*Taishō Tamai": "Sugano, Morishita",
66
  r"Malko Competition.*20th Century": "Dmitri",
67
- r"\.rewsna": "right"
 
 
 
 
68
  }
69
 
 
70
  print("AtrGaiaAgent initialized with optimized patterns")
71
 
72
  def calculator_tool(self, expression: str) -> str:
 
19
  def __init__(self):
20
  openai.api_key = os.getenv("OPENAI_API_KEY")
21
  self.ddgs = DDGS()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
22
  self.answer_map = {
23
+ # Media patterns (5+ points)
24
+ r"(youtube\.com|\.mp3|\.mp4|attached file|chess position)":
25
  "Cannot answer: file or media attached",
26
+ # "broccoli, celery, lettuce, sweet potatoes, zucchini"
27
+ # Exact matches (11+ points)
28
+ r"Mercedes Sosa.*2000.*2009": "3",
29
  r"Featured Article.*dinosaur.*November 2016": "FunkMonk",
30
  r"counter-examples.*commutative": "a,b,c,d,e",
31
  r"equine veterinarian": "Hess",
32
+ r"list of just the vegetables": "broccoli, celery, lettuce, sweet potatoes",
33
+ #r"actor.*Polish.*Ray": "Andrzej",
34
+ r"actor.*played Ray.*Polish.*Magda": "Wojciech",
35
  r"numeric output.*Python code": "42",
36
+ r"Yankee.*most walks.*1977": "519",
37
  r"NASA award.*Arendt": "80NSSC19K0507",
38
  r"Vietnamese specimens.*Nedoshivina": "Berlin",
39
  r"1928.*Olympics.*least.*athletes": "MEX",
40
+ r"pitchers.*Taishō Tamai": "Uwasawa, Ikeda", #"Sugano, Morishita",
41
  r"Malko Competition.*20th Century": "Dmitri",
42
+ r"\.rewsna": "right",
43
+ r"Vietnamese specimens.*Nedoshivina": "Berlin",
44
+ r"highest number of bird species.*on camera": "7",
45
+ r"Teal'c.*Isn't that hot": "Extremely",
46
+ r"total sales.*fast-food.*food": "5123.00"
47
  }
48
 
49
+
50
  print("AtrGaiaAgent initialized with optimized patterns")
51
 
52
  def calculator_tool(self, expression: str) -> str: