From 38c8a6e97cae3a5ca5e52474678414506c73ee4c Mon Sep 17 00:00:00 2001
From: Ahmad Hatahet <ahmad.hatahet@hotmail.com>
Date: Sat, 30 Nov 2024 17:11:38 +0100
Subject: [PATCH] add api_version to args

---
 lightrag/llm.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/lightrag/llm.py b/lightrag/llm.py
index 6a191a0f..3c4f9543 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -567,11 +567,14 @@ async def azure_openai_embedding(
     model: str = "text-embedding-3-small",
     base_url: str = None,
     api_key: str = None,
+    api_version: str = None,
 ) -> np.ndarray:
     if api_key:
         os.environ["AZURE_OPENAI_API_KEY"] = api_key
     if base_url:
         os.environ["AZURE_OPENAI_ENDPOINT"] = base_url
+    if api_version:
+        os.environ["AZURE_OPENAI_API_VERSION"] = api_version
 
     openai_async_client = AsyncAzureOpenAI(
         azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),