fix(db): Correct database path and chat history logic

This commit is contained in:
PromptEngineer 2025-07-12 01:51:57 -07:00
parent 2421514f3e
commit c93b8639ab
3 changed files with 6 additions and 6 deletions

View File

@ -5,7 +5,7 @@ from datetime import datetime
from typing import List, Dict, Optional, Tuple
class ChatDatabase:
def __init__(self, db_path: str = "chat_history.db"):
def __init__(self, db_path: str = "backend/chat_data.db"):
self.db_path = db_path
self.init_database()

View File

@ -289,13 +289,13 @@ class ChatHandler(http.server.BaseHTTPRequestHandler):
self.send_json_response({"error": "Message is required"}, status_code=400)
return
# Add user message to database first
user_message_id = db.add_message(session_id, message, "user")
if session['message_count'] == 0:
title = generate_session_title(message)
db.update_session_title(session_id, title)
# Add user message to database first
user_message_id = db.add_message(session_id, message, "user")
# 🎯 SMART ROUTING: Decide between direct LLM vs RAG
idx_ids = db.get_indexes_for_session(session_id)
force_rag = bool(data.get("force_rag", False))

View File

@ -17,7 +17,7 @@ from rag_system.main import get_agent
from rag_system.factory import get_indexing_pipeline
# Initialize database connection once at module level
db = ChatDatabase("backend/chat_history.db")
db = ChatDatabase("backend/chat_data.db")
# Get the desired agent mode from environment variables, defaulting to 'default'
# This allows us to easily switch between 'default', 'fast', 'react', etc.