kuzakhmetovartur
a649fb1192
-добавлены нетворки в докер композ -исправлен рутинг (баг пайчарма) -запросы к ии агентам не проходят из-за ссл сертификата (пробовали отключить, но пока не выходит, нужно доделать)
78 lines
2.4 KiB
Python
78 lines
2.4 KiB
Python
from pydantic import BaseModel, Field
|
|
from typing import Optional, List, Dict, Any
|
|
from datetime import datetime
|
|
|
|
|
|
class ChatRequest(BaseModel):
|
|
message: str = Field(..., min_length=1, max_length=2000)
|
|
conversation_id: Optional[str] = None
|
|
|
|
|
|
class ChatResponse(BaseModel):
|
|
response: str
|
|
conversation_id: str
|
|
tokens_used: Optional[int] = None
|
|
model: Optional[str] = None
|
|
|
|
|
|
class ScheduleGenerateRequest(BaseModel):
|
|
child_age: int = Field(..., ge=1, le=18)
|
|
preferences: List[str] = Field(default_factory=list)
|
|
date: str # ISO format date string
|
|
description: Optional[str] = None
|
|
|
|
|
|
class ScheduleGenerateResponse(BaseModel):
|
|
schedule_id: str
|
|
title: str
|
|
tasks: List[Dict[str, Any]]
|
|
tokens_used: Optional[int] = None
|
|
|
|
|
|
class ConversationHistory(BaseModel):
|
|
conversation_id: str
|
|
messages: List[Dict[str, Any]]
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ConversationListItem(BaseModel):
|
|
"""Элемент списка разговоров"""
|
|
conversation_id: str
|
|
last_message: Optional[str] = None
|
|
created_at: datetime
|
|
updated_at: datetime
|
|
message_count: int = 0
|
|
|
|
class Config:
|
|
from_attributes = True
|
|
|
|
|
|
class ScheduleUpdateRequest(BaseModel):
|
|
"""Запрос на обновление расписания через ИИ"""
|
|
user_request: str = Field(..., min_length=1, max_length=1000, description="Описание желаемых изменений")
|
|
|
|
|
|
class ScheduleUpdateResponse(BaseModel):
|
|
"""Ответ после обновления расписания"""
|
|
schedule_id: str
|
|
title: str
|
|
tasks: List[Dict[str, Any]]
|
|
tokens_used: Optional[int] = None
|
|
|
|
|
|
class RecommendationRequest(BaseModel):
|
|
"""Запрос на получение рекомендаций"""
|
|
preferences: List[str] = Field(default_factory=list, description="Предпочтения пользователя")
|
|
category: Optional[str] = Field(None, description="Категория заданий")
|
|
completed_tasks: Optional[List[str]] = Field(default_factory=list, description="Уже выполненные задания")
|
|
top_k: int = Field(5, ge=1, le=20, description="Количество рекомендаций")
|
|
|
|
|
|
class RecommendationResponse(BaseModel):
|
|
"""Ответ с рекомендациями"""
|
|
recommendations: List[Dict[str, Any]]
|
|
total: int |