abdoh-alkhateeb commited on
Commit
ad719fb
1 Parent(s): bc3e1cb

Apply minor fixes and improvements

Browse files
agents/google_serper_search_agent.py CHANGED
@@ -16,7 +16,7 @@ class GoogleSerperSearchAgent:
16
 
17
  nltk.download("punkt")
18
 
19
- def run(self, query: str, limit_per_source: int = 5) -> tuple[pd.DataFrame, dict[str, float]]:
20
  results = []
21
  for site in self._sources:
22
  results.extend(self._search(query, site, limit_per_source))
 
16
 
17
  nltk.download("punkt")
18
 
19
+ def run(self, query: str, limit_per_source: int = 3) -> tuple[pd.DataFrame, dict[str, float]]:
20
  results = []
21
  for site in self._sources:
22
  results.extend(self._search(query, site, limit_per_source))
agents/synthesis_agent.py CHANGED
@@ -77,7 +77,10 @@ class SynthesisAgent:
77
 
78
  elements = [element[element.index(":") + 1 :].strip() for element in all_sub_themes[i].strip().split("\n") if element and ":" in element]
79
  for i in range(0, len(elements), 3):
80
- section_b += f"###### {elements[i]}\n{elements[i + 1]}\n\n{elements[i + 2]}\n\n"
 
 
 
81
 
82
  section_b += f"###### References:\n" + "\n".join(references[theme["title"]]) + "\n\n\n"
83
 
 
77
 
78
  elements = [element[element.index(":") + 1 :].strip() for element in all_sub_themes[i].strip().split("\n") if element and ":" in element]
79
  for i in range(0, len(elements), 3):
80
+ try:
81
+ section_b += f"###### {elements[i]}\n{elements[i + 1]}\n\n{elements[i + 2]}\n\n"
82
+ except IndexError:
83
+ pass
84
 
85
  section_b += f"###### References:\n" + "\n".join(references[theme["title"]]) + "\n\n\n"
86