services/vibe-bot
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

fix db location

Browse Source
This commit is contained in:
Reese Wells
2026-03-03 10:30:41 -05:00
parent b44cdfbeb3
commit 493b6b8425
2 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
database.py
View File

@@ -1,5 +1,4 @@
import sqlite3 import sqlite3
import json
import os import os
from typing import Optional, List, Tuple from typing import Optional, List, Tuple
from datetime import datetime from datetime import datetime
@@ -7,7 +6,7 @@ import numpy as np
from openai import OpenAI from openai import OpenAI
# Database configuration # 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_MODEL = os.getenv("EMBEDDING_MODEL", "qwen3-embed-4b")
EMBEDDING_DIMENSION = 2048 # Default for qwen3-embed-4b EMBEDDING_DIMENSION = 2048 # Default for qwen3-embed-4b
MAX_HISTORY_MESSAGES = int(os.getenv("MAX_HISTORY_MESSAGES", "1000")) MAX_HISTORY_MESSAGES = int(os.getenv("MAX_HISTORY_MESSAGES", "1000"))
@@ -234,11 +233,13 @@ class ChatDatabase:
cursor = conn.cursor() cursor = conn.cursor()
# Join chat_messages and message_embeddings to get content and embeddings # Join chat_messages and message_embeddings to get content and embeddings
cursor.execute(""" cursor.execute(
"""
SELECT cm.message_id, cm.content, me.embedding SELECT cm.message_id, cm.content, me.embedding
FROM chat_messages cm FROM chat_messages cm
JOIN message_embeddings me ON cm.message_id = me.message_id JOIN message_embeddings me ON cm.message_id = me.message_id
""") """
)
rows = cursor.fetchall() rows = cursor.fetchall()
results = [] results = []

2
vibe-bot.container
View File

@@ -4,8 +4,10 @@ After=network.target
[Container] [Container]
Image=localhost/vibe-bot:latest Image=localhost/vibe-bot:latest
Volume=vibe-bot-db:/db
Environment=DISCORD_TOKEN=your_token_here Environment=DISCORD_TOKEN=your_token_here
Environment=OPENAI_API_ENDPOINT=http://host.containers.internal:8000 Environment=OPENAI_API_ENDPOINT=http://host.containers.internal:8000
Environment=DB_PATH=/db/chat_history.db
[Service] [Service]
Restart=on-failure Restart=on-failure