45 lines
1.9 KiB
Python
45 lines
1.9 KiB
Python
|
from dataclasses import field, dataclass
|
||
|
|
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class ProjectContext:
|
||
|
|
"""Static, per-project knowledge injected into every decomposition."""
|
||
|
|
project_key: str # e.g. "ARCH"
|
||
|
|
language: str = ""
|
||
|
|
backend_framework: str = "" # "FastAPI", "Django", "Spring Boot"
|
||
|
|
frontend_framework: str = "" # "React + Vite", "Vue 3"
|
||
|
|
ui_library: str = "" # "shadcn/ui", "MUI", "Ant Design"
|
||
|
|
database: str = "" # "PostgreSQL 16 via SQLAlchemy 2"
|
||
|
|
orm: str = ""
|
||
|
|
test_framework: str = "" # "pytest + httpx.AsyncClient"
|
||
|
|
package_manager: str = "" # "uv", "poetry", "npm"
|
||
|
|
auth: str = "" # "JWT via fastapi-users"
|
||
|
|
deployment: str = "" # "Docker Compose on Hetzner"
|
||
|
|
conventions: list[str] = field(default_factory=list) # free-form notes
|
||
|
|
extra: dict[str, str] = field(default_factory=dict) # anything else
|
||
|
|
|
||
|
|
def to_prompt_text(self) -> str:
|
||
|
|
lines = [f"## Project Context ({self.project_key})"]
|
||
|
|
fields = [
|
||
|
|
("Language", self.language),
|
||
|
|
("Backend framework", self.backend_framework),
|
||
|
|
("Frontend framework", self.frontend_framework),
|
||
|
|
("UI library", self.ui_library),
|
||
|
|
("Database", self.database),
|
||
|
|
("ORM", self.orm),
|
||
|
|
("Test framework", self.test_framework),
|
||
|
|
("Package manager", self.package_manager),
|
||
|
|
("Auth", self.auth),
|
||
|
|
("Deployment", self.deployment),
|
||
|
|
]
|
||
|
|
for label, value in fields:
|
||
|
|
if value:
|
||
|
|
lines.append(f"- {label}: {value}")
|
||
|
|
for k, v in self.extra.items():
|
||
|
|
if v:
|
||
|
|
lines.append(f"- {k}: {v}")
|
||
|
|
if self.conventions:
|
||
|
|
lines.append("")
|
||
|
|
lines.append("Conventions:")
|
||
|
|
lines.extend(f"- {c}" for c in self.conventions)
|
||
|
|
return "\n".join(lines)
|