gardel/lightrag
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
parent 28a84b2aa2 df69d386c5
commit 0a9089ff74
1 changed files with 2 additions and 2 deletions
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( 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: ) -> str:
keyword_extraction = kwargs.pop("keyword_extraction", None) keyword_extraction = kwargs.pop("keyword_extraction", None)
result = await azure_openai_complete_if_cache( result = await azure_openai_complete_if_cache(
"conversation-4o-mini", model,
prompt, prompt,
system_prompt=system_prompt, system_prompt=system_prompt,
history_messages=history_messages, history_messages=history_messages,