gmakstutis/LightRAG
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge pull request #597 from nickjfrench/fix-hardcoded-model

Browse source 
Fixes #596 - Hardcoded model deployment name in azure_openai_complete
This commit is contained in:
zrguo 2025-01-19 12:56:43 +08:00 • committed by GitHub
commit 0a9089ff74
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lightrag/llm.py
View file

@ -622,11 +622,11 @@ async def nvidia_openai_complete(
async def azure_openai_complete(
prompt, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
model: str = "gpt-4o-mini", prompt, system_prompt=None, history_messages=[], keyword_extraction=False, **kwargs
) -> str:
keyword_extraction = kwargs.pop("keyword_extraction", None)
result = await azure_openai_complete_if_cache(
"conversation-4o-mini",
model,
prompt,
system_prompt=system_prompt,
history_messages=history_messages,