поправил правила хранения промтов
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
This commit is contained in:
@@ -118,8 +118,7 @@ def create_table_config_gpt():
|
|||||||
with conn.cursor() as cur:
|
with conn.cursor() as cur:
|
||||||
cur.execute("""
|
cur.execute("""
|
||||||
CREATE TABLE IF NOT EXISTS config_gpt (
|
CREATE TABLE IF NOT EXISTS config_gpt (
|
||||||
url TEXT PRIMARY KEY,
|
name VARCHAR(20) PRIMARY KEY,
|
||||||
name VARCHAR(20),
|
|
||||||
promt TEXT
|
promt TEXT
|
||||||
);
|
);
|
||||||
""")
|
""")
|
||||||
@@ -127,17 +126,16 @@ def create_table_config_gpt():
|
|||||||
finally:
|
finally:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
def update_promt(url: str, name: str, promt: str):
|
def update_promt(name: str, promt: str):
|
||||||
conn = get_connection()
|
conn = get_connection()
|
||||||
try:
|
try:
|
||||||
with conn.cursor() as cur:
|
with conn.cursor() as cur:
|
||||||
cur.execute("""
|
cur.execute("""
|
||||||
INSERT INTO config_gpt (url, name, promt)
|
INSERT INTO config_gpt (url, name, promt)
|
||||||
VALUES (%s, %s, %s)
|
VALUES ( %s, %s)
|
||||||
ON CONFLICT (url) DO UPDATE SET
|
ON CONFLICT (name) DO UPDATE SET
|
||||||
name = EXCLUDED.name,
|
|
||||||
promt = EXCLUDED.promt
|
promt = EXCLUDED.promt
|
||||||
""", (url, name, promt))
|
""", (name, promt))
|
||||||
conn.commit()
|
conn.commit()
|
||||||
finally:
|
finally:
|
||||||
pass
|
pass
|
||||||
@@ -146,7 +144,7 @@ def get_promt(promt_name_url):
|
|||||||
conn = get_connection()
|
conn = get_connection()
|
||||||
try:
|
try:
|
||||||
with conn.cursor(cursor_factory=RealDictCursor) as cur:
|
with conn.cursor(cursor_factory=RealDictCursor) as cur:
|
||||||
cur.execute("SELECT promt FROM config_gpt WHERE url = %s", (promt_name_url,))
|
cur.execute("SELECT promt FROM config_gpt WHERE name = %s", (promt_name_url,))
|
||||||
promt = cur.fetchone()
|
promt = cur.fetchone()
|
||||||
return promt['promt']
|
return promt['promt']
|
||||||
finally:
|
finally:
|
||||||
@@ -160,7 +158,7 @@ def get_all_promt():
|
|||||||
cur.execute("SELECT * FROM config_gpt")
|
cur.execute("SELECT * FROM config_gpt")
|
||||||
rows = cur.fetchall()
|
rows = cur.fetchall()
|
||||||
|
|
||||||
sources = [{"url": row["url"], "name": row["name"], "promt": row["promt"]} for row in rows]
|
sources = [{"name": row["name"], "promt": row["promt"]} for row in rows]
|
||||||
return {"sources": sources}
|
return {"sources": sources}
|
||||||
finally:
|
finally:
|
||||||
pass
|
pass
|
||||||
|
|||||||
Reference in New Issue
Block a user