diff --git a/postgres.py b/postgres.py index d751168..ae5e209 100644 --- a/postgres.py +++ b/postgres.py @@ -99,48 +99,3 @@ def get_similar_chunks_with_distance( ) for r in res ] - - -def get_random_chunks(limit=5) -> List[Chunk]: - cur = get_connection().cursor() - cur.execute( - """ - SELECT c.id, c.page_id, p.title, p.description, c.text - FROM chunks c - JOIN pages p ON c.page_id = p.id - ORDER BY RANDOM() - LIMIT %s; - """, - (limit,), - ) - res = cur.fetchall() - cur.close() - - return [ - Chunk(id=r[0], pageId=r[1], title=r[2], description=r[3], text=r[4]) - for r in res - ] - - -def get_pages(limit=10) -> List[Page]: - cur = get_connection().cursor() - cur.execute( - "SELECT id, title, description, text FROM pages LIMIT %s;", - (limit,), - ) - res = cur.fetchall() - cur.close() - - return [Page(id=r[0], title=r[1], description=r[2], text=r[3]) for r in res] - - -def get_random_pages(limit=10) -> List[Page]: - cur = get_connection().cursor() - cur.execute( - "SELECT id, title, description, text FROM pages ORDER BY RANDOM() LIMIT %s;", - (limit,), - ) - res = cur.fetchall() - cur.close() - - return [Page(id=r[0], title=r[1], description=r[2], text=r[3]) for r in res]