Merge pull request #74 from Shaunwei/pcui/add-postgres

Add postgres support
Shaunwei · Jul 16, 2023 · 99148f0 · 99148f0
2 parents a4c8abf + dc16c80
commit 99148f0
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -11,10 +11,10 @@ COPY ./ /realtime_ai_character
 # Install Python dependencies
 RUN pip install -r requirements.txt
 
-# Build the application
-RUN alembic upgrade head
-
 EXPOSE 8000
 
+# Make the entrypoint script executable
+RUN chmod +x /realtime_ai_character/entrypoint.sh
+
 # Run the application
-CMD ["uvicorn", "realtime_ai_character.main:app", "--host", "0.0.0.0"]
+CMD ["/bin/sh", "/realtime_ai_character/entrypoint.sh"]
diff --git a/entrypoint.sh b/entrypoint.sh
@@ -0,0 +1,4 @@
+#!/bin/sh
+set -e
+alembic upgrade head
+uvicorn realtime_ai_character.main:app --host 0.0.0.0
diff --git a/realtime_ai_character/database/connection.py b/realtime_ai_character/database/connection.py
@@ -7,8 +7,11 @@
 
 SQLALCHEMY_DATABASE_URL = os.getenv("DATABASE_URL")
 
+connect_args = {"check_same_thread": False} if SQLALCHEMY_DATABASE_URL.startswith(
+    "sqlite") else {}
+
 engine = create_engine(
-    SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False}
+    SQLALCHEMY_DATABASE_URL, connect_args=connect_args
 )
 
 SessionLocal = sessionmaker(

diff --git a/requirements.txt b/requirements.txt
@@ -25,3 +25,4 @@ soundfile
 click
 google-cloud-speech
 anthropic
+psycopg2-binary