diff --git a/scripts/api.py b/scripts/api.py index 2c29c9e..c156a79 100644 --- a/scripts/api.py +++ b/scripts/api.py @@ -56,7 +56,7 @@ DEFAULT_SETTINGS = { } print("Loading Aaron AI...") -PG_DSN = os.getenv("PG_DSN", "dbname=aaronai user=aaronai password=aaronai_db_password host=localhost") +PG_DSN = os.getenv("PG_DSN") def get_pg(): return psycopg2.connect(PG_DSN) diff --git a/scripts/dream.py b/scripts/dream.py index 235eaae..dd537e6 100644 --- a/scripts/dream.py +++ b/scripts/dream.py @@ -23,7 +23,7 @@ import psycopg2 load_dotenv(Path.home() / "aaronai" / ".env") -PG_DSN = os.getenv("PG_DSN", "dbname=aaronai user=aaronai password=aaronai_db_password host=localhost") +PG_DSN = os.getenv("PG_DSN") def get_pg(): return psycopg2.connect(PG_DSN) diff --git a/scripts/ingest.py b/scripts/ingest.py index e984bae..383d235 100644 --- a/scripts/ingest.py +++ b/scripts/ingest.py @@ -16,7 +16,7 @@ load_dotenv(Path.home() / "aaronai" / ".env") print("Loading embedding model...") embedder = SentenceTransformer("all-MiniLM-L6-v2") -PG_DSN = os.getenv("PG_DSN", "dbname=aaronai user=aaronai password=aaronai_db_password host=localhost") +PG_DSN = os.getenv("PG_DSN") def get_pg(): return psycopg2.connect(PG_DSN) diff --git a/scripts/ingest_conversations.py b/scripts/ingest_conversations.py index 0e752f7..2bd51cb 100644 --- a/scripts/ingest_conversations.py +++ b/scripts/ingest_conversations.py @@ -15,7 +15,7 @@ from sentence_transformers import SentenceTransformer load_dotenv(Path.home() / "aaronai" / ".env") CONVERSATIONS_DB = str(Path.home() / "aaronai" / "conversations.db") -PG_DSN = os.getenv("PG_DSN", "dbname=aaronai user=aaronai password=aaronai_db_password host=localhost") +PG_DSN = os.getenv("PG_DSN") MIN_EXCHANGES = 3 print("Loading embedding model...") diff --git a/scripts/migrate_to_pgvector.py b/scripts/migrate_to_pgvector.py index 2f570f6..aff92b6 100644 --- a/scripts/migrate_to_pgvector.py +++ b/scripts/migrate_to_pgvector.py @@ -10,7 +10,10 @@ from pathlib import Path from sentence_transformers import SentenceTransformer CHROMA_SQLITE = str(Path.home() / "aaronai" / "db" / "chroma.sqlite3") -PG_DSN = "dbname=aaronai user=aaronai password=aaronai_db_password host=localhost" +import os +PG_DSN = os.getenv("PG_DSN") +if not PG_DSN: + raise RuntimeError("PG_DSN environment variable not set") print("Loading embedding model...") embedder = SentenceTransformer("all-MiniLM-L6-v2")