Commit
·
1ca2b4f
1
Parent(s):
387c720
add api_version to azure_openai_complete_if_cache
Browse files- 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"),
|