Merge branch 'main' into rag_scoring_fn_1

This commit is contained in:
Xi Yan 2024-12-19 14:10:48 -08:00
commit 13720cbedf

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,
) )