automate database creation

sunholo-data · Mar 10, 2024 · 841e6fb · 841e6fb
1 parent 8864ba2
commit 841e6fb
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/config/llm_config.yaml b/config/llm_config.yaml
@@ -32,7 +32,7 @@ edmonbrain_vertex:
     region: europe-west1
     cluster: cluster_name
     instance: instance_name
-    database: database_name
+    database: database_name # if not included will default to 'db_<project_id>'
 #   ip_type: PUBLIC or PRIVATE
 codey:
   llm: codey

diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup, find_packages
 
 # Define your base version
-version = '0.21.22'
+version = '0.21.23'
 
 setup(
     name='sunholo',

diff --git a/sunholo/components/vectorstore.py b/sunholo/components/vectorstore.py
@@ -91,7 +91,7 @@ def pick_vectorstore(vs_str, vector_name, embeddings):
             region=alloydb_config["region"],
             cluster=alloydb_config["cluster"],
             instance=alloydb_config["instance"],
-            database=alloydb_config["database"],
+            database=alloydb_config.get("database") or f'db_{alloydb_config["project_id"]}',
             ip_type=alloydb_config.get("ip_type") or IPTypes.PRIVATE
         )