administrator
f9dc61e32c
PostgreSQL-backed knowledge store with hybrid search: - Full-text search via tsvector (always available) - Semantic reranking via Voyage AI embeddings + Python cosine similarity - Tables: kb_memories, kb_sessions, kb_messages - Tools: store_memory, store_conversation, search, get_context, get_recent, stats Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Inicializace Knowledgebase databáze.
|
|
Vytvoří databázi 'knowledgebase' a aplikuje schema.sql.
|
|
|
|
Spustit jednou:
|
|
python setup_db.py
|
|
"""
|
|
import os
|
|
import sys
|
|
import psycopg
|
|
from psycopg.rows import dict_row
|
|
from pathlib import Path
|
|
|
|
PG_HOST = os.getenv("PG_HOST", "192.168.1.76")
|
|
PG_PORT = int(os.getenv("PG_PORT", "5432"))
|
|
PG_USER = os.getenv("PG_USER", "vladimir.buzalka")
|
|
PG_PASSWORD = os.getenv("PG_PASSWORD", "Vlado7309208104++")
|
|
|
|
SCHEMA_FILE = Path(__file__).parent / "schema.sql"
|
|
|
|
|
|
def main():
|
|
# ── Připoj se k postgres a vytvoř DB ──
|
|
print("Connecting to postgres...")
|
|
with psycopg.connect(
|
|
host=PG_HOST, port=PG_PORT,
|
|
user=PG_USER, password=PG_PASSWORD,
|
|
dbname="postgres",
|
|
autocommit=True,
|
|
) as conn:
|
|
exists = conn.execute(
|
|
"SELECT 1 FROM pg_database WHERE datname = 'knowledgebase'"
|
|
).fetchone()
|
|
if not exists:
|
|
conn.execute("CREATE DATABASE knowledgebase")
|
|
print("Created database 'knowledgebase'")
|
|
else:
|
|
print("Database 'knowledgebase' already exists")
|
|
|
|
# ── Aplikuj schema ──
|
|
print("Applying schema...")
|
|
schema = SCHEMA_FILE.read_text(encoding="utf-8")
|
|
with psycopg.connect(
|
|
host=PG_HOST, port=PG_PORT,
|
|
user=PG_USER, password=PG_PASSWORD,
|
|
dbname="knowledgebase",
|
|
autocommit=True,
|
|
) as conn:
|
|
conn.execute(schema)
|
|
print("Schema applied successfully.")
|
|
print()
|
|
print("Done! Knowledgebase database is ready.")
|
|
print()
|
|
print("Next steps:")
|
|
print(" 1. pip install -r requirements.txt")
|
|
print(" 2. Optional: set VOYAGE_API_KEY=...")
|
|
print(" 3. Add server to Claude Code settings (see below)")
|
|
print()
|
|
print(' "knowledgebase": {')
|
|
print(' "command": "python",')
|
|
print(' "args": ["U:/janssen/Knowledgebase/server.py"],')
|
|
print(' "env": {"VOYAGE_API_KEY": "..."}')
|
|
print(' }')
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|