fix context_retriever model->model_id

2024-12-19 12:52:00 -08:00 · 2024-12-19 12:52:00 -08:00 · 5be2ea37b1
commit 5be2ea37b1
parent 03607a68c7
1 changed files with 1 additions and 1 deletions
--- a/llama_stack/providers/inline/agents/meta_reference/rag/context_retriever.py
+++ b/llama_stack/providers/inline/agents/meta_reference/rag/context_retriever.py
@ -64,7 +64,7 @@ async def llm_rag_query_generator(
    model = config.model
    message = UserMessage(content=content)
    response = await inference_api.chat_completion(
-        model=model,
+        model_id=model,
        messages=[message],
        stream=False,
    )