diff --git a/python/graphrag/graphrag/index/default_config/parameters/factories.py b/python/graphrag/graphrag/index/default_config/parameters/factories.py index 34452d6d4e..e56e38a718 100644 --- a/python/graphrag/graphrag/index/default_config/parameters/factories.py +++ b/python/graphrag/graphrag/index/default_config/parameters/factories.py @@ -172,7 +172,7 @@ def section(key: Section): if api_key is None: raise ValueError(LLM_KEY_REQUIRED) llm_type = _str(Fragment.type) - llm_type = LLMType(type) + llm_type = LLMType(llm_type) deployment_name = str(Fragment.deployment_name) is_azure = _is_azure(llm_type) api_base = _str(Fragment.api_base, _api_base) @@ -218,7 +218,7 @@ def section(key: Section): async_mode_enum = AsyncType(async_mode) if async_mode else None deployment_name = _str(Fragment.deployment_name) llm_type = _str(Fragment.type) - llm_type = LLMType(type) + llm_type = LLMType(llm_type) is_azure = _is_azure(llm_type) api_base = _str(Fragment.api_base, _api_base)