ahmadhatahet commited on
Commit
1ca2b4f
·
1 Parent(s): 387c720

add api_version to azure_openai_complete_if_cache

Browse files
Files changed (1) hide show
  1. lightrag/llm.py +3 -0
lightrag/llm.py CHANGED
@@ -92,12 +92,15 @@ async def azure_openai_complete_if_cache(
92
  history_messages=[],
93
  base_url=None,
94
  api_key=None,
 
95
  **kwargs,
96
  ):
97
  if api_key:
98
  os.environ["AZURE_OPENAI_API_KEY"] = api_key
99
  if base_url:
100
  os.environ["AZURE_OPENAI_ENDPOINT"] = base_url
 
 
101
 
102
  openai_async_client = AsyncAzureOpenAI(
103
  azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
 
92
  history_messages=[],
93
  base_url=None,
94
  api_key=None,
95
+ api_version=None,
96
  **kwargs,
97
  ):
98
  if api_key:
99
  os.environ["AZURE_OPENAI_API_KEY"] = api_key
100
  if base_url:
101
  os.environ["AZURE_OPENAI_ENDPOINT"] = base_url
102
+ if api_version:
103
+ os.environ["AZURE_OPENAI_API_VERSION"] = api_version
104
 
105
  openai_async_client = AsyncAzureOpenAI(
106
  azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),