Spaces:
Running
Running
File size: 1,113 Bytes
0739c8b 9700f95 0739c8b 9700f95 0739c8b 9700f95 |
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 |
# src/models/rag.py
from dataclasses import dataclass
from typing import List, Optional, Dict
from datetime import datetime
@dataclass
class Message:
"""Single message in a conversation"""
role: str # 'user' or 'assistant'
content: str
timestamp: datetime
@dataclass
class ConversationContext:
"""Conversation context with history"""
messages: List[Message]
max_messages: int = 10
def add_message(self, role: str, content: str):
"""Add a message while maintaining max size"""
self.messages.append(Message(
role=role,
content=content,
timestamp=datetime.now()
))
# Keep only the most recent messages
if len(self.messages) > self.max_messages:
self.messages = self.messages[-self.max_messages:]
@dataclass
class RAGResponse:
"""Enhanced RAG response with conversation context"""
response: str
context_docs: Optional[List[str]] = None
sources: Optional[List[Dict]] = None
scores: Optional[List[float]] = None
conversation_context: Optional[ConversationContext] = None |