Files
instructor/examples/evals/models.py
T
2023-09-11 20:16:10 -04:00

25 lines
507 B
Python

from typing import List, Optional
from pydantic import BaseModel, Field
from enum import Enum
class SourceType(str, Enum):
CRM = "CRM"
WEB = "WEB"
EMAIL = "EMAIL"
SOCIAL_MEDIA = "SOCIAL_MEDIA"
OTHER = "OTHER"
class Search(BaseModel):
query: str
source_type: SourceType
results_limit: Optional[int] = Field(10)
is_priority: Optional[bool] = None
tags: Optional[List[str]] = None
class MultiSearch(BaseModel):
queries: List[Search]
user_id: Optional[str]