rocmport/models.py

from pydantic import BaseModel
from typing import List, Optional

class AnalysisResult(BaseModel):
    has_kernels: bool = False
    cuda_apis: List[str] = []
    risk_level: str = "unknown"
    raw: str = ""

class TranslationResult(BaseModel):
    code: str
    warnings: List[str] = []

class CompileResult(BaseModel):
    success: bool
    errors: List[str] = []
    stdout: str = ""

class PipelineResult(Bas ÿÿ eModel):
    original: str
    migrated: str
    analysis: Optional[AnalysisResult] = None
    compile:  Optional[CompileResult]  = None
    score:    float = 0.0