From ed5f94e1bbf2950091b407b9e07073f974d9fe3d Mon Sep 17 00:00:00 2001 From: Nathan Evans Date: Fri, 3 Oct 2025 15:21:06 -0700 Subject: [PATCH] Add model_provider --- graphrag/config/init_content.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/graphrag/config/init_content.py b/graphrag/config/init_content.py index 37ad3f04..2775b131 100644 --- a/graphrag/config/init_content.py +++ b/graphrag/config/init_content.py @@ -20,6 +20,7 @@ INIT_YAML = f"""\ models: {defs.DEFAULT_CHAT_MODEL_ID}: type: {defs.DEFAULT_CHAT_MODEL_TYPE.value} + model_provider: openai auth_type: {defs.DEFAULT_CHAT_MODEL_AUTH_TYPE.value} # or azure_managed_identity api_key: ${{GRAPHRAG_API_KEY}} # set this in the generated .env file, or remove if managed identity model: {defs.DEFAULT_CHAT_MODEL} @@ -34,6 +35,7 @@ models: requests_per_minute: null {defs.DEFAULT_EMBEDDING_MODEL_ID}: type: {defs.DEFAULT_EMBEDDING_MODEL_TYPE.value} + model_provider: openai auth_type: {defs.DEFAULT_EMBEDDING_MODEL_AUTH_TYPE.value} api_key: ${{GRAPHRAG_API_KEY}} model: {defs.DEFAULT_EMBEDDING_MODEL}