File size: 1,269 Bytes
5889992 |
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 |
from pydantic import BaseModel, Field
from typing import Dict, Any, Optional, List
class EmotionalAnalysis(BaseModel):
primary_emotion: str
intensity: int = Field(..., ge=1, le=10)
secondary_emotions: List[str]
triggers: List[str]
coping_strategies: List[str] = []
confidence_score: float = Field(..., ge=0, le=1)
class ContextInfo(BaseModel):
query: str
web_context: str = ""
vector_context: List[str] = Field(default_factory=list)
combined_context: str = ""
class SessionData(BaseModel):
user_id: str = Field(..., description="Unique user identifier")
session_id: str = Field(..., description="Current session identifier")
is_new_user: bool = Field(False, description="Flag for new user detection")
is_new_session: bool = Field(False, description="Flag for new session detection")
class ConversationResponse(BaseModel):
session_data: SessionData
response: str = Field(..., description="Primary assistant response")
emotion_analysis: EmotionalAnalysis
context: ContextInfo = Field(default_factory=ContextInfo)
query: str
safety_level: str = Field("unknown", description="Assessment of response safety") # Default value
suggested_resources: List[str] = Field(default_factory=list) |