configurable cluster llm model

Author: anhnh2002

env.example

@@ -6,6 +6,7 @@
 
 MAIN_MODEL = claude-sonnet-4
 FALLBACK_MODEL_1 = glm-4p5
+CLUSTER_MODEL = claude-sonnet-4
 LLM_BASE_URL = http://litellm:4000/
 LLM_API_KEY = sk-1234
 

src/cluster_modules.py

@@ -6,7 +6,7 @@
 from dependency_analyzer.models.core import Node
 from llm_services import call_llm
 from utils import count_tokens
-from config import MAX_TOKEN_PER_MODULE, MAIN_MODEL
+from config import MAX_TOKEN_PER_MODULE, CLUSTER_MODEL
 from prompt_template import format_cluster_prompt
 
 
@@ -57,7 +57,7 @@ def cluster_modules(
         return {}
 
     prompt = format_cluster_prompt(potential_core_components, current_module_tree, current_module_name)
-    response = call_llm(prompt, model=MAIN_MODEL)
+    response = call_llm(prompt, model=CLUSTER_MODEL)
 
     #parse the response
     try:

src/config.py

@@ -18,6 +18,7 @@
 # LLM services
 MAIN_MODEL = os.getenv('MAIN_MODEL', 'claude-sonnet-4')
 FALLBACK_MODEL_1 = os.getenv('FALLBACK_MODEL_1', 'glm-4p5')
+CLUSTER_MODEL = os.getenv('CLUSTER_MODEL', MAIN_MODEL)
 LLM_BASE_URL = os.getenv('LLM_BASE_URL', 'http://0.0.0.0:4000/')
 LLM_API_KEY = os.getenv('LLM_API_KEY', 'sk-1234')