7000pctAUTO/codechunk-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

fix: resolve CI/CD issues - Poetry setup, type annotations, MyPy errors
Some checks failed
CI / build (push) Has been cancelled
Details
CI / test (push) Has been cancelled
Details

Browse Source
This commit is contained in:
7000pctAUTO
2026-02-02 00:08:11 +00:00
parent 8186d226f2
commit 275119db73
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
codechunk/config.py
View File

@@ -1,15 +1,15 @@
from pathlib import Path from pathlib import Path
from typing import List, Optional, Dict, Any from typing import Any, Optional
from dataclasses import dataclass, field from dataclasses import dataclass, field
import yaml import yaml
@dataclass @dataclass
class ChunkingConfig: class ChunkingConfig:
include_patterns: List[str] = field(default_factory=lambda: [ include_patterns: list[str] = field(default_factory=lambda: [
"*.py", "*.js", "*.ts", "*.go", "*.rs", "*.java", "*.cpp", "*.c", "*.h" "*.py", "*.js", "*.ts", "*.go", "*.rs", "*.java", "*.cpp", "*.c", "*.h"
]) ])
exclude_patterns: List[str] = field(default_factory=lambda: [ exclude_patterns: list[str] = field(default_factory=lambda: [
"**/test_*.py", "**/__pycache__/**", "**/node_modules/**", "**/test_*.py", "**/__pycache__/**", "**/node_modules/**",
"**/.git/**", "**/venv/**", "**/.env/**" "**/.git/**", "**/venv/**", "**/.env/**"
]) ])
@@ -17,7 +17,7 @@ class ChunkingConfig:
min_chunk_size: int = 3 min_chunk_size: int = 3
preserve_docstrings: bool = True preserve_docstrings: bool = True
remove_comments: bool = False remove_comments: bool = False
boilerplate_patterns: List[str] = field(default_factory=lambda: [ boilerplate_patterns: list[str] = field(default_factory=lambda: [
r"@property\s*\n\s*def\s+\w+\s*\(\s*\)\s*:", r"@property\s*\n\s*def\s+\w+\s*\(\s*\)\s*:",
r"@abstractmethod", r"@abstractmethod",
r"@staticmethod", r"@staticmethod",
@@ -27,14 +27,14 @@ class ChunkingConfig:
@dataclass @dataclass
class PrioritizationConfig: class PrioritizationConfig:
keywords: List[str] = field(default_factory=lambda: [ keywords: list[str] = field(default_factory=lambda: [
"main", "core", "handler", "controller", "service", "model" "main", "core", "handler", "controller", "service", "model"
]) ])
size_limit: int = 10000 size_limit: int = 10000
exclude_patterns: List[str] = field(default_factory=lambda: [ exclude_patterns: list[str] = field(default_factory=lambda: [
"**/test_*.py", "**/*_test.py", "**/conftest.py" "**/test_*.py", "**/*_test.py", "**/conftest.py"
]) ])
include_only: List[str] = field(default_factory=list) include_only: list[str] = field(default_factory=list)
weight_by_depth: bool = True weight_by_depth: bool = True
@@ -51,21 +51,21 @@ class Config:
chunking: ChunkingConfig = field(default_factory=ChunkingConfig) chunking: ChunkingConfig = field(default_factory=ChunkingConfig)
prioritization: PrioritizationConfig = field(default_factory=PrioritizationConfig) prioritization: PrioritizationConfig = field(default_factory=PrioritizationConfig)
output: OutputConfig = field(default_factory=OutputConfig) output: OutputConfig = field(default_factory=OutputConfig)
env_overrides: Dict[str, str] = field(default_factory=dict) env_overrides: dict[str, str] = field(default_factory=dict)
def load_config(config_path: Optional[str] = None) -> Config: def load_config(config_path: Optional[str] = None) -> Config:
"""Load configuration from YAML file.""" """Load configuration from YAML file."""
if config_path is None: if config_path is None:
config_path = Path.cwd() / ".codechunk.yaml" config_file = Path.cwd() / ".codechunk.yaml"
else:
config_file = Path(config_path) config_file = Path(config_path)
if not config_file.exists(): if not config_file.exists():
return Config() return Config()
try: try:
with open(config_file, 'r') as f: with open(config_file) as f:
data = yaml.safe_load(f) data = yaml.safe_load(f)
if data is None: if data is None: