env_variables

Author: DoyleDev

fastapi/.env.example

@@ -1,19 +1,10 @@
-# Local Lakebase Configuration (legacy - replaced by Lakebase config below)
-# DATABRICKS_DATABASE_INSTANCE=<INSTANCE_NAME>
-# DATABRICKS_DATABASE_NAME=<DATABASE_NAME>
-DATABRICKS_DATABASE_PORT=5432
-
-# Table Configuration
-DEFAULT_POSTGRES_SCHEMA=<SCHEMA_NAME>
-DEFAULT_POSTGRES_TABLE=<TABLE_NAME>
-
 # Lakebase Configuration
 LAKEBASE_INSTANCE_NAME=<LAKEBASE_INSTANCE_NAME> # Name of the Lakebase instance
 LAKEBASE_DATABASE_NAME=<LAKEBASE_DATABASE_NAME> # Name of the Lakebase database
 LAKEBASE_CATALOG_NAME=<LAKEBASE_CATALOG_NAME> # Name of the Lakebase catalog
 SYNCHED_TABLE_STORAGE_CATALOG=<SYNCED_TABLE_CATALOG> # Catalog where you have permissions to create tables. Used to store metadata from the lakebase synced table pipeline.
 SYNCHED_TABLE_STORAGE_SCHEMA=<SYNCED_TABLE_SCHEMA> # Schema where you have permissions to create tables. Used to store metadata from the lakebase synced table pipeline.
-
+DATABRICKS_DATABASE_PORT=5432
 DATABRICKS_HOST=<DATABRICKS_HOST> # e.g., https://adb-1234567890123456.7.azuredatabricks.net
 DATABRICKS_TOKEN=<PAT_TOKEN> # e.g., dapi
 DATABRICKS_USER_NAME=<YOUR_USER_NAME> # e.g., [email protected]

fastapi/app.yaml

@@ -12,11 +12,6 @@ env:
   - name: 'DATABRICKS_DATABASE_PORT'
     value: '5432'
 
-  - name: 'DEFAULT_POSTGRES_SCHEMA'
-    value: 'public'
-  - name: 'DEFAULT_POSTGRES_TABLE'
-    value: 'orders_synced'
-
   - name: 'DB_POOL_SIZE'
     value: '5'
   - name: 'DB_MAX_OVERFLOW'

fastapi/models/orders.py

@@ -1,4 +1,3 @@
-import os
 from datetime import date
 from decimal import Decimal
 from typing import Optional
@@ -18,8 +17,8 @@ class OrderBase(SQLModel):
 
 
 class Order(OrderBase, table=True):
-    __tablename__ = os.getenv("DEFAULT_POSTGRES_TABLE", "orders_synced")
-    __table_args__ = {"schema": os.getenv("DEFAULT_POSTGRES_SCHEMA", "public")}
+    __tablename__ = "orders_synced"
+    __table_args__ = "public"
     o_orderkey: Optional[int] = Field(default=None, primary_key=True)
 
 

fastapi/routes/v1/lakebase.py

@@ -45,8 +45,10 @@ async def create_lakebase_resources(
     ),
 ):
     if create_resources:
-        instance_name = os.getenv("LAKEBASE_INSTANCE_NAME", f"{current_user_id}-lakebase-demo")
-        
+        instance_name = os.getenv(
+            "LAKEBASE_INSTANCE_NAME", f"{current_user_id}-lakebase-demo"
+        )
+
         # Check if instance already exists
         try:
             instance_exists = w.database.get_database_instance(name=instance_name)
@@ -59,12 +61,14 @@ async def create_lakebase_resources(
             )
         except Exception as e:
             if "not found" in str(e).lower() or "resource not found" in str(e).lower():
-                logger.info(f"Instance {instance_name} does not exist. Proceeding with creation.")
+                logger.info(
+                    f"Instance {instance_name} does not exist. Proceeding with creation."
+                )
             else:
                 logger.error(f"Error checking instance existence: {e}")
                 raise HTTPException(
-                    status_code=500, 
-                    detail=f"Error checking instance existence: {str(e)}"
+                    status_code=500,
+                    detail=f"Error checking instance existence: {str(e)}",
                 )
 
         lakebase_database_name = os.getenv("LAKEBASE_DATABASE_NAME", "demo_database")
@@ -101,12 +105,15 @@ async def create_lakebase_resources(
         logger.info(f"Creating superuser role for: {superuser_role.name}")
         try:
             created_role = w.database.create_database_instance_role(
-                instance_name=instance_create.name, database_instance_role=superuser_role
+                instance_name=instance_create.name,
+                database_instance_role=superuser_role,
             )
             logger.info(f"Successfully created superuser role: {created_role.name}")
         except Exception as e:
             logger.error(f"Failed to create superuser role (continuing anyway): {e}")
-            logger.info("Database instance is still functional - role can be created manually if needed")
+            logger.info(
+                "Database instance is still functional - role can be created manually if needed"
+            )
 
         catalog = DatabaseCatalog(
             name=catalog_name,
@@ -145,17 +152,18 @@ async def create_lakebase_resources(
             logger.info(f"Initiated sync pipeline creation: {synced_table_create.id}")
             pipeline_id = synced_table_create.id
         except Exception as e:
-            logger.error(f"API error during synced table creation (pipeline likely created anyway): {e}")
+            logger.error(
+                f"API error during synced table creation (pipeline likely created anyway): {e}"
+            )
             pipeline_id = "check-workspace-ui"
-        
-        # Get workspace URL from current workspace client
+
         workspace_url = w.config.host
         if pipeline_id != "check-workspace-ui":
             pipeline_url = f"{workspace_url}/pipelines/{pipeline_id}"
             message = f"Resources created successfully. Synced table pipeline {pipeline_id} is provisioning asynchronously. Monitor progress at: {pipeline_url}"
         else:
             message = f"Resources created successfully. Synced table pipeline initiated (API response error). Check pipelines in workspace: {workspace_url}/pipelines"
-        
+
         return LakebaseResourcesResponse(
             instance=instance_create.name,
             catalog=database_create.name,
@@ -192,7 +200,9 @@ async def delete_lakebase_resources(
             message="No resources were deleted (confirm_deletion=False)",
         )
 
-    instance_name = os.getenv("LAKEBASE_INSTANCE_NAME", f"{current_user_id}-lakebase-demo")
+    instance_name = os.getenv(
+        "LAKEBASE_INSTANCE_NAME", f"{current_user_id}-lakebase-demo"
+    )
     catalog_name = os.getenv("LAKEBASE_CATALOG_NAME", f"{current_user_id}-pg-catalog")
     synced_table_name = f"{catalog_name}.public.orders_synced"