Spaces:
Running
Running
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 =
|
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 |
-
|
|
|
|
|
|
|
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 |
|