fix typing

leehuwuj · leehuwuj · commit c2136e5a84a1 · 2025-01-21T17:05:22.000+07:00
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
         """

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def retrieve(self, ctx: Context, ev: StartEvent) -> PlanResearchEvent:`
`112`	`112`	`)`
`113`	`113`	`)`
`114`	`114`	`retriever = self.index.as_retriever(`
`115`		`- similarity_top_k=os.getenv("TOP_K", 10),`
	`115`	`+ similarity_top_k=int(os.getenv("TOP_K", 10)),`
`116`	`116`	`)`
`117`	`117`	`nodes = retriever.retrieve(self.user_request)`
`118`	`118`	`self.context_nodes.extend(nodes)`
`@@ -266,7 +266,7 @@ async def answer(self, ctx: Context, ev: ResearchEvent) -> CollectAnswersEvent:`
`266`	`266`	`@step`
`267`	`267`	`async def collect_answers(`
`268`	`268`	`self, ctx: Context, ev: CollectAnswersEvent`
`269`		`- ) -> ReportEvent:`
	`269`	`+ ) -> PlanResearchEvent:`
`270`	`270`	`"""`
`271`	`271`	`Collect answers to all questions`
`272`	`272`	`"""`