openweights 0.7.2__tar.gz → 0.7.3__tar.gz

{openweights-0.7.2 → openweights-0.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openweights
-Version: 0.7.2
+Version: 0.7.3
 Summary: An openai-like sdk for finetuning and batch inference
 License: MIT
 License-File: LICENSE.txt

{openweights-0.7.2 → openweights-0.7.3}/openweights/client/__init__.py

@@ -98,13 +98,22 @@ def register(name: str):
     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 @@ class OpenWeights:
             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-0.7.2 → openweights-0.7.3}/openweights/dashboard/backend/database.py

@@ -25,15 +25,20 @@ from utils import (
     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 = _SUPABASE_URL
+        self.supabase_anon_key = _SUPABASE_ANON_KEY
         self.auth_token = auth_token
         self._current_org_id = None
 
+        if not self.supabase_url or not self.supabase_anon_key:
+            raise ValueError("SUPABASE_URL and SUPABASE_ANON_KEY must be set")
+
         # Initialize regular client for normal operations
         self.client = create_client(self.supabase_url, self.supabase_anon_key)
         if auth_token:
@@ -45,7 +50,9 @@ class Database:
         if not self._current_org_id:
             raise ValueError("No organization set. Call set_organization_id first.")
         return OpenWeights(
-            auth_token=self.auth_token,
+            self.supabase_url,
+            self.supabase_anon_key,
+            self.auth_token,
             organization_id=self._current_org_id,
         )
 

{openweights-0.7.2 → openweights-0.7.3}/openweights/dashboard/backend/main.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-0.7.2 → openweights-0.7.3}/pyproject.toml

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

openweights-0.7.2/.github/workflows/ci.yaml

@@ -1,99 +0,0 @@
-name: "ci"
-on: # rebuild any PRs and main branch changes
-  pull_request:
-  push:
-    branches:
-      - main
-
-concurrency:
-  group: ${{ github.repository_owner }}-${{ github.repository }}-${{ github.ref }}
-  cancel-in-progress: true
-
-env:
-  FORCE_COLOR: "1"
-  PYTHONUNBUFFERED: "1"
-
-jobs:
-  build:
-    runs-on: ubuntu-latest
-    permissions:
-      contents: read
-      packages: write
-
-    strategy:
-      matrix:
-        python-version: ["3.10", "3.11", "3.12"]
-
-    steps:
-      - name: Checkout repository
-        uses: actions/checkout@v4
-      - name: Setup PDM
-        uses: pdm-project/setup-pdm@v4
-        with:
-          cache: false # project doesn't use pdm lock files
-          python-version: ${{ matrix.python-version }}
-      - name: Setup UV
-        uses: astral-sh/setup-uv@v5
-      - name: Configure PDM to use UV
-        run: pdm config use_uv true
-      - name: Cache Huggingface assets
-        uses: actions/cache@v4
-        with:
-          key: huggingface-${{ runner.os }}-${{ matrix.python-version }}-${{ hashFiles('**/pyproject.toml') }}
-          path: ~/.cache/huggingface
-          restore-keys: |
-            huggingface-${{ runner.os }}-${{ matrix.python-version }}-
-      - name: Install dependencies
-        run: pdm install --group dev --no-lock
-
-      # TODO: re-enable when we have pre-commit hooks
-      # - name: Install pre-commit hooks
-      #   run: pdm run pre-commit install
-      # - name: Ensure pre-commit hooks pass
-      #   run: pdm run pre-commit run --all-files
-
-      # TODO: tests don't currently pass due to missing torch dependency
-      # see: https://github.com/longtermrisk/openweights/actions/runs/13210898278/job/36883965796?pr=11#step:8:29
-
-      # - name: Run tests
-      #   env:
-      #     HF_TOKEN: ${{ secrets.HF_TOKEN }}
-      #   run: pdm run python -m pytest tests
-
-  release:
-    needs: build
-    environment: pypi
-    permissions:
-      contents: write
-      id-token: write
-    # https://github.community/t/how-do-i-specify-job-dependency-running-in-another-workflow/16482
-    if: github.event_name == 'push' && github.ref == 'refs/heads/main' && !contains(github.event.head_commit.message, 'chore')
-    runs-on: ubuntu-latest
-    concurrency: release
-    steps:
-      - uses: actions/checkout@v4
-        with:
-          fetch-depth: 0
-      - uses: pdm-project/setup-pdm@v4
-        with:
-          cache: true
-          python-version: "3.12"
-      - name: build
-        run: pdm build
-      - name: Semantic Release
-        id: release
-        uses: python-semantic-release/python-semantic-release@v8.0.7
-        with:
-          github_token: ${{ secrets.GITHUB_TOKEN }}
-
-      # - name: Publish package distributions to PyPI
-      #   uses: pypa/gh-action-pypi-publish@release/v1
-      #   if: steps.release.outputs.released == 'true'
-      - name: Publish package distributions to GitHub Releases
-        uses: python-semantic-release/upload-to-gh-release@main
-        if: steps.release.outputs.released == 'true'
-        with:
-          github_token: ${{ secrets.GITHUB_TOKEN }}
-
-      # TODO: build and deploy docs to gh-pages
-      # do this as part of the semantic release step to ensure version is bumped