kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-01-14 09:07:20 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Take out temporary retry_strategy = none since it is not fnllm compatible

Browse Source
This commit is contained in:
Nathan Evans 2025-10-01 15:50:58 -07:00
parent d0a4c33fa3
commit 8d6db47a1c
1 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphrag/index/validate_config.py
View File

@ -17,10 +17,6 @@ logger = logging.getLogger(__name__)
def validate_config_names(parameters: GraphRagConfig) -> None:
"""Validate config file for model deployment name typos, by running a quick test message for each."""
for id, config in parameters.models.items():
# temporarily disable retry since this is just a connectivity test
retry_strategy = config.retry_strategy
config.retry_strategy = "none"
if config.type in ["chat", "azure_openai", "openai"]:
llm = ModelManager().register_chat(
name="test-llm",
@ -55,5 +51,3 @@ def validate_config_names(parameters: GraphRagConfig) -> None:
logger.error(f"Embedding configuration error detected.\n{e}") # noqa
print(f"Failed to validate embedding model ({id}) params", e) # noqa: T201
sys.exit(1)
config.retry_strategy = retry_strategy