fix db location
This commit is contained in:
@@ -1,5 +1,4 @@
|
||||
import sqlite3
|
||||
import json
|
||||
import os
|
||||
from typing import Optional, List, Tuple
|
||||
from datetime import datetime
|
||||
@@ -7,7 +6,7 @@ import numpy as np
|
||||
from openai import OpenAI
|
||||
|
||||
# Database configuration
|
||||
DB_PATH = os.getenv("CHAT_DB_PATH", "chat_history.db")
|
||||
DB_PATH = os.getenv("DB_PATH", "chat_history.db")
|
||||
EMBEDDING_MODEL = os.getenv("EMBEDDING_MODEL", "qwen3-embed-4b")
|
||||
EMBEDDING_DIMENSION = 2048 # Default for qwen3-embed-4b
|
||||
MAX_HISTORY_MESSAGES = int(os.getenv("MAX_HISTORY_MESSAGES", "1000"))
|
||||
@@ -234,11 +233,13 @@ class ChatDatabase:
|
||||
cursor = conn.cursor()
|
||||
|
||||
# Join chat_messages and message_embeddings to get content and embeddings
|
||||
cursor.execute("""
|
||||
cursor.execute(
|
||||
"""
|
||||
SELECT cm.message_id, cm.content, me.embedding
|
||||
FROM chat_messages cm
|
||||
JOIN message_embeddings me ON cm.message_id = me.message_id
|
||||
""")
|
||||
"""
|
||||
)
|
||||
rows = cursor.fetchall()
|
||||
|
||||
results = []
|
||||
|
||||
Reference in New Issue
Block a user