MdNazishArman commited on
Commit
5941eda
·
1 Parent(s): ff6f836

Improved variable assignment to use your own azure open ai embedding model

Browse files
Files changed (1) hide show
  1. lightrag/llm/azure_openai.py +1 -1
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,