Merge branch 'main' into fix/agenerate-wrong-exception

Author: natoverse

.semversioner/next-release/major-20250219010328495371.json

@@ -0,0 +1,4 @@
+{
+  "type": "major",
+  "description": "Rework API to accept callbacks."
+}

.semversioner/next-release/minor-20250219193155428669.json

@@ -0,0 +1,4 @@
+{
+  "type": "minor",
+  "description": "Add LMM Manager and Factory, to support provider registration"
+}

.semversioner/next-release/patch-20250219182755748457.json

@@ -0,0 +1,4 @@
+{
+  "type": "patch",
+  "description": "Refactor config defaults from constants to type-safe, hierarchical dataclass."
+}

graphrag/api/prompt_tune.py

@@ -13,10 +13,10 @@
 
 from pydantic import PositiveInt, validate_call
 
-import graphrag.config.defaults as defs
 from graphrag.callbacks.noop_workflow_callbacks import NoopWorkflowCallbacks
+from graphrag.config.defaults import language_model_defaults
 from graphrag.config.models.graph_rag_config import GraphRagConfig
-from graphrag.index.llm.load_llm import load_llm
+from graphrag.language_model.manager import ModelManager
 from graphrag.logger.print_progress import PrintProgressLogger
 from graphrag.prompt_tune.defaults import MAX_TOKEN_COUNT, PROMPT_TUNING_MODEL_ID
 from graphrag.prompt_tune.generator.community_report_rating import (
@@ -102,13 +102,16 @@ async def generate_indexing_prompts(
     # if max_retries is not set, inject a dynamically assigned value based on the number of expected LLM calls
     # to be made or fallback to a default value in the worst case
     if default_llm_settings.max_retries == -1:
-        default_llm_settings.max_retries = min(len(doc_list), defs.LLM_MAX_RETRIES)
+        default_llm_settings.max_retries = min(
+            len(doc_list), language_model_defaults.max_retries
+        )
 
-    llm = load_llm(
-        "prompt_tuning",
-        default_llm_settings,
-        cache=None,
+    llm = ModelManager().register_chat(
+        name="prompt_tuning",
+        model_type=default_llm_settings.type,
+        config=default_llm_settings,
         callbacks=NoopWorkflowCallbacks(),
+        cache=None,
     )
 
     if not domain: