phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-29 11:24:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix context_retriever model->model_id

Browse source
This commit is contained in:
Xi Yan 2024-12-19 12:52:00 -08:00
parent 03607a68c7
commit 5be2ea37b1
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/inline/agents/meta_reference/rag/context_retriever.py
View file

@ -64,7 +64,7 @@ async def llm_rag_query_generator(
model = config.model model = config.model
message = UserMessage(content=content) message = UserMessage(content=content)
response = await inference_api.chat_completion( response = await inference_api.chat_completion(
model=model, model_id=model,
messages=[message], messages=[message],
stream=False, stream=False,
) )