2023-11-27 16:01:20 -08:00
|
|
|
import datetime
|
2024-08-22 14:11:23 -07:00
|
|
|
from typing import List, Optional
|
2023-11-27 16:01:20 -08:00
|
|
|
|
2024-07-28 15:21:51 -07:00
|
|
|
from pydantic import BaseModel, ConfigDict
|
2023-11-27 16:01:20 -08:00
|
|
|
|
|
|
|
from ..models.api_call_status import ApiCallStatus
|
2024-08-22 14:11:23 -07:00
|
|
|
from ..models.ml_feedback import MlFeedback
|
|
|
|
from ..models.source_range_prompt import SourceRangePrompt
|
|
|
|
from ..models.text_to_cad_model import TextToCadModel
|
2023-11-27 16:01:20 -08:00
|
|
|
from ..models.uuid import Uuid
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
class TextToCadIteration(BaseModel):
|
|
|
|
"""A response from a text to CAD iteration."""
|
|
|
|
|
|
|
|
code: str
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
completed_at: Optional[datetime.datetime] = None
|
|
|
|
|
|
|
|
created_at: datetime.datetime
|
|
|
|
|
|
|
|
error: Optional[str] = None
|
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
feedback: Optional[MlFeedback] = None
|
2023-11-28 23:50:50 -08:00
|
|
|
|
2023-11-29 10:41:21 -08:00
|
|
|
id: Uuid
|
2023-11-28 23:50:50 -08:00
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
model: TextToCadModel
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
model_version: str
|
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
original_source_code: str
|
|
|
|
|
|
|
|
prompt: Optional[str] = None
|
2023-11-28 23:50:50 -08:00
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
source_ranges: List[SourceRangePrompt]
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
started_at: Optional[datetime.datetime] = None
|
|
|
|
|
|
|
|
status: ApiCallStatus
|
|
|
|
|
|
|
|
updated_at: datetime.datetime
|
|
|
|
|
|
|
|
user_id: Uuid
|
2024-01-06 18:32:21 -08:00
|
|
|
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|