From a3306c46335a89b57d9d17cbcafedae118c908fb Mon Sep 17 00:00:00 2001 From: Soichi Sumi Date: Thu, 23 Jan 2025 21:30:57 +0900 Subject: [PATCH] add "azure_openai" in llm_binding and embedding_binding --- lightrag/api/lightrag_server.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lightrag/api/lightrag_server.py b/lightrag/api/lightrag_server.py index ffa65464..16f3ae49 100644 --- 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