fix context retriever (#75)

2024-09-18 08:24:36 -07:00 · 2024-09-18 08:24:36 -07:00 · e6fdb9df29
commit e6fdb9df29
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: