update config construction

microsoft · Oct 25, 2024 · 9d607a1 · 9d607a1
1 parent dfe687d
commit 9d607a1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphrag/index/llm/load_llm.py b/graphrag/index/llm/load_llm.py
@@ -121,7 +121,7 @@ def _load_openai_chat_llm(
 ):
     configuration = validator.validate_python({
         "api_key": config.api_key,
-        "api_base": config.api_base,
+        "endpoint": config.api_base,
         "api_version": config.api_version,
         "organization": config.organization,
         "max_retries": config.max_retries,
@@ -156,7 +156,7 @@ def _load_openai_embeddings_llm(
 ):
     configuration = validator.validate_python({
         "api_key": config.api_key,
-        "api_base": config.api_base,
+        "endpoint": config.api_base,
         "api_version": config.api_version,
         "organization": config.organization,
         "max_retries": config.max_retries,