feat: Add Chat endpoint for AI conversation

src/api/agents/chat/router.py

@@ -0,0 +1,123 @@
+"""
+Chat Agent Router - API Endpoints for Chat functionality
+"""
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel, Field
+from typing import List, Optional, Dict, Any
+from src.services.groq_service import generate_response
+from src.api.core.logger import logger
+
+router = APIRouter()
+
+
+class ChatMessage(BaseModel):
+    """Chat message model"""
+    role: str = Field(..., description="Role: 'user' or 'assistant'")
+    content: str = Field(..., description="Message content")
+
+
+class ChatRequest(BaseModel):
+    """Request model for chat"""
+    message: str = Field(..., description="User message")
+    context: Optional[Dict[str, Any]] = Field(default=None, description="Additional context")
+    conversation_id: Optional[str] = Field(default=None, description="Conversation ID for history")
+
+
+class ChatResponse(BaseModel):
+    """Response model for chat"""
+    response: str
+    conversation_id: Optional[str] = None
+    model: str = "llama-3.1-8b-instant"
+
+
+# Simple in-memory conversation storage
+conversations: Dict[str, List[Dict[str, str]]] = {}
+
+
+@router.post("/chat", response_model=ChatResponse)
+async def chat(request: ChatRequest) -> ChatResponse:
+    """
+    Chat endpoint - AI conversation
+    
+    - **message**: User's message
+    - **context**: Optional context data
+    - **conversation_id**: Optional conversation ID for history
+    """
+    try:
+        logger.info(f"Chat request: {request.message[:50]}...")
+        
+        # Get or create conversation
+        conv_id = request.conversation_id or "default"
+        if conv_id not in conversations:
+            conversations[conv_id] = []
+        
+        # Build messages with history
+        messages = [
+            {
+                "role": "system",
+                "content": """You are AmkyawDev AI Assistant, a helpful and friendly AI assistant. 
+You can help with coding, answering questions, and various tasks.
+Always be helpful, polite, and provide accurate information."""
+            }
+        ]
+        
+        # Add conversation history
+        messages.extend(conversations[conv_id][-10:])  # Keep last 10 messages
+        
+        # Add current message
+        messages.append({"role": "user", "content": request.message})
+        
+        # Generate response
+        response_text = await generate_response(
+            messages=messages,
+            model="llama-3.1-8b-instant",
+            temperature=0.7,
+            max_tokens=2048
+        )
+        
+        # Save to conversation history
+        conversations[conv_id].append({"role": "user", "content": request.message})
+        conversations[conv_id].append({"role": "assistant", "content": response_text})
+        
+        # Keep history limited
+        if len(conversations[conv_id]) > 20:
+            conversations[conv_id] = conversations[conv_id][-20:]
+        
+        return ChatResponse(
+            response=response_text,
+            conversation_id=conv_id,
+            model="llama-3.1-8b-instant"
+        )
+        
+    except Exception as e:
+        logger.error(f"Chat error: {str(e)}")
+        raise HTTPException(status_code=500, detail=str(e))
+
+
+@router.get("/conversations")
+async def get_conversations() -> Dict[str, Any]:
+    """Get all conversation IDs"""
+    return {
+        "conversations": list(conversations.keys()),
+        "count": len(conversations)
+    }
+
+
+@router.delete("/conversations/{conversation_id}")
+async def delete_conversation(conversation_id: str) -> Dict[str, str]:
+    """Delete a conversation"""
+    if conversation_id in conversations:
+        del conversations[conversation_id]
+        return {"status": "deleted", "conversation_id": conversation_id}
+    raise HTTPException(status_code=404, detail="Conversation not found")
+
+
+@router.get("/conversations/{conversation_id}")
+async def get_conversation(conversation_id: str) -> Dict[str, Any]:
+    """Get conversation history"""
+    if conversation_id in conversations:
+        return {
+            "conversation_id": conversation_id,
+            "messages": conversations[conversation_id]
+        }
+    raise HTTPException(status_code=404, detail="Conversation not found")