phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix context retriever (#75)

Browse source
This commit is contained in:
Xi Yan 2024-09-18 08:24:36 -07:00 • committed by GitHub
parent 055770a791
commit e6fdb9df29
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/impls/meta_reference/agents/rag/context_retriever.py
View file

@ -63,11 +63,9 @@ async def llm_rag_query_generator(
model = config.model model = config.model
message = UserMessage(content=content) message = UserMessage(content=content)
response = inference_api.chat_completion( response = inference_api.chat_completion(
ChatCompletionRequest( model=model,
model=model, messages=[message],
messages=[message], stream=False,
stream=False,
)
) )
async for chunk in response: async for chunk in response: