xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-06-11 00:06:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: create knwoledge base issue

Browse Source
This commit is contained in:
WangCham
2025-07-11 18:14:03 +08:00
parent 815cdf8b4a
commit 14c161b733
2 changed files with 27 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
pkg/entity/persistence/rag.py
View File

@@ -1,19 +1,17 @@
from sqlalchemy import create_engine, Column, Integer, String, Text, DateTime, LargeBinary
from sqlalchemy import create_engine, Column, String, Text, DateTime, LargeBinary, Integer
from sqlalchemy.orm import declarative_base, sessionmaker
from datetime import datetime
import os
Base = declarative_base()
DATABASE_URL = os.getenv('DATABASE_URL', 'sqlite:///./rag_knowledge.db')
print("Using database URL:", DATABASE_URL)
engine = create_engine(DATABASE_URL, connect_args={'check_same_thread': False})
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
def create_db_and_tables():
"""Creates all database tables defined in the Base."""
Base.metadata.create_all(bind=engine)
@@ -22,35 +20,31 @@ def create_db_and_tables():
class KnowledgeBase(Base):
__tablename__ = 'kb'
id = Column(Integer, primary_key=True, index=True)
id = Column(String, primary_key=True, index=True)
name = Column(String, index=True)
description = Column(Text)
created_at = Column(DateTime, default=datetime.utcnow)
embedding_model_uuid = Column(String, default='')
top_k = Column(Integer, default=5)
class File(Base):
__tablename__ = 'file'
id = Column(Integer, primary_key=True, index=True)
kb_id = Column(Integer, nullable=True)
id = Column(String, primary_key=True, index=True)
kb_id = Column(String, nullable=True)
file_name = Column(String)
path = Column(String)
created_at = Column(DateTime, default=datetime.utcnow)
file_type = Column(String)
status = Column(Integer, default=0)
status = Column(String, default='0')
class Chunk(Base):
__tablename__ = 'chunks'
id = Column(Integer, primary_key=True, index=True)
file_id = Column(Integer, nullable=True)
id = Column(String, primary_key=True, index=True)
file_id = Column(String, nullable=True)
text = Column(Text)
class Vector(Base):
__tablename__ = 'vectors'
id = Column(Integer, primary_key=True, index=True)
chunk_id = Column(Integer, nullable=True)
id = Column(String, primary_key=True, index=True)
chunk_id = Column(String, nullable=True)
embedding = Column(LargeBinary)