Class EnhancedRagApiHandler

rag_system/api_server_with_progress.py:238–411

Enhanced API handler with progress tracking support

        raise

class EnhancedRagApiHandler(http.server.BaseHTTPRequestHandler):
    """Enhanced API handler with progress tracking support"""
    
    def do_OPTIONS(self):
        """Handle CORS preflight requests for frontend integration."""
        self.send_response(200)
        self.send_header('Access-Control-Allow-Origin', '*')
        self.send_header('Access-Control-Allow-Methods', 'POST, GET, OPTIONS')
        self.send_header('Access-Control-Allow-Headers', 'Content-Type')
        self.end_headers()

    def do_GET(self):
        """Handle GET requests for progress status and SSE streams"""
        parsed_path = urlparse(self.path)
        
        if parsed_path.path == '/progress':
            self.handle_progress_status()
        elif parsed_path.path == '/stream':
            self.handle_progress_stream()
        else:
            self.send_json_response({"error": "Not Found"}, status_code=404)

    def do_POST(self):
        """Handle POST requests for chat and indexing."""
        parsed_path = urlparse(self.path)

        if parsed_path.path == '/chat':
            self.handle_chat()
        elif parsed_path.path == '/index':
            self.handle_index_with_progress()
        else:
            self.send_json_response({"error": "Not Found"}, status_code=404)

    def handle_chat(self):
        """Handles a chat query by calling the agentic RAG pipeline."""
        try:
            content_length = int(self.headers['Content-Length'])
            post_data = self.rfile.read(content_length)
            data = json.loads(post_data.decode('utf-8'))
            
            query = data.get('query')
            if not query:
                self.send_json_response({"error": "Query is required"}, status_code=400)
                return

            # Use the single, persistent agent instance to run the query
            result = RAG_AGENT.run(query)
            
            # The result is a dict, so we need to dump it to a JSON string
            self.send_json_response(result)

        except json.JSONDecodeError:
            self.send_json_response({"error": "Invalid JSON"}, status_code=400)
        except Exception as e:
            self.send_json_response({"error": f"Server error: {str(e)}"}, status_code=500)

    def handle_index_with_progress(self):
        """Triggers the document indexing pipeline with real-time progress tracking."""