2023-11-27 16:01:20 -08:00
|
|
|
import datetime
|
2024-08-09 20:45:26 -07:00
|
|
|
from typing import 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.ml_prompt_metadata import MlPromptMetadata
|
|
|
|
from ..models.ml_prompt_type import MlPromptType
|
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 MlPrompt(BaseModel):
|
|
|
|
"""A ML prompt."""
|
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
|
|
|
metadata: Optional[MlPromptMetadata] = None
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
model_version: str
|
|
|
|
|
|
|
|
output_file: Optional[str] = None
|
|
|
|
|
|
|
|
prompt: str
|
|
|
|
|
|
|
|
started_at: Optional[datetime.datetime] = None
|
|
|
|
|
|
|
|
status: ApiCallStatus
|
|
|
|
|
2024-08-22 14:11:23 -07:00
|
|
|
type: MlPromptType
|
2023-11-28 23:50:50 -08:00
|
|
|
|
|
|
|
updated_at: datetime.datetime
|
|
|
|
|
|
|
|
user_id: Uuid
|
2024-01-06 18:32:21 -08:00
|
|
|
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|