Create research_agent.py

agents/research_agent.py

@@ -0,0 +1,19 @@
+from .dataset_semantic_search_agent import DatasetSemanticSearchAgent
+from .google_serper_search_agent import GoogleSerperSearchAgent
+
+
+class ResearchAgent:
+    def __init__(self, web_sources_path: str = "sources.txt", local_sources_path: str = "articles") -> None:
+        self._google_serper_search_agent = GoogleSerperSearchAgent(web_sources_path)
+        self._dataset_semantic_search_agent = DatasetSemanticSearchAgent(local_sources_path)
+
+    def run(self, themes: list[list[str]]) -> dict[str, dict[str, list]]:
+        sources = {}
+
+        for theme in themes:
+            web_sources = self._google_serper_search_agent.run(theme["query"])
+            dataset_sources = self._dataset_semantic_search_agent.run(theme["query"])
+
+            sources[theme] = {"web_sources": web_sources, "dataset_sources": dataset_sources}
+
+        return sources