weiweiw
/
Langchain-Chatchat
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update table_name and column_name in SQL

This commit is contained in:
imClumsyPanda 2023-08-05 21:02:14 +08:00
parent cd0f2a2c41
commit f68e432570
1 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
server/knowledge_base/knowledge_base.py
View File

@ -32,16 +32,16 @@ def add_kb_to_db(kb_name, vs_type, embed_model):
conn = sqlite3.connect(DB_ROOT) conn = sqlite3.connect(DB_ROOT)
c = conn.cursor() c = conn.cursor()
# Create table # Create table
c.execute('''CREATE TABLE if not exists KNOWLEDGE_BASE c.execute('''CREATE TABLE if not exists knowledge_base
(ID INTEGER PRIMARY KEY AUTOINCREMENT, (id INTEGER PRIMARY KEY AUTOINCREMENT,
KB_NAME TEXT, kb_name TEXT,
VS_TYPE TEXT, vs_type TEXT,
EMBED_MODEL TEXT, embed_model TEXT,
FILE_COUNT INTEGER, file_count INTEGER,
CREATE_TIME DATETIME) ''') create_time DATETIME) ''')
# Insert a row of data # Insert a row of data
c.execute(f"""INSERT INTO KNOWLEDGE_BASE c.execute(f"""INSERT INTO knowledge_base
(KB_NAME, VS_TYPE, EMBED_MODEL, FILE_COUNT, CREATE_TIME) (kb_name, vs_type, embed_model, file_count, create_time)
VALUES VALUES
('{kb_name}','{vs_type}','{embed_model}', ('{kb_name}','{vs_type}','{embed_model}',
0,'{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}')""") 0,'{datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")}')""")
@ -53,8 +53,8 @@ def kb_exists(kb_name):
conn = sqlite3.connect(DB_ROOT) conn = sqlite3.connect(DB_ROOT)
c = conn.cursor() c = conn.cursor()
c.execute(f'''SELECT COUNT(*) c.execute(f'''SELECT COUNT(*)
FROM KNOWLEDGE_BASE FROM knowledge_base
WHERE KB_NAME="{kb_name}" ''') WHERE kb_name="{kb_name}" ''')
status = True if c.fetchone()[0] else False status = True if c.fetchone()[0] else False
conn.commit() conn.commit()
conn.close() conn.close()
@ -64,9 +64,9 @@ def kb_exists(kb_name):
def load_kb_from_db(kb_name): def load_kb_from_db(kb_name):
conn = sqlite3.connect(DB_ROOT) conn = sqlite3.connect(DB_ROOT)
c = conn.cursor() c = conn.cursor()
c.execute(f'''SELECT KB_NAME, VS_TYPE, EMBED_MODEL c.execute(f'''SELECT kb_name, vs_type, embed_model
FROM KNOWLEDGE_BASE FROM knowledge_base
WHERE KB_NAME="{kb_name}" ''') WHERE kb_name="{kb_name}" ''')
resp = c.fetchone() resp = c.fetchone()
if resp: if resp:
kb_name, vs_type, embed_model = resp kb_name, vs_type, embed_model = resp
@ -81,8 +81,8 @@ def delete_kb_from_db(kb_name):
conn = sqlite3.connect(DB_ROOT) conn = sqlite3.connect(DB_ROOT)
c = conn.cursor() c = conn.cursor()
c.execute(f'''DELETE c.execute(f'''DELETE
FROM KNOWLEDGE_BASE FROM knowledge_base
WHERE KB_NAME="{kb_name}" ''') WHERE kb_name="{kb_name}" ''')
conn.commit() conn.commit()
conn.close() conn.close()
return True return True