Spaces:
Sleeping
Sleeping
File size: 2,948 Bytes
02d6bde ec6867d 3973360 02d6bde ec6867d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 |
from pydantic import BaseModel, Field, EmailStr
# from typing import Optional
# from datetime import datetime, timezone
# class BaseDocument(BaseModel):
# created_at: Optional[datetime] = Field(
# default_factory=lambda: datetime.now(timezone.utc)
# )
# updated_at: Optional[datetime] = Field(
# default_factory=lambda: datetime.now(timezone.utc)
# )
# class Config:
# arbitrary_types_allowed = True
# class Activity(BaseDocument):
# id: Optional[str] = Field("", description="Activity's id")
# user_id: str = Field("", description="User's id")
# activity_category: str = Field("", description="Activity's category")
# description: str = Field("", description="Activity's description")
# start_time: Optional[datetime] = Field("", description="Activity's start time")
# end_time: Optional[datetime] = Field("", description="Activity's end time")
# class BookHotel(BaseDocument):
# hotel_name: str = Field("", description="Hotel's name")
# address: str = Field("", description="Hotel's address")
# phone_number: str = Field("", description="Hotel's phone number")
# hotel_email: EmailStr = Field("", description="Hotel's email")
# start_time: Optional[datetime] = Field("", description="Start time of the booking")
# end_time: Optional[datetime] = Field("", description="End time of the booking")
# rating: str = Field("", description="Hotel's rating")
# website: str = Field("", description="Hotel's website")
# class Config:
# json_schema_extra = {
# "example": {
# "hotel_name": "Blue Lagoon Resort",
# "address": "123 Beachside Blvd, Paradise City, Island Nation 54321",
# "phone_number": "+1234567890",
# "hotel_email": "baohtqe170017@fpt.edu.vn",
# "start_time": "2025-01-05T14:00:00.000+00:00",
# "end_time": "2025-01-10T11:00:00.000+00:00",
# "rating": "4.5",
# "website": "https://www.bluelagoonresort.com",
# }
# }
class NearlyDestinationRecommendation(BaseModel):
query: str = Field(
...,
title="Query related to wanting to go somewhere",
description="Auto extracted from user's message. Using Vietnamese language for better results.",
)
places: str = Field(..., title="Places", description="place mention in query")
kinds: str = Field(
...,
title="Kinds",
description="""
Kinds of places you want to find choose one of options:
["architecture", "historic", "cultural", "natural", "sport", "religion", "accomodations", "foods", "shops", "other"]""",
)
class Config:
json_schema_extra = {
"example": {"query": "Địa điểm gần Hà Nội" "places" "Hà Nội"}
}
|