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
|
from typing import List, Dict, Optional, Tuple
|
||||||
|
|
||||||
class ChatDatabase:
|
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.db_path = db_path
|
||||||
self.init_database()
|
self.init_database()
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -289,15 +289,15 @@ class ChatHandler(http.server.BaseHTTPRequestHandler):
|
||||||
self.send_json_response({"error": "Message is required"}, status_code=400)
|
self.send_json_response({"error": "Message is required"}, status_code=400)
|
||||||
return
|
return
|
||||||
|
|
||||||
# Add user message to database first
|
|
||||||
user_message_id = db.add_message(session_id, message, "user")
|
|
||||||
|
|
||||||
if session['message_count'] == 0:
|
if session['message_count'] == 0:
|
||||||
title = generate_session_title(message)
|
title = generate_session_title(message)
|
||||||
db.update_session_title(session_id, title)
|
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
|
# 🎯 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))
|
force_rag = bool(data.get("force_rag", False))
|
||||||
use_rag = True if force_rag else self._should_use_rag(message, idx_ids)
|
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
|
from rag_system.factory import get_indexing_pipeline
|
||||||
|
|
||||||
# Initialize database connection once at module level
|
# 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'
|
# Get the desired agent mode from environment variables, defaulting to 'default'
|
||||||
# This allows us to easily switch between 'default', 'fast', 'react', etc.
|
# This allows us to easily switch between 'default', 'fast', 'react', etc.
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user