Spaces:
Running
Running
File size: 680 Bytes
3a07545 c3cf389 3a07545 c3cf389 3a07545 |
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 |
"""
Pydantic models for the fitness agent.
"""
from pydantic import BaseModel, Field
from typing import Optional, List
from datetime import datetime, date
class AgentResponse(BaseModel):
"""Standard agent response format."""
content: str
metadata: Optional[dict] = None
class ConversationMessage(BaseModel):
"""Individual conversation message."""
role: str # "user" or "assistant"
content: str
timestamp: Optional[str] = None
class AgentConfig(BaseModel):
"""Configuration for the fitness agent."""
model_name: str
temperature: Optional[float] = 0.7
max_tokens: Optional[int] = None
custom_instructions: Optional[str] = None
|