fyp

app/ai/agent/reply_intelligence.py

@@ -0,0 +1,73 @@
+"""
+Intelligent reply detection for AIDA.
+Determines which message AIDA should quote when responding.
+"""
+
+from typing import Optional, Dict, Any, List
+
+
+def determine_reply_to_message(
+    conversation_history: List[Dict[str, Any]],
+    reply_context: Optional[Dict[str, Any]] = None
+) -> Optional[str]:
+    """
+    Analyze conversation and determine which message AIDA should quote.
+
+    Strategy:
+    1. If user just replied to a specific message (reply_context exists),
+       quote the LATEST USER MESSAGE, not the message they swiped
+    2. Otherwise, quote the most recent user message
+
+    Args:
+        conversation_history: List of messages with structure:
+            [{"id": str, "sender_id": str, "content": str, "is_ai": bool, ...}, ...]
+        reply_context: Optional context of message user replied to
+
+    Returns:
+        Message ID to quote, or None if no appropriate message found
+    """
+
+    if not conversation_history:
+        return None
+
+    # Find the most recent user message (not AI)
+    latest_user_message = None
+    for msg in conversation_history:
+        if not msg.get("is_ai", False):
+            latest_user_message = msg
+            break  # conversation_history is sorted newest first
+
+    if latest_user_message:
+        return latest_user_message.get("id")
+
+    return None
+
+
+def build_reply_context_for_response(
+    conversation_history: List[Dict[str, Any]],
+    reply_to_id: str
+) -> Optional[Dict[str, Any]]:
+    """
+    Build reply context for AIDA's response.
+
+    Returns:
+        {
+            "id": message_id,
+            "sender_name": sender name,
+            "content": message content
+        }
+    """
+
+    if not reply_to_id:
+        return None
+
+    # Find the message to quote
+    for msg in conversation_history:
+        if msg.get("id") == reply_to_id:
+            return {
+                "id": msg.get("id"),
+                "sender_name": msg.get("sender_name", "User"),
+                "content": msg.get("content", ""),
+            }
+
+    return None

app/ai/agent/schemas.py

@@ -172,6 +172,10 @@ class AgentResponse(BaseModel):
     my_listings: Optional[List[Dict[str, Any]]] = None  # For user's own listings
     tool_result: Optional[ToolResult] = None
     error: Optional[str] = None
+    # Intelligent reply context
+    reply_to_id: Optional[str] = None
+    replied_to_content: Optional[str] = None
+    replied_to_sender: Optional[str] = None
     metadata: Dict[str, Any] = Field(default_factory=dict)
 
 

app/ai/routes/chat.py

@@ -8,7 +8,7 @@ See app/ai/services/agent_executor.py for execution logic.
 
 from fastapi import APIRouter, Request, Response
 from pydantic import BaseModel
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, Any, List
 from structlog import get_logger
 from uuid import uuid4
 from datetime import datetime
@@ -38,6 +38,7 @@ class AskBody(BaseModel):
     is_voice_message: Optional[bool] = False  # True if message came from voice input
     source: Optional[str] = "default"  # "dm", "search_bar", etc.
     reply_context: Optional[Dict[str, Any]] = None  # Context of message user is replying to
+    conversation_history: Optional[List[Dict[str, Any]]] = None  # Full conversation for intelligent replies
 
 
 
@@ -88,6 +89,7 @@ async def ask_ai(request: Request, response: Response, body: AskBody) -> AgentRe
         is_voice_message=body.is_voice_message or False,
         source=body.source or "default",
         reply_context=body.reply_context,
+        conversation_history=body.conversation_history,
     )
     
     logger.info(

app/ai/services/agent_executor.py

@@ -21,7 +21,7 @@ import json
 import hashlib
 import logging
 from uuid import uuid4
-from typing import Optional, Dict, Any
+from typing import Optional, Dict, Any, List
 from dataclasses import dataclass
 
 from app.ai.agent.graph import get_aida_graph
@@ -47,6 +47,7 @@ class AgentInput:
     is_voice_message: bool = False
     source: str = "default"
     reply_context: Optional[Dict[str, Any]] = None
+    conversation_history: Optional[List[Dict[str, Any]]] = None
     
     def to_dict(self) -> Dict[str, Any]:
         """Convert to dict for graph invocation"""
@@ -63,12 +64,14 @@ class AgentInput:
             "source": self.source,
             "temp_data": {"reply_context": self.reply_context} if self.reply_context else {},
         }
-        
+
         if self.start_new_session:
             result["conversation_history"] = []
             result["provided_fields"] = {}
             result["missing_required_fields"] = []
-            
+        elif self.conversation_history:
+            result["conversation_history"] = self.conversation_history
+
         return result
     
     @property
@@ -222,17 +225,38 @@ class AgentExecutor:
     def _extract_response(self, state: Dict[str, Any], input: AgentInput) -> AgentResponse:
         """Extract clean response from graph state"""
         temp_data = state.get("temp_data", {})
-        
+
         # Check for pre-built response
         if final_response := temp_data.get("final_response"):
             return final_response
-        
+
         # Build response from state
         response_text = temp_data.get("response_text") or "I'm here to help! What would you like to do?"
-        
+
         current_flow = state.get("current_flow")
         flow_str = current_flow.value if hasattr(current_flow, 'value') else str(current_flow)
-        
+
+        # Determine intelligent reply
+        reply_to_id = None
+        reply_to_context = None
+
+        if input.conversation_history:
+            from app.ai.agent.reply_intelligence import (
+                determine_reply_to_message,
+                build_reply_context_for_response
+            )
+
+            reply_to_id = determine_reply_to_message(
+                conversation_history=input.conversation_history,
+                reply_context=input.reply_context
+            )
+
+            if reply_to_id:
+                reply_to_context = build_reply_context_for_response(
+                    conversation_history=input.conversation_history,
+                    reply_to_id=reply_to_id
+                )
+
         return AgentResponse(
             success=state.get("last_error") is None,
             text=response_text,
@@ -244,7 +268,12 @@ class AgentExecutor:
             },
             draft=temp_data.get("draft"),
             draft_ui=temp_data.get("draft_ui"),
+            search_results=temp_data.get("search_results"),
+            my_listings=temp_data.get("my_listings"),
             error=state.get("last_error"),
+            reply_to_id=reply_to_id,
+            replied_to_content=reply_to_context["content"] if reply_to_context else None,
+            replied_to_sender=reply_to_context["sender_name"] if reply_to_context else None,
             metadata={
                 "intent": state.get("intent_type"),
                 "intent_confidence": state.get("intent_confidence", 0),