File size: 3,361 Bytes
9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 5889992 9111274 |
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 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 |
from fastapi import APIRouter, HTTPException, BackgroundTasks
from typing import List, Optional
from datetime import datetime
from src.llm.models.schemas import ConversationResponse, SessionData
from src.llm.utils.logging import TheryBotLogger
from src.llm.memory.history import RedisHistory
from src.llm.memory.memory_manager import RedisMemoryManager
from src.llm.memory.session_manager import SessionManager
from src.llm.agents.conversation_agent import ConversationAgent
router = APIRouter(
prefix="/api/v1",
tags=["TheryAI Core Services"],
responses={500: {"description": "Internal Server Error"}}
)
# Initialize core components
session_manager = SessionManager()
memory_manager = RedisMemoryManager()
history = RedisHistory()
logger = TheryBotLogger()
conversation_agent = ConversationAgent()
@router.post("/users", response_model=dict)
async def create_user():
"""Create a new user ID"""
try:
user_id, _ = session_manager.generate_ids()
return {"user_id": user_id}
except Exception as e:
logger.error(f"User creation failed: {str(e)}")
raise HTTPException(500, "User creation failed")
@router.post("/sessions", response_model=SessionData)
async def create_session(user_id: str):
"""Create a new session ID for a user"""
try:
_, session_id = session_manager.generate_ids(existing_user_id=user_id)
return SessionData(
user_id=user_id,
session_id=session_id,
is_new_user=False,
is_new_session=True
)
except Exception as e:
logger.error(f"Session creation failed: {str(e)}")
raise HTTPException(500, "Session creation failed")
@router.get("/sessions/{session_id}/messages", response_model=List[ConversationResponse])
async def get_messages(session_id: str, limit: int = 50):
"""Get message history for a session"""
try:
if not session_manager.validate_session(session_id):
raise HTTPException(404, "Session not found")
messages = history.get_conversation_history(session_id, limit=limit)
return [msg["response"] for msg in messages]
except Exception as e:
logger.error(f"Message retrieval failed: {str(e)}")
raise HTTPException(500, "Message retrieval failed")
@router.post("/sessions/{session_id}/messages", response_model=ConversationResponse)
async def create_message(
session_id: str,
message: str,
background_tasks: BackgroundTasks
):
"""Process and store a new message"""
try:
user_id = session_manager.validate_session(session_id)
if not user_id:
raise HTTPException(404, "Invalid session")
response = await conversation_agent.process_async(
query=message,
session_data=SessionData(
user_id=user_id,
session_id=session_id,
is_new_user=False,
is_new_session=False
)
)
background_tasks.add_task(
memory_manager.store_conversation,
session_id,
datetime.now().isoformat(),
response
)
return response
except Exception as e:
logger.error(f"Message processing failed: {str(e)}")
raise HTTPException(500, "Message processing failed") |