Skip to content

Commit

Permalink
Cleanup - remove old db models
Browse files Browse the repository at this point in the history
  • Loading branch information
aybruhm committed Jan 10, 2024
1 parent 873fb19 commit 3ca00ad
Showing 1 changed file with 0 additions and 67 deletions.
67 changes: 0 additions & 67 deletions agenta-backend/agenta_backend/models/db_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -360,70 +360,3 @@ class TraceDB(Document):

class Settings:
name = "traces"


class OldEvaluationTypeSettings(BaseModel):
similarity_threshold: Optional[float]
regex_pattern: Optional[str]
regex_should_match: Optional[bool]
webhook_url: Optional[str]
llm_app_prompt_template: Optional[str]
custom_code_evaluation_id: Optional[str]
evaluation_prompt_template: Optional[str]


class OldEvaluationScenarioInput(BaseModel):
input_name: str
input_value: str


class OldEvaluationScenarioOutput(BaseModel):
variant_id: str
variant_output: str


class OldEvaluationDB(Document):
app: Link[AppDB]
organization: Link[OrganizationDB]
user: Link[UserDB]
status: str
evaluation_type: str
evaluation_type_settings: OldEvaluationTypeSettings
variants: List[PydanticObjectId]
testset: Link[TestSetDB]
created_at: Optional[datetime] = Field(default=datetime.utcnow())
updated_at: Optional[datetime] = Field(default=datetime.utcnow())

class Settings:
name = "evaluations"


class OldEvaluationScenarioDB(Document):
user: Link[UserDB]
organization: Link[OrganizationDB]
evaluation: Link[EvaluationDB]
inputs: List[OldEvaluationScenarioInput]
outputs: List[OldEvaluationScenarioOutput] # EvaluationScenarioOutput
vote: Optional[str]
score: Optional[Any]
correct_answer: Optional[str]
created_at: Optional[datetime] = Field(default=datetime.utcnow())
updated_at: Optional[datetime] = Field(default=datetime.utcnow())
is_pinned: Optional[bool]
note: Optional[str]

class Settings:
name = "evaluation_scenarios"


class OldCustomEvaluationDB(Document):
evaluation_name: str
python_code: str
app: Link[AppDB]
user: Link[UserDB]
organization: Link[OrganizationDB]
created_at: Optional[datetime] = Field(default=datetime.utcnow())
updated_at: Optional[datetime] = Field(default=datetime.utcnow())

class Settings:
name = "custom_evaluations"

0 comments on commit 3ca00ad

Please sign in to comment.