config fallback model

Author: anhnh2002

README.md

@@ -49,7 +49,8 @@ codewiki config set \
   --api-key YOUR_API_KEY \
   --base-url https://api.anthropic.com \
   --main-model claude-sonnet-4 \
-  --cluster-model claude-sonnet-4
+  --cluster-model claude-sonnet-4 \
+  --fallback-model glm-4p5
 ```
 
 ### 3. Generate Documentation
@@ -101,7 +102,8 @@ codewiki config set \
   --api-key <your-api-key> \
   --base-url <provider-url> \
   --main-model <model-name> \
-  --cluster-model <model-name>
+  --cluster-model <model-name> \
+  --fallback-model <model-name>
 
 # Show current configuration
 codewiki config show

codewiki/cli/adapters/doc_generator.py

@@ -135,7 +135,8 @@ def generate(self) -> DocumentationJob:
                 llm_base_url=self.config.get('base_url'),
                 llm_api_key=self.config.get('api_key'),
                 main_model=self.config.get('main_model'),
-                cluster_model=self.config.get('cluster_model')
+                cluster_model=self.config.get('cluster_model'),
+                fallback_model=self.config.get('fallback_model')
             )
 
             # Run backend documentation generation

codewiki/cli/commands/config.py

@@ -50,11 +50,17 @@ def config_group():
     type=str,
     help="Model for module clustering (recommend top-tier)"
 )
+@click.option(
+    "--fallback-model",
+    type=str,
+    help="Fallback model for documentation generation"
+)
 def config_set(
     api_key: Optional[str],
     base_url: Optional[str],
     main_model: Optional[str],
-    cluster_model: Optional[str]
+    cluster_model: Optional[str],
+    fallback_model: Optional[str]
 ):
     """
     Set configuration values for CodeWiki.
@@ -69,15 +75,15 @@ def config_set(
     \b
     # Set all configuration
     $ codewiki config set --api-key sk-abc123 --base-url https://api.anthropic.com \\
-        --main-model claude-sonnet-4 --cluster-model claude-sonnet-4
+        --main-model claude-sonnet-4 --cluster-model claude-sonnet-4 --fallback-model glm-4p5
     
     \b
     # Update only API key
     $ codewiki config set --api-key sk-new-key
     """
     try:
         # Check if at least one option is provided
-        if not any([api_key, base_url, main_model, cluster_model]):
+        if not any([api_key, base_url, main_model, cluster_model, fallback_model]):
             click.echo("No options provided. Use --help for usage information.")
             sys.exit(EXIT_CONFIG_ERROR)
 
@@ -96,6 +102,9 @@ def config_set(
         if cluster_model:
             validated_data['cluster_model'] = validate_model_name(cluster_model)
 
+        if fallback_model:
+            validated_data['fallback_model'] = validate_model_name(fallback_model)
+        
         # Create config manager and save
         manager = ConfigManager()
         manager.load()  # Load existing config if present
@@ -104,7 +113,8 @@ def config_set(
             api_key=validated_data.get('api_key'),
             base_url=validated_data.get('base_url'),
             main_model=validated_data.get('main_model'),
-            cluster_model=validated_data.get('cluster_model')
+            cluster_model=validated_data.get('cluster_model'),
+            fallback_model=validated_data.get('fallback_model')
         )
 
         # Display success messages
@@ -138,6 +148,9 @@ def config_set(
                     "   Recommended models: claude-opus, claude-sonnet-4, gpt-4, gpt-4-turbo"
                 )
 
+        if fallback_model:
+            click.secho(f"✓ Fallback model: {fallback_model}", fg="green")
+        
         click.echo("\n" + click.style("Configuration updated successfully.", fg="green", bold=True))
 
     except ConfigurationError as e:
@@ -177,7 +190,7 @@ def config_show(output_json: bool):
             click.secho("\n✗ Configuration not found.", fg="red", err=True)
             click.echo("\nPlease run 'codewiki config set' to configure your API credentials:")
             click.echo("  codewiki config set --api-key <key> --base-url <url> \\")
-            click.echo("    --main-model <model> --cluster-model <model>")
+            click.echo("    --main-model <model> --cluster-model <model> --fallback-model <model>")
             click.echo("\nFor more help: codewiki config set --help")
             sys.exit(EXIT_CONFIG_ERROR)
 
@@ -192,6 +205,7 @@ def config_show(output_json: bool):
                 "base_url": config.base_url if config else "",
                 "main_model": config.main_model if config else "",
                 "cluster_model": config.cluster_model if config else "",
+                "fallback_model": config.fallback_model if config else "glm-4p5",
                 "default_output": config.default_output if config else "docs",
                 "config_file": str(manager.config_file_path)
             }
@@ -216,6 +230,7 @@ def config_show(output_json: bool):
                 click.echo(f"  Base URL:         {config.base_url or 'Not set'}")
                 click.echo(f"  Main Model:       {config.main_model or 'Not set'}")
                 click.echo(f"  Cluster Model:    {config.cluster_model or 'Not set'}")
+                click.echo(f"  Fallback Model:   {config.fallback_model or 'Not set'}")
             else:
                 click.secho("  Not configured", fg="yellow")
 
@@ -339,8 +354,9 @@ def config_validate(quick: bool, verbose: bool):
             click.echo("[4/5] Checking model configuration...")
             click.echo(f"      Main model: {config.main_model}")
             click.echo(f"      Cluster model: {config.cluster_model}")
+            click.echo(f"      Fallback model: {config.fallback_model}")
 
-        if not config.main_model or not config.cluster_model:
+        if not config.main_model or not config.cluster_model or not config.fallback_model:
             click.secho("✗ Models not configured", fg="red")
             sys.exit(EXIT_CONFIG_ERROR)
 
@@ -349,6 +365,7 @@ def config_validate(quick: bool, verbose: bool):
         else:
             click.secho(f"✓ Main model configured: {config.main_model}", fg="green")
             click.secho(f"✓ Cluster model configured: {config.cluster_model}", fg="green")
+            click.secho(f"✓ Fallback model configured: {config.fallback_model}", fg="green")
 
         # Warn about non-top-tier cluster model
         if not is_top_tier_model(config.cluster_model):

codewiki/cli/commands/generate.py

@@ -201,6 +201,7 @@ def generate_command(
             config={
                 'main_model': config.main_model,
                 'cluster_model': config.cluster_model,
+                'fallback_model': config.fallback_model,
                 'base_url': config.base_url,
                 'api_key': api_key,
             },

codewiki/cli/config_manager.py

@@ -87,6 +87,7 @@ def save(
         base_url: Optional[str] = None,
         main_model: Optional[str] = None,
         cluster_model: Optional[str] = None,
+        fallback_model: Optional[str] = None,
         default_output: Optional[str] = None
     ):
         """
@@ -97,6 +98,7 @@ def save(
             base_url: LLM API base URL
             main_model: Primary model
             cluster_model: Clustering model
+            fallback_model: Fallback model
             default_output: Default output directory
         """
         # Ensure config directory exists
@@ -114,6 +116,7 @@ def save(
                     base_url="",
                     main_model="",
                     cluster_model="",
+                    fallback_model="glm-4p5",
                     default_output="docs"
                 )
 
@@ -124,6 +127,8 @@ def save(
             self._config.main_model = main_model
         if cluster_model is not None:
             self._config.cluster_model = cluster_model
+        if fallback_model is not None:
+            self._config.fallback_model = fallback_model
         if default_output is not None:
             self._config.default_output = default_output
 

codewiki/cli/models/config.py

@@ -26,11 +26,13 @@ class Configuration:
         base_url: LLM API base URL
         main_model: Primary model for documentation generation
         cluster_model: Model for module clustering
+        fallback_model: Fallback model for documentation generation
         default_output: Default output directory
     """
     base_url: str
     main_model: str
     cluster_model: str
+    fallback_model: str = "glm-4p5"
     default_output: str = "docs"
 
     def validate(self):
@@ -43,6 +45,7 @@ def validate(self):
         validate_url(self.base_url)
         validate_model_name(self.main_model)
         validate_model_name(self.cluster_model)
+        validate_model_name(self.fallback_model)
 
     def to_dict(self) -> dict:
         """Convert to dictionary."""
@@ -63,6 +66,7 @@ def from_dict(cls, data: dict) -> 'Configuration':
             base_url=data.get('base_url', ''),
             main_model=data.get('main_model', ''),
             cluster_model=data.get('cluster_model', ''),
+            fallback_model=data.get('fallback_model', 'glm-4p5'),
             default_output=data.get('default_output', 'docs'),
         )
 
@@ -71,7 +75,8 @@ def is_complete(self) -> bool:
         return bool(
             self.base_url and 
             self.main_model and 
-            self.cluster_model
+            self.cluster_model and
+            self.fallback_model
         )
 
     def to_backend_config(self, repo_path: str, output_dir: str, api_key: str):
@@ -97,6 +102,7 @@ def to_backend_config(self, repo_path: str, output_dir: str, api_key: str):
             llm_base_url=self.base_url,
             llm_api_key=api_key,
             main_model=self.main_model,
-            cluster_model=self.cluster_model
+            cluster_model=self.cluster_model,
+            fallback_model=self.fallback_model
         )