mirror of
https://github.com/zebrajr/localGPT.git
synced 2025-12-06 12:20:53 +01:00
fix(db): Correct database path and chat history logic
This commit is contained in:
parent
2421514f3e
commit
c93b8639ab
|
|
@ -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()
|
||||
|
||||
|
|
|
|||
|
|
@ -289,15 +289,15 @@ 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)
|
||||
idx_ids = db.get_indexes_for_session(session_id)
|
||||
force_rag = bool(data.get("force_rag", False))
|
||||
use_rag = True if force_rag else self._should_use_rag(message, idx_ids)
|
||||
|
||||
|
|
|
|||
|
|
@ -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.
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user