Merge pull request #630 from soichisumi/support-azureopenai

add "azure_openai" in llm_binding and embedding_binding
HKUDS · Jan 24, 2025 · b71ef4a · b71ef4a
2 parents 63bd066 + a3306c4
commit b71ef4a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py
@@ -551,10 +551,10 @@ async def api_key_auth(api_key_header_value: str | None = Security(api_key_heade
 
 def create_app(args):
     # Verify that bindings arer correctly setup
-    if args.llm_binding not in ["lollms", "ollama", "openai"]:
+    if args.llm_binding not in ["lollms", "ollama", "openai", "azure_openai"]:
         raise Exception("llm binding not supported")
 
-    if args.embedding_binding not in ["lollms", "ollama", "openai"]:
+    if args.embedding_binding not in ["lollms", "ollama", "openai", "azure_openai"]:
         raise Exception("embedding binding not supported")
 
     # Add SSL validation