cvec 1.5.2__tar.gz → 2.0.0__tar.gz

cvec-2.0.0/.agents/skills/pull-request/SKILL.md

@@ -0,0 +1,12 @@
+---
+name: pull-request
+description: Open a GitHub pull request using the project's template and including proof of work.
+user-invocable: true
+allowed-tools: Bash(cat .github/pull_request_template.md)
+---
+
+Use the following template for the PR description:
+
+```
+!`cat .github/pull_request_template.md`
+```

cvec-2.0.0/.github/pull_request_template.md

@@ -0,0 +1,19 @@
+<!--- If this PR is closing or supporting a Linear ticket, please link it to the ticket. For example, include the ticket ID in the PR title in brackets, like "[PD1-999] Port to highcharts" -->
+
+# Description
+<!--- Describe your changes in detail -->
+<!--- Why is this change required? What problem does it solve? -->
+
+# Risks
+
+<!--- The author should explain any risks posed by this change. -->
+
+# Testing
+
+## Author
+
+<!--- The author should show proof of work, such as the output of a script calling the changed SDK against a CVector API. -->
+
+## Reviewer
+
+<!--- The author should add instructions for manual test steps that the reviewer should cover, and any special setup instructions. State 'No testing is required.' if no additional testing is required by the reviewer. -->

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

@@ -1,4 +1,4 @@
-name: Continuous Integration
+name: CI
 
 on:
   pull_request:
@@ -9,6 +9,7 @@ on:
       - src/**
       - tests/**
   push:
+    branches: [main]
     paths:
       - .github/workflows/ci.yml
       - uv.lock
@@ -19,15 +20,19 @@ on:
 jobs:
   build:
     runs-on: ubuntu-24.04
+    permissions:
+      contents: read
     strategy:
       matrix:
-        python-version: ["3.10", "3.11", "3.12", "3.13", "3.14"]
+        python-version: ["3.11", "3.12", "3.13", "3.14", "3.15"]
 
     steps:
       - name: Checkout
         uses: actions/checkout@8e8c483db84b4bee98b60c0593521ed34d9990e8 # v6.0.1
+        with:
+          persist-credentials: false
 
-      - uses: CVector-Energy/python-test@uv
+      - uses: CVector-Energy/python-test@d223d8bc5764691c1c87f9609eddad67519d44bd # v2.0.1
         with:
           python-version: ${{ matrix.python-version }}
           src-dirs: .

cvec-2.0.0/.github/workflows/gha.yml

@@ -0,0 +1,65 @@
+name: GitHub Actions Static Analysis
+
+on:
+  push:
+    branches: [main]
+    paths:
+      - ".github/workflows/**"
+      - ".github/actions/**"
+      - "action.yml"
+      - "action.yaml"
+      - "**/action.yml"
+      - "**/action.yaml"
+  pull_request:
+    paths:
+      - ".github/workflows/**"
+      - ".github/actions/**"
+      - "action.yml"
+      - "action.yaml"
+      - "**/action.yml"
+      - "**/action.yaml"
+
+permissions: {}
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  zizmor:
+    name: zizmor
+    runs-on: ubuntu-24.04
+    permissions:
+      contents: read # check out the repository
+      actions: read # read workflow definitions for analysis
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+
+      # zizmor's online audits (impostor-commit) resolve every `uses:` against the GitHub API. The default GITHUB_TOKEN only reaches this repository, so a reusable workflow or action pinned to a private CVector repository, such as CVector-Energy/.github, aborts the whole run with "missing or you have no access".
+      - name: Create GitHub App Token able to read the shared workflow repository
+        id: app
+        # A public repository cannot read the APP_PRIVATE_KEY organization secret, whose visibility is limited to private repositories, and a pull request from a fork receives no secrets at all. Neither case can reference a private workflow, so skip the token instead of failing the job.
+        if: ${{ github.event.repository.private && github.event.pull_request.head.repo.fork != true }}
+        uses: actions/create-github-app-token@bcd2ba49218906704ab6c1aa796996da409d3eb1 # v3.2.0
+        with:
+          client-id: ${{ vars.APP_ID }}
+          private-key: ${{ secrets.APP_PRIVATE_KEY }}
+          owner: ${{ github.repository_owner }}
+          repositories: |
+            .github
+            ${{ github.event.repository.name }}
+          permission-contents: read
+
+      - name: Run zizmor
+        uses: zizmorcore/zizmor-action@3dc1ecc9bcb9e94e9b2c709687979e1298497054 # v0.6.2
+        with:
+          version: 1.25.2
+          advanced-security: false
+          annotations: true
+          token: ${{ steps.app.outputs.token || github.token }}
+
+      - name: Run actionlint
+        uses: raven-actions/actionlint@3d39aea434753780c3b3d4a1a31c854b4dbf49d7 # v2.2.0

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

@@ -0,0 +1,56 @@
+name: Python License Check
+
+on:
+  push:
+
+permissions:
+  contents: read
+  id-token: write
+
+jobs:
+  license-check:
+    runs-on: ubuntu-24.04
+
+    steps:
+      - uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
+        with:
+          persist-credentials: false
+
+      - name: Configure AWS credentials
+        uses: aws-actions/configure-aws-credentials@e6de054238d6b7531b4efff3b6587d9aade6a06c # v6.2.3
+        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@fd767a3655df6df012e3f72bd6ad11d2bca99311 # v0.5.2
+        with:
+          app-id: ${{ vars.APP_ID }}
+          app-private-key: ${{ secrets.APP_PRIVATE_KEY }}
+          # A project with no third-party runtime dependencies has no licenses to check, and LicenseCheck reports that as exit code 3, which the action fails on by default. Accepting it hides nothing: a dependency that cannot be resolved is still listed, with no license, and fails with exit code 1 — as does an incompatible license. Exit code 3 only ever means the project declares no dependencies at all.
+          allow-no-packages: true
+          # Source-aware license resolution (editable projects, private indexes, exact artifacts) is still unreleased, and the `allowed-license-references` and `license-overrides` inputs below depend on it. Move back to a tagged LicenseCheck release once the pull request lands.
+          licensecheck-version: git+https://github.com/FHPythonUtils/LicenseCheck.git@3d26581de610b08a44e697d7d5bb1ee6af6639a2 # LicenseCheck#151
+          allowed-license-references: LicenseRef-NVIDIA-Proprietary LicenseRef-NVIDIA-SOFTWARE-LICENSE
+          license-overrides: |
+            # Public Domain/BSD; the wheel excludes the GPL-covered Emacs mode.
+            "docutils==0.22.4" = "BSD-2-Clause"
+            # BSD-3-Clause; the wheel publishes the full license text without an SPDX identifier.
+            "us==3.2.0" = "BSD-3-Clause"
+            # NVIDIA CUDA Toolkit EULA; reviewed at 13.0.3.0.
+            "cuda-toolkit==13.0.3.0" = "LicenseRef-NVIDIA-Proprietary"
+            # NVIDIA CUDA Toolkit EULA; reviewed at 0.8.1.
+            "nvidia-cusparselt-cu13==0.8.1" = "LicenseRef-NVIDIA-Proprietary"
+            # PSF-2.0 (Python Software Foundation License); the wheel metadata has no SPDX identifier. Windows-only certbot dependency.
+            "pywin32==312" = "PSF-2.0"
+            # Apache-2.0; 1.7.5 publishes no CPython 3.14 wheel, so the audit resolves the private-index sdist and cannot read wheel metadata.
+            "cvxpy==1.7.5" = "Apache-2.0"
+            # Apache-2.0 OR BSD-3-Clause, matching dd-trace-py, which publishes this Datadog stub as a ddtrace 4.16 dependency; the wheel declares no license metadata.
+            "ddtrace-internal==0.1.0" = "Apache-2.0;; BSD-3-Clause"

cvec-1.5.2/README.md → cvec-2.0.0/PKG-INFO

@@ -1,3 +1,25 @@
+Metadata-Version: 2.5
+Name: cvec
+Version: 2.0.0
+Summary: SDK for CVector Energy
+Author-email: CVector <support@cvector.energy>
+License-Expression: MIT
+License-File: LICENSE
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Programming Language :: Python :: 3.15
+Requires-Python: >=3.11
+Requires-Dist: brotli<2,>=1.2.0
+Requires-Dist: pyarrow>=22.0.0
+Requires-Dist: pydantic>=2.12.0
+Description-Content-Type: text/markdown
+
 # CVec Client Library
 
 The "cvec" package is the Python SDK for CVector Energy.
@@ -244,35 +266,6 @@ Add multiple metric data points to the database.
 
 Return a list of metrics that had at least one transition in the given [`start_at`, `end_at`) interval. All metrics are returned if no `start_at` and `end_at` are given.
 
-## `get_modeling_metrics(?start_at, ?end_at)`
-
-Fetch modeling metrics from the modeling database. This method returns a list of available modeling metrics that had transitions in the specified time range.
-
-- `start_at`: Optional start date for the query range (uses class default if not specified)
-- `end_at`: Optional end date for the query range (uses class default if not specified)
-
-Returns a list of `Metric` objects containing modeling metrics.
-
-## `get_modeling_metrics_data(?names, ?start_at, ?end_at)`
-
-Fetch actual data values from modeling metrics within a time range. This method returns the actual data points (values) for the specified modeling metrics, similar to `get_metric_data()` but for the modeling database.
-
-- `names`: Optional list of modeling metric names to filter by
-- `start_at`: Optional start time for the query (uses class default if not specified)
-- `end_at`: Optional end time for the query (uses class default if not specified)
-
-Returns a list of `MetricDataPoint` objects containing the actual data values.
-
-## `get_modeling_metrics_data_arrow(?names, ?start_at, ?end_at)`
-
-Fetch actual data values from modeling metrics within a time range in Apache Arrow format. This method returns the actual data points (values) for the specified modeling metrics in Arrow IPC format, which is more efficient for large datasets.
-
-- `names`: Optional list of modeling metric names to filter by
-- `start_at`: Optional start time for the query (uses class default if not specified)
-- `end_at`: Optional end time for the query (uses class default if not specified)
-
-Returns Arrow IPC format data that can be read using `pyarrow.ipc.open_file()`.
-
 ## `get_eav_tables()`
 
 Get all EAV (Entity-Attribute-Value) tables for the tenant. EAV tables store semi-structured data where each row represents an entity with flexible attributes.

cvec-1.5.2/PKG-INFO → cvec-2.0.0/README.md

@@ -1,25 +1,3 @@
-Metadata-Version: 2.4
-Name: cvec
-Version: 1.5.2
-Summary: SDK for CVector Energy
-Author-email: CVector <support@cvector.energy>
-License-Expression: MIT
-License-File: LICENSE
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Operating System :: OS Independent
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
-Requires-Python: >=3.10
-Requires-Dist: brotli<2,>=1.2.0
-Requires-Dist: pyarrow>=22.0.0
-Requires-Dist: pydantic>=2.12.0
-Description-Content-Type: text/markdown
-
 # CVec Client Library
 
 The "cvec" package is the Python SDK for CVector Energy.
@@ -266,35 +244,6 @@ Add multiple metric data points to the database.
 
 Return a list of metrics that had at least one transition in the given [`start_at`, `end_at`) interval. All metrics are returned if no `start_at` and `end_at` are given.
 
-## `get_modeling_metrics(?start_at, ?end_at)`
-
-Fetch modeling metrics from the modeling database. This method returns a list of available modeling metrics that had transitions in the specified time range.
-
-- `start_at`: Optional start date for the query range (uses class default if not specified)
-- `end_at`: Optional end date for the query range (uses class default if not specified)
-
-Returns a list of `Metric` objects containing modeling metrics.
-
-## `get_modeling_metrics_data(?names, ?start_at, ?end_at)`
-
-Fetch actual data values from modeling metrics within a time range. This method returns the actual data points (values) for the specified modeling metrics, similar to `get_metric_data()` but for the modeling database.
-
-- `names`: Optional list of modeling metric names to filter by
-- `start_at`: Optional start time for the query (uses class default if not specified)
-- `end_at`: Optional end time for the query (uses class default if not specified)
-
-Returns a list of `MetricDataPoint` objects containing the actual data values.
-
-## `get_modeling_metrics_data_arrow(?names, ?start_at, ?end_at)`
-
-Fetch actual data values from modeling metrics within a time range in Apache Arrow format. This method returns the actual data points (values) for the specified modeling metrics in Arrow IPC format, which is more efficient for large datasets.
-
-- `names`: Optional list of modeling metric names to filter by
-- `start_at`: Optional start time for the query (uses class default if not specified)
-- `end_at`: Optional end time for the query (uses class default if not specified)
-
-Returns Arrow IPC format data that can be read using `pyarrow.ipc.open_file()`.
-
 ## `get_eav_tables()`
 
 Get all EAV (Entity-Attribute-Value) tables for the tenant. EAV tables store semi-structured data where each row represents an entity with flexible attributes.

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

@@ -1,10 +1,10 @@
 [project]
 name = "cvec"
-version = "1.5.2"
+version = "2.0.0"
 description = "SDK for CVector Energy"
 authors = [{ name = "CVector", email = "support@cvector.energy" }]
 readme = "README.md"
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 dependencies = [
     "pydantic>=2.12.0",
     "pyarrow>=22.0.0",
@@ -17,16 +17,16 @@ classifiers = [
     "Operating System :: OS Independent",
     "Programming Language :: Python :: 3",
     'Intended Audience :: Developers',
-    'Programming Language :: Python :: 3.10',
     'Programming Language :: Python :: 3.11',
     'Programming Language :: Python :: 3.12',
     'Programming Language :: Python :: 3.13',
     'Programming Language :: Python :: 3.14',
+    'Programming Language :: Python :: 3.15',
 ]
 
 [dependency-groups]
 dev = [
-    "pytest>=8.3.5,<9",
+    "pytest>=8.3.5,<10",
     "mypy>=1.15.0,<2",
     "ruff>=0.11.10,<1",
     "python-claude>=0.3.0,<0.4",

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

@@ -14,7 +14,6 @@ from urllib.request import Request, urlopen
 
 from cvec.http_cache import MAX_CACHE_ENTRIES, CacheEntry, parse_max_age
 
-from cvec.models.agent_post import AgentPost, AgentPostRecommendation, AgentPostTag
 from cvec.models.eav_column import EAVColumn
 from cvec.models.eav_filter import EAVFilter
 from cvec.models.eav_table import EAVTable
@@ -36,6 +35,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 +49,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 +212,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 +236,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(
@@ -402,137 +404,6 @@ class CVec:
             ]
             self._make_request("POST", endpoint, json_data=data_dicts)  # type: ignore[arg-type]
 
-    def get_modeling_metrics(
-        self,
-        start_at: Optional[datetime] = None,
-        end_at: Optional[datetime] = None,
-    ) -> List[Metric]:
-        """
-        Return a list of modeling metrics that had at least one transition in the given [start_at, end_at) interval.
-        All metrics are returned if no start_at and end_at are given.
-
-        Args:
-            start_at: Optional start time for the query (uses class default if not specified)
-            end_at: Optional end time for the query (uses class default if not specified)
-
-        Returns:
-            List of Metric objects containing modeling metrics
-        """
-        _start_at = start_at or self.default_start_at
-        _end_at = end_at or self.default_end_at
-
-        params: Dict[str, Any] = {
-            "start_at": _start_at.isoformat() if _start_at else None,
-            "end_at": _end_at.isoformat() if _end_at else None,
-        }
-
-        response_data = self._make_request(
-            "GET", "/api/modeling/metrics", params=params
-        )
-        return [Metric.model_validate(metric_data) for metric_data in response_data]
-
-    def get_modeling_metrics_data(
-        self,
-        names: Optional[List[str]] = None,
-        start_at: Optional[datetime] = None,
-        end_at: Optional[datetime] = None,
-    ) -> List[MetricDataPoint]:
-        """
-        Return all data-points within a given [start_at, end_at) interval,
-        optionally selecting a given list of modeling metric names.
-        Returns a list of MetricDataPoint objects, one for each metric value transition.
-
-        Args:
-            names: Optional list of modeling metric names to filter by
-            start_at: Optional start time for the query
-            end_at: Optional end time for the query
-        """
-        _start_at = start_at or self.default_start_at
-        _end_at = end_at or self.default_end_at
-
-        params: Dict[str, Any] = {
-            "start_at": _start_at.isoformat() if _start_at else None,
-            "end_at": _end_at.isoformat() if _end_at else None,
-            "names": ",".join(names) if names else None,
-        }
-
-        response_data = self._make_request(
-            "GET", "/api/modeling/metrics/data", params=params
-        )
-        return [
-            MetricDataPoint.model_validate(point_data) for point_data in response_data
-        ]
-
-    def get_modeling_metrics_data_arrow(
-        self,
-        names: Optional[List[str]] = None,
-        start_at: Optional[datetime] = None,
-        end_at: Optional[datetime] = None,
-    ) -> bytes:
-        """
-        Return all data-points within a given [start_at, end_at) interval,
-        optionally selecting a given list of modeling metric names.
-        Returns Arrow IPC format data that can be read using pyarrow.ipc.open_file.
-
-        Args:
-            names: Optional list of modeling metric names to filter by
-            start_at: Optional start time for the query
-            end_at: Optional end time for the query
-        """
-        _start_at = start_at or self.default_start_at
-        _end_at = end_at or self.default_end_at
-
-        params: Dict[str, Any] = {
-            "start_at": _start_at.isoformat() if _start_at else None,
-            "end_at": _end_at.isoformat() if _end_at else None,
-            "names": ",".join(names) if names else None,
-        }
-
-        endpoint = "/api/modeling/metrics/data/arrow"
-        result = self._make_request("GET", endpoint, params=params)
-        assert isinstance(result, bytes)
-        return result
-
-    def add_agent_post(
-        self,
-        title: str,
-        author: str,
-        image_id: Optional[str] = None,
-        content: Optional[str] = None,
-        recommendations: Optional[List[AgentPostRecommendation]] = None,
-        tags: Optional[List[AgentPostTag]] = None,
-        is_private: bool = True,
-    ) -> None:
-        """
-        Add an agent post.
-
-        Note: If image_id is provided, the image must be uploaded to S3 beforehand.
-        The image_id should be the UUID used as the filename (without .png extension)
-        in the S3 bucket at the tenant's path.
-
-        Args:
-            title: Post title
-            author: Post author
-            image_id: Optional UUID of the uploaded image
-            content: Optional post content
-            recommendations: Optional list of recommendations
-            tags: Optional list of tags
-            is_private: Whether the post is private (only visible to superadmins). Defaults to True.
-        """
-
-        post = AgentPost(
-            title=title,
-            author=author,
-            image_id=image_id,
-            content=content,
-            recommendations=recommendations,
-            tags=tags,
-            is_private=is_private,
-        )
-        payload = post.model_dump(mode="json", exclude_none=True)
-
-        self._make_request("POST", "/api/agent_posts/add", json_data=payload)
-
     def _login_with_supabase(self, email: str, password: str) -> None:
         """
         Login to Supabase and get access/refresh tokens.
@@ -557,7 +428,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 +457,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 +480,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 +535,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 +553,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 +602,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 +618,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-2.0.0}/src/cvec/models/__init__.py

@@ -1,9 +1,3 @@
-from .agent_post import (
-    AgentPost,
-    AgentPostRecommendation,
-    AgentPostTag,
-    Severity,
-)
 from .eav_column import EAVColumn
 from .eav_filter import EAVFilter
 from .eav_table import EAVTable
@@ -11,14 +5,10 @@ from .metric import Metric, MetricDataPoint
 from .span import Span
 
 __all__ = [
-    "AgentPost",
-    "AgentPostRecommendation",
-    "AgentPostTag",
     "EAVColumn",
     "EAVFilter",
     "EAVTable",
     "Metric",
     "MetricDataPoint",
-    "Severity",
     "Span",
 ]