We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8edb266 commit 36d2f88Copy full SHA for 36d2f88
1 file changed
src/google/adk/tools/retrieval/vertex_ai_rag_retrieval.py
@@ -16,8 +16,8 @@
16
17
from __future__ import annotations
18
19
-import logging
20
import asyncio
+import logging
21
from typing import Any
22
from typing import TYPE_CHECKING
23
@@ -97,7 +97,7 @@ async def run_async(
97
98
response = await asyncio.to_thread(
99
rag.retrieval_query,
100
- text=args["query"],
+ text=args['query'],
101
rag_resources=self.vertex_rag_store.rag_resources,
102
rag_corpora=self.vertex_rag_store.rag_corpora,
103
similarity_top_k=self.vertex_rag_store.similarity_top_k,
0 commit comments