cvec 1.5.2__tar.gz → 1.5.3__tar.gz

{cvec-1.5.2 → cvec-1.5.3}/.github/workflows/ci.yml

@@ -27,7 +27,7 @@ jobs:
       - name: Checkout
         uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1
 
-      - uses: CVector-Energy/python-test@uv
+      - uses: CVector-Energy/python-test@b956bb181831f0ddca3496505d5823c678a926e7 # v1.1.0
         with:
           python-version: ${{ matrix.python-version }}
           src-dirs: .

{cvec-1.5.2 → cvec-1.5.3}/.github/workflows/license-check-python.yml

@@ -11,7 +11,7 @@ jobs:
       - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
 
       - name: Check Python licenses
-        uses: CVector-Energy/pyproject-license-check@main
+        uses: CVector-Energy/pyproject-license-check@977092b2a13e0766082dfdfb7c57b67985c675e8 # main
         with:
           app-id: ${{ vars.APP_ID }}
           app-private-key: ${{ secrets.APP_PRIVATE_KEY }}

cvec-1.5.3/.github/workflows/pyproject-license-check.yml

@@ -0,0 +1,34 @@
+name: Python License Check
+
+on:
+  push:
+
+permissions:
+  contents: read
+  id-token: write
+
+jobs:
+  license-check:
+    runs-on: ubuntu-24.04
+
+    steps:
+      - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
+
+      - name: Configure AWS credentials
+        uses: aws-actions/configure-aws-credentials@7474bc4690e29a8392af63c5b98e7449536d5c3a # v4.3.1
+        with:
+          role-to-assume: arn:aws:iam::904233131484:role/CvecCodeArtifactInternalReadOnlyAccess
+          aws-region: us-east-1
+
+      - name: Configure CodeArtifact
+        run: |
+          CODEARTIFACT_AUTH_TOKEN=$(aws codeartifact get-authorization-token --domain cvector --domain-owner 904233131484 --region us-east-1 --query authorizationToken --output text)
+          echo "::add-mask::$CODEARTIFACT_AUTH_TOKEN"
+          echo "UV_INDEX_CVECTOR_INTERNAL_USERNAME=aws" >> $GITHUB_ENV
+          echo "UV_INDEX_CVECTOR_INTERNAL_PASSWORD=${CODEARTIFACT_AUTH_TOKEN}" >> $GITHUB_ENV
+
+      - name: Check Python licenses
+        uses: CVector-Energy/pyproject-license-check@075858d33ce078e040088fbef523db1e0cfd2d63 # v0.3.0
+        with:
+          app-id: ${{ vars.APP_ID }}
+          app-private-key: ${{ secrets.APP_PRIVATE_KEY }}

{cvec-1.5.2 → cvec-1.5.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: cvec
-Version: 1.5.2
+Version: 1.5.3
 Summary: SDK for CVector Energy
 Author-email: CVector <support@cvector.energy>
 License-Expression: MIT

{cvec-1.5.2 → cvec-1.5.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cvec"
-version = "1.5.2"
+version = "1.5.3"
 description = "SDK for CVector Energy"
 authors = [{ name = "CVector", email = "support@cvector.energy" }]
 readme = "README.md"

{cvec-1.5.2 → cvec-1.5.3}/src/cvec/cvec.py

@@ -36,6 +36,7 @@ class CVec:
     host: Optional[str]
     default_start_at: Optional[datetime]
     default_end_at: Optional[datetime]
+    timeout: Optional[float]
     # Supabase authentication
     _access_token: Optional[str]
     _refresh_token: Optional[str]
@@ -49,10 +50,12 @@ class CVec:
         default_start_at: Optional[datetime] = None,
         default_end_at: Optional[datetime] = None,
         api_key: Optional[str] = None,
+        timeout: Optional[float] = 30,
     ) -> None:
         self.host = host or os.environ.get("CVEC_HOST")
         self.default_start_at = default_start_at
         self.default_end_at = default_end_at
+        self.timeout = timeout
 
         # Supabase authentication
         self._access_token = None
@@ -210,7 +213,7 @@ class CVec:
             req = Request(
                 url, data=request_body, headers=request_headers, method=method
             )
-            with urlopen(req) as response:
+            with urlopen(req, timeout=self.timeout) as response:
                 return self._process_response(response, url, method)
 
         try:
@@ -234,7 +237,7 @@ class CVec:
                     req = Request(
                         url, data=request_body, headers=request_headers, method=method
                     )
-                    with urlopen(req) as response:
+                    with urlopen(req, timeout=self.timeout) as response:
                         return self._process_response(response, url, method)
                 except (HTTPError, URLError, ValueError, KeyError) as refresh_error:
                     logger.warning(
@@ -557,7 +560,7 @@ class CVec:
         request_body = json.dumps(payload).encode("utf-8")
         req = Request(supabase_url, data=request_body, headers=headers, method="POST")
 
-        with urlopen(req) as response:
+        with urlopen(req, timeout=self.timeout) as response:
             response_data = response.read()
             data = json.loads(response_data.decode("utf-8"))
 
@@ -586,7 +589,7 @@ class CVec:
         request_body = json.dumps(payload).encode("utf-8")
         req = Request(supabase_url, data=request_body, headers=headers, method="POST")
 
-        with urlopen(req) as response:
+        with urlopen(req, timeout=self.timeout) as response:
             response_data = response.read()
             data = json.loads(response_data.decode("utf-8"))
 
@@ -609,7 +612,7 @@ class CVec:
             config_url = f"{self.host}/config"
             req = Request(config_url, method="GET")
 
-            with urlopen(req) as response:
+            with urlopen(req, timeout=self.timeout) as response:
                 response_data = response.read()
                 config_data = json.loads(response_data.decode("utf-8"))
 
@@ -664,7 +667,7 @@ class CVec:
         def make_rpc_request() -> Any:
             """Inner function to make the actual RPC request."""
             req = Request(url, data=request_body, headers=headers, method="POST")
-            with urlopen(req) as response:
+            with urlopen(req, timeout=self.timeout) as response:
                 response_data = response.read()
                 return json.loads(response_data.decode("utf-8"))
 
@@ -682,7 +685,7 @@ class CVec:
                     req = Request(
                         url, data=request_body, headers=headers, method="POST"
                     )
-                    with urlopen(req) as response:
+                    with urlopen(req, timeout=self.timeout) as response:
                         response_data = response.read()
                         return json.loads(response_data.decode("utf-8"))
                 except (HTTPError, URLError, ValueError, KeyError) as refresh_error:
@@ -731,7 +734,7 @@ class CVec:
         def make_query_request() -> Any:
             """Inner function to make the actual query request."""
             req = Request(url, headers=headers, method="GET")
-            with urlopen(req) as response:
+            with urlopen(req, timeout=self.timeout) as response:
                 response_data = response.read()
                 return json.loads(response_data.decode("utf-8"))
 
@@ -747,7 +750,7 @@ class CVec:
 
                     # Retry the request
                     req = Request(url, headers=headers, method="GET")
-                    with urlopen(req) as response:
+                    with urlopen(req, timeout=self.timeout) as response:
                         response_data = response.read()
                         return json.loads(response_data.decode("utf-8"))
                 except (HTTPError, URLError, ValueError, KeyError) as refresh_error:

{cvec-1.5.2 → cvec-1.5.3}/tests/test_cvec.py

@@ -1,5 +1,8 @@
 import io
 import os
+import socket
+import threading
+import time
 from datetime import datetime
 from typing import Any
 from unittest.mock import patch
@@ -897,3 +900,89 @@ class TestCVecSelectFromEAVId:
         )
 
         assert result == []
+
+
+class TestCVecTimeout:
+    """Test that HTTP requests respect the timeout parameter."""
+
+    @patch.object(CVec, "_login_with_supabase", return_value=None)
+    @patch.object(
+        CVec, "_fetch_config", autospec=True, side_effect=mock_fetch_config_side_effect
+    )
+    def test_default_timeout_is_set(self, mock_fetch_key: Any, mock_login: Any) -> None:
+        """Test that the default timeout is 30 seconds."""
+        client = CVec(
+            host="test_host",
+            api_key="cva_hHs0CbkKALxMnxUdI9hanF0TBPvvvr1HjG6O",
+        )
+        assert client.timeout == 30
+
+    @patch.object(CVec, "_login_with_supabase", return_value=None)
+    @patch.object(
+        CVec, "_fetch_config", autospec=True, side_effect=mock_fetch_config_side_effect
+    )
+    def test_custom_timeout_is_set(self, mock_fetch_key: Any, mock_login: Any) -> None:
+        """Test that a custom timeout can be provided."""
+        client = CVec(
+            host="test_host",
+            api_key="cva_hHs0CbkKALxMnxUdI9hanF0TBPvvvr1HjG6O",
+            timeout=60,
+        )
+        assert client.timeout == 60
+
+    @patch.object(CVec, "_login_with_supabase", return_value=None)
+    @patch.object(
+        CVec, "_fetch_config", autospec=True, side_effect=mock_fetch_config_side_effect
+    )
+    def test_none_timeout_disables_timeout(
+        self, mock_fetch_key: Any, mock_login: Any
+    ) -> None:
+        """Test that timeout=None disables the timeout."""
+        client = CVec(
+            host="test_host",
+            api_key="cva_hHs0CbkKALxMnxUdI9hanF0TBPvvvr1HjG6O",
+            timeout=None,
+        )
+        assert client.timeout is None
+
+    def test_make_request_times_out_on_hung_server(self) -> None:
+        """Test that _make_request raises when the server hangs."""
+        srv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        srv.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+        srv.bind(("127.0.0.1", 0))
+        srv.listen(1)
+        port = srv.getsockname()[1]
+
+        def accept_and_hang() -> None:
+            conn, _ = srv.accept()
+            time.sleep(30)
+            conn.close()
+            srv.close()
+
+        t = threading.Thread(target=accept_and_hang, daemon=True)
+        t.start()
+
+        with (
+            patch.object(CVec, "_login_with_supabase", return_value=None),
+            patch.object(
+                CVec,
+                "_fetch_config",
+                autospec=True,
+                side_effect=mock_fetch_config_side_effect,
+            ),
+        ):
+            client = CVec(
+                host=f"http://127.0.0.1:{port}",
+                api_key="cva_hHs0CbkKALxMnxUdI9hanF0TBPvvvr1HjG6O",
+                timeout=2,
+            )
+            client._access_token = "fake_token"
+
+            start = time.time()
+            with pytest.raises(OSError):
+                client._make_request("GET", "/api/test")
+            elapsed = time.time() - start
+
+            assert elapsed < 5, f"Expected timeout around 2s, took {elapsed:.1f}s"
+
+        srv.close()

{cvec-1.5.2 → cvec-1.5.3}/uv.lock

@@ -399,11 +399,11 @@ wheels = [
 
 [[package]]
 name = "pygments"
-version = "2.19.1"
+version = "2.20.0"
 source = { registry = "https://pypi.org/simple" }
-sdist = { url = "https://files.pythonhosted.org/packages/7c/2d/c3338d48ea6cc0feb8446d8e6937e1408088a72a39937982cc6111d17f84/pygments-2.19.1.tar.gz", hash = "sha256:61c16d2a8576dc0649d9f39e089b5f02bcd27fba10d8fb4dcc28173f7a45151f", size = 4968581, upload-time = "2025-01-06T17:26:30.443Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/c3/b2/bc9c9196916376152d655522fdcebac55e66de6603a76a02bca1b6414f6c/pygments-2.20.0.tar.gz", hash = "sha256:6757cd03768053ff99f3039c1a36d6c0aa0b263438fcab17520b30a303a82b5f", size = 4955991, upload-time = "2026-03-29T13:29:33.898Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/8a/0b/9fcc47d19c48b59121088dd6da2488a49d5f72dacf8262e2790a1d2c7d15/pygments-2.19.1-py3-none-any.whl", hash = "sha256:9ea1544ad55cecf4b8242fab6dd35a93bbce657034b0611ee383099054ab6d8c", size = 1225293, upload-time = "2025-01-06T17:26:25.553Z" },
+    { url = "https://files.pythonhosted.org/packages/f4/7e/a72dd26f3b0f4f2bf1dd8923c85f7ceb43172af56d63c7383eb62b332364/pygments-2.20.0-py3-none-any.whl", hash = "sha256:81a9e26dd42fd28a23a2d169d86d7ac03b46e2f8b59ed4698fb4785f946d0176", size = 1231151, upload-time = "2026-03-29T13:29:30.038Z" },
 ]
 
 [[package]]

cvec-1.5.2/.claude/settings.local.json

@@ -1,28 +0,0 @@
-{
-  "permissions": {
-    "allow": [
-      "Bash(chmod:*)",
-      "Bash(python:*)",
-      "Bash(git commit:*)",
-      "Bash(pytest:*)",
-      "Bash(python3 -m pytest:*)",
-      "Bash(poetry run pytest:*)",
-      "Bash(git add:*)",
-      "Bash(poetry run mypy:*)",
-      "Bash(git checkout:*)",
-      "Bash(poetry run ruff:*)",
-      "Bash(poetry install:*)",
-      "Bash(tree:*)",
-      "Bash(git mv:*)",
-      "WebFetch(domain:github.com)",
-      "WebFetch(domain:pypi.org)",
-      "Bash(poetry run python:*)",
-      "Bash(poetry add:*)",
-      "Bash(git fetch:*)",
-      "Bash(git rebase:*)",
-      "Bash(git rm:*)"
-    ],
-    "deny": [],
-    "ask": []
-  }
-}

cvec-1.5.2/.github/workflows/pyproject-license-check.yml

@@ -1,17 +0,0 @@
-name: Python License Check
-
-on:
-  push:
-
-jobs:
-  license-check:
-    runs-on: ubuntu-24.04
-
-    steps:
-      - uses: actions/checkout@1af3b93b6815bc44a9784bd300feb67ff0d1eeb3 # v6.0.0
-
-      - name: Check Python licenses
-        uses: CVector-Energy/pyproject-license-check@main
-        with:
-          app-id: ${{ vars.APP_ID }}
-          app-private-key: ${{ secrets.APP_PRIVATE_KEY }}