Fix Discovery.discover()

RussellLuo · RussellLuo · commit 3e913c3941b7 · 2025-08-07T11:02:14.000+08:00
diff --git a/coagent/core/discovery.py b/coagent/core/discovery.py
@@ -180,7 +180,7 @@ async def discover(self, msg: DiscoveryQuery, ctx: Context) -> DiscoveryReply:
             extensions=msg.extensions,
             queries=[msg],
         )
-        batch_reply = await self.batch_discover(batch_query)
+        batch_reply = await self.batch_discover(batch_query, ctx)
         return batch_reply.results[0]
 
     @handler

Original file line number	Diff line number	Diff line change
`@@ -180,7 +180,7 @@ async def discover(self, msg: DiscoveryQuery, ctx: Context) -> DiscoveryReply:`
`180`	`180`	`extensions=msg.extensions,`
`181`	`181`	`queries=[msg],`
`182`	`182`	`)`
`183`		`- batch_reply = await self.batch_discover(batch_query)`
	`183`	`+ batch_reply = await self.batch_discover(batch_query, ctx)`
`184`	`184`	`return batch_reply.results[0]`
`185`	`185`
`186`	`186`	`@handler`