Don't require supabase related envs

Author: nielsrolf

.github/workflows/ci.yaml

@@ -98,13 +98,22 @@ def register_job(cls):
     return register_job
 
 
+_SUPABASE_URL = os.environ.get(
+    "SUPABASE_URL", "https://cmaguqyuzweixkrqjvnf.supabase.co"
+)
+_SUPABASE_ANON_KEY = os.environ.get(
+    "SUPABASE_ANON_KEY",
+    "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImNtYWd1cXl1endlaXhrcnFqdm5mIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NjE2ODA1ODIsImV4cCI6MjA3NzI1NjU4Mn0.SlD0g3sWHsc3_SKEofR6Y6H01oWiEYBlmXYQiw0379s",
+)
+
+
 class OpenWeights:
     _INSTANCES = []
 
     def __init__(
         self,
-        supabase_url: Optional[str] = None,
-        supabase_key: Optional[str] = None,
+        supabase_url: Optional[str] = _SUPABASE_URL,
+        supabase_key: Optional[str] = _SUPABASE_ANON_KEY,
         auth_token: Optional[str] = None,
         organization_id: Optional[str] = None,
         use_async: bool = False,
@@ -118,21 +127,11 @@ def __init__(
             auth_token: Authentication token (or OPENWEIGHTS_API_KEY env var)
                        Can be either a session token or a service account JWT token
         """
-        self.supabase_url = supabase_url or os.environ.get(
-            "SUPABASE_URL", "https://cmaguqyuzweixkrqjvnf.supabase.co"
-        )
-        self.supabase_key = supabase_key or os.environ.get(
-            "SUPABASE_ANON_KEY",
-            "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJzdXBhYmFzZSIsInJlZiI6ImNtYWd1cXl1endlaXhrcnFqdm5mIiwicm9sZSI6ImFub24iLCJpYXQiOjE3NjE2ODA1ODIsImV4cCI6MjA3NzI1NjU4Mn0.SlD0g3sWHsc3_SKEofR6Y6H01oWiEYBlmXYQiw0379s",
-        )
+        self.supabase_url = supabase_url
+        self.supabase_key = supabase_key
         self.auth_token = auth_token or os.getenv("OPENWEIGHTS_API_KEY")
         self.deploy_kwargs = deploy_kwargs
 
-        if not self.supabase_url or not self.supabase_key:
-            raise ValueError(
-                "Supabase URL and key must be provided either as arguments or environment variables"
-            )
-
         if not self.auth_token:
             raise ValueError(
                 "Authentication token must be provided either as argument or OPENWEIGHTS_API_KEY environment variable"

openweights/client/__init__.py

@@ -25,14 +25,14 @@
     validate_organization_secrets,
 )
 
-from openweights import OpenWeights
+from openweights.client import _SUPABASE_ANON_KEY, _SUPABASE_URL, OpenWeights
 from supabase import Client, create_client
 
 
 class Database:
     def __init__(self, auth_token: Optional[str] = None):
-        self.supabase_url = os.getenv("SUPABASE_URL")
-        self.supabase_anon_key = os.getenv("SUPABASE_ANON_KEY")
+        self.supabase_url = _SUPABASE_URL
+        self.supabase_anon_key = _SUPABASE_ANON_KEY
         self.auth_token = auth_token
         self._current_org_id = None
 

openweights/dashboard/backend/database.py

@@ -116,11 +116,8 @@ async def create_organization(
 @app.get("/organizations/", response_model=List[Organization])
 async def get_organizations(db: Database = Depends(get_db)):
     """Get list of organizations the current user has access to."""
-    try:
-        result = db.client.from_("organizations").select("*").execute()
-        return result.data
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    result = db.client.from_("organizations").select("*").execute()
+    return result.data
 
 
 @app.get("/organizations/{organization_id}", response_model=Organization)

openweights/dashboard/backend/main.py

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "openweights"
-version = "0.7.1"
+version = "0.7.3"
 description = "An openai-like sdk for finetuning and batch inference"
 readme = "README.md"
 requires-python = ">=3.11"