fix typing

run-llama · Jan 21, 2025 · c2136e5 · c2136e5
1 parent dd7e79d
commit c2136e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/templates/components/agents/python/deep_research/app/workflows/deep_research.py b/templates/components/agents/python/deep_research/app/workflows/deep_research.py
@@ -112,7 +112,7 @@ def retrieve(self, ctx: Context, ev: StartEvent) -> PlanResearchEvent:
             )
         )
         retriever = self.index.as_retriever(
-            similarity_top_k=os.getenv("TOP_K", 10),
+            similarity_top_k=int(os.getenv("TOP_K", 10)),
         )
         nodes = retriever.retrieve(self.user_request)
         self.context_nodes.extend(nodes)
@@ -266,7 +266,7 @@ async def answer(self, ctx: Context, ev: ResearchEvent) -> CollectAnswersEvent:
     @step
     async def collect_answers(
         self, ctx: Context, ev: CollectAnswersEvent
-    ) -> ReportEvent:
+    ) -> PlanResearchEvent:
         """
         Collect answers to all questions
         """