feat: split connection

app.py

@@ -1,7 +1,6 @@
 import json
 import os
 
-import sqlalchemy
 import streamlit as st
 import streamlit.components.v1 as components
 from langchain.callbacks import get_openai_callback
@@ -14,8 +13,7 @@ from chat_history import insert_chat_history, insert_chat_history_articles
 from css import load_css
 from custom_pgvector import CustomPGVector
 from message import Message
-
-CONNECTION_STRING = "postgresql+psycopg2://postgres@/sorbobot?host=0.0.0.0"
+from connection import connect
 
 st.set_page_config(layout="wide")
 
@@ -23,14 +21,6 @@ st.title("Sorbobot - Le futur de la recherche scientifique interactive")
 
 chat_column, doc_column = st.columns([2, 1])
 
-
-def connect() -> sqlalchemy.engine.Connection:
-    engine = sqlalchemy.create_engine(CONNECTION_STRING, pool_pre_ping=True)
-
-    conn = engine.connect()
-    return conn
-
-
 conn = connect()
 
 

connection.py

@@ -0,0 +1,11 @@
+import sqlalchemy
+
+
+CONNECTION_STRING = "postgresql+psycopg2://postgres@/sorbobot?host=0.0.0.0"
+
+
+def connect() -> sqlalchemy.engine.Connection:
+    engine = sqlalchemy.create_engine(CONNECTION_STRING, pool_pre_ping=True)
+
+    conn = engine.connect()
+    return conn