Merge pull request #24 from FSoft-AI4Code/feat/traceback

add traceback when errors

Author: anhnh2002

codewiki/cli/commands/generate.py

@@ -4,6 +4,7 @@
 
 import sys
 import logging
+import traceback
 from pathlib import Path
 from typing import Optional
 import click
@@ -245,12 +246,15 @@ def generate_command(
 
     except ConfigurationError as e:
         logger.error(e.message)
+        logger.error(f"Traceback: {traceback.format_exc()}")
         sys.exit(e.exit_code)
     except RepositoryError as e:
         logger.error(e.message)
+        logger.error(f"Traceback: {traceback.format_exc()}")
         sys.exit(e.exit_code)
     except APIError as e:
         logger.error(e.message)
+        logger.error(f"Traceback: {traceback.format_exc()}")
         sys.exit(e.exit_code)
     except KeyboardInterrupt:
         click.echo("\n\nInterrupted by user")

codewiki/src/be/agent_orchestrator.py

@@ -2,6 +2,7 @@
 # import logfire
 import logging
 import os
+import traceback
 from typing import Dict, List, Any
 
 # Configure logging and monitoring
@@ -144,4 +145,5 @@ async def process_module(self, module_name: str, components: Dict[str, Node],
 
         except Exception as e:
             logger.error(f"Error processing module {module_name}: {str(e)}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
             raise

codewiki/src/be/cluster_modules.py

@@ -1,6 +1,7 @@
 from typing import List, Dict, Any
 from collections import defaultdict
 import logging
+import traceback
 logger = logging.getLogger(__name__)
 
 from codewiki.src.be.dependency_analyzer.models.core import Node
@@ -75,6 +76,7 @@ def cluster_modules(
 
     except Exception as e:
         logger.error(f"Failed to parse LLM response: {e}. Response: {response[:200]}...")
+        logger.error(f"Traceback: {traceback.format_exc()}")
         return {}
 
     # check if the module tree is valid

codewiki/src/be/dependency_analyzer/analysis/analysis_service.py

@@ -7,6 +7,7 @@
 """
 
 import logging
+import traceback
 from typing import Dict, List, Optional, Any
 from pathlib import Path
 from codewiki.src.be.dependency_analyzer.utils.security import safe_open_text, assert_safe_path
@@ -212,6 +213,7 @@ def analyze_repository_structure_only(
             if temp_dir:
                 self._cleanup_repository(temp_dir)
             logger.error(f"Structure analysis failed for {github_url}: {str(e)}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
             raise RuntimeError(f"Structure analysis failed: {str(e)}") from e
 
     def _clone_repository(self, github_url: str) -> str:

codewiki/src/be/dependency_analyzer/analysis/call_graph_analyzer.py

@@ -8,6 +8,7 @@
 
 from typing import Dict, List
 import logging
+import traceback
 from pathlib import Path
 from codewiki.src.be.dependency_analyzer.models.core import Node, CallRelationship
 from codewiki.src.be.dependency_analyzer.utils.patterns import CODE_EXTENSIONS
@@ -138,6 +139,7 @@ def _analyze_code_file(self, repo_dir: str, file_info: Dict):
 
         except Exception as e:
             logger.error(f"⚠️ Error analyzing {file_path}: {str(e)}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
 
     def _analyze_python_file(self, file_path: str, content: str, base_dir: str):
         """

codewiki/src/be/dependency_analyzer/analyzers/javascript.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import traceback
 from typing import List, Set, Optional, Tuple
 from pathlib import Path
 import sys
@@ -33,6 +34,7 @@ def __init__(self, file_path: str, content: str, repo_path: str = None):
 
         except Exception as e:
             logger.error(f"Failed to initialize JavaScript parser: {e}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
             self.parser = None
             self.js_language = None
 

codewiki/src/be/dependency_analyzer/analyzers/typescript.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import traceback
 from typing import List, Set, Optional, Tuple
 from pathlib import Path
 import sys
@@ -31,6 +32,7 @@ def __init__(self, file_path: str, content: str, repo_path: str = None):
 
         except Exception as e:
             logger.error(f"Failed to initialize TypeScript parser: {e}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
             self.parser = None
             self.ts_language = None
 

codewiki/src/be/documentation_generator.py

@@ -43,7 +43,7 @@ def create_documentation_metadata(self, working_dir: str, components: Dict[str,
             "generation_info": {
                 "timestamp": datetime.now().isoformat(),
                 "main_model": self.config.main_model,
-                "generator_version": "1.0.0",
+                "generator_version": "1.0.1",
                 "repo_path": self.config.repo_path,
                 "commit_id": self.commit_id
             },
@@ -171,6 +171,7 @@ async def generate_module_documentation(self, components: Dict[str, Any], leaf_n
 
                 except Exception as e:
                     logger.error(f"Failed to process module {module_key}: {str(e)}")
+                    logger.error(f"Traceback: {traceback.format_exc()}")
                     continue
 
             # Generate repo overview
@@ -242,6 +243,7 @@ async def generate_parent_module_docs(self, module_path: List[str],
 
         except Exception as e:
             logger.error(f"Error generating parent documentation for {module_name}: {str(e)}")
+            logger.error(f"Traceback: {traceback.format_exc()}")
             raise
 
     async def run(self) -> None:

codewiki/src/be/main.py

@@ -11,6 +11,7 @@
 import logging
 import argparse
 import asyncio
+import traceback
 
 # Configure logging and monitoring
 from codewiki.src.be.dependency_analyzer.utils.logging_config import setup_logging
@@ -57,6 +58,7 @@ async def main() -> None:
         logger.debug("Documentation generation interrupted by user")
     except Exception as e:
         logger.error(f"Unexpected error: {str(e)}")
+        logger.error(f"Traceback: {traceback.format_exc()}")
         raise
 
 

codewiki/src/be/utils.py

@@ -3,6 +3,7 @@
 from typing import List, Tuple
 import logging
 import tiktoken
+import traceback
 
 
 logger = logging.getLogger(__name__)
@@ -169,6 +170,7 @@ async def validate_single_diagram(diagram_content: str, diagram_num: int, line_s
                 core_error = core_error
             else:
                 logger.error(f"No match found for error pattern, fallback to mermaid-py\n{error_str}")
+                logger.error(f"Traceback: {traceback.format_exc()}")
                 raise Exception(error_str)
 
     except Exception as e: