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

fix agents context retriever

Browse source
This commit is contained in:
Xi Yan 2024-10-10 20:17:29 -07:00
parent 1ff0476002
commit ca29980c6b
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -63,13 +63,12 @@ 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 = await inference_api.chat_completion(
model=model, model=model,
messages=[message], messages=[message],
stream=False, stream=False,
) )
async for chunk in response: query = response.completion_message.content
query = chunk.completion_message.content
return query return query