fix(pipeline): return query from QueryPool.add_query (#2198)

2026-06-07 14:26:03 +00:00 · 2026-05-16 11:36:10 +08:00
parent 245e798b79
commit e425cf079a
2 changed files with 76 additions and 0 deletions
--- a/src/langbot/pkg/pipeline/pool.py
+++ b/src/langbot/pkg/pipeline/pool.py
@@ -63,6 +63,7 @@ class QueryPool:
            self.cached_queries[query_id] = query
            self.query_id_counter += 1
            self.condition.notify_all()
+            return query

    async def __aenter__(self):
        await self.pool_lock.acquire()