fix context retriever

2025-07-29 15:23:51 +00:00 · 2024-09-18 08:13:44 -07:00 · 2024-09-18 08:13:44 -07:00 · a2fac5d281
commit a2fac5d281
parent 055770a791
1 changed files with 3 additions and 5 deletions
--- a/llama_stack/providers/impls/meta_reference/agents/rag/context_retriever.py
+++ b/llama_stack/providers/impls/meta_reference/agents/rag/context_retriever.py
@ -63,11 +63,9 @@ async def llm_rag_query_generator(
    model = config.model
    message = UserMessage(content=content)
    response = inference_api.chat_completion(
-        ChatCompletionRequest(
-            model=model,
-            messages=[message],
-            stream=False,
-        )
+        model=model,
+        messages=[message],
+        stream=False,
    )

    async for chunk in response: