Commit
·
5941eda
1
Parent(s):
ff6f836
Improved variable assignment to use your own azure open ai embedding model
Browse files
lightrag/llm/azure_openai.py
CHANGED
@@ -165,7 +165,7 @@ async def azure_openai_complete(
|
|
165 |
)
|
166 |
async def azure_openai_embed(
|
167 |
texts: list[str],
|
168 |
-
model: str = "text-embedding-3-small",
|
169 |
base_url: str = None,
|
170 |
api_key: str = None,
|
171 |
api_version: str = None,
|
|
|
165 |
)
|
166 |
async def azure_openai_embed(
|
167 |
texts: list[str],
|
168 |
+
model: str = os.getenv("EMBEDDING_MODEL", "text-embedding-3-small"),
|
169 |
base_url: str = None,
|
170 |
api_key: str = None,
|
171 |
api_version: str = None,
|