From 5be2ea37b1102f38d7dd8f7df5ce8b47a175686f Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 19 Dec 2024 12:52:00 -0800 Subject: [PATCH] fix context_retriever model->model_id --- .../inline/agents/meta_reference/rag/context_retriever.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/inline/agents/meta_reference/rag/context_retriever.py b/llama_stack/providers/inline/agents/meta_reference/rag/context_retriever.py index 1dbe7a91c..7b5c8b4b0 100644 --- 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, )