Merge pull request #802 from MdNazishArmanShorthillsAI/main

Improved variable assignment to use your own azure open ai embedding …
2025-02-17 15:53:37 +08:00
parent fc6dff04ff 44ef234002
commit 5581df0b06
1 changed files with 1 additions and 1 deletions
--- a/lightrag/llm/azure_openai.py
+++ b/lightrag/llm/azure_openai.py
@@ -165,7 +165,7 @@ async def azure_openai_complete(
 )
 async def azure_openai_embed(
    texts: list[str],
-    model: str = "text-embedding-3-small",
+    model: str = os.getenv("EMBEDDING_MODEL", "text-embedding-3-small"),
    base_url: str = None,
    api_key: str = None,
    api_version: str = None,