openlayer-ai · stainless-app · May 22, 2024 · May 21, 2024 · May 21, 2024 · May 22, 2024
diff --git a/.release-please-manifest.json b/.release-please-manifest.json
@@ -1,3 +1,3 @@
 {
-  ".": "0.1.0-alpha.2"
+  ".": "0.1.0-alpha.3"
 }
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -20,6 +20,16 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 ### Removed
 * Deprecated and removed `publish_ground_truths` method. Use `update_data` instead.
 
+## 0.1.0-alpha.3 (2024-05-22)
+
+Full Changelog: [v0.1.0-alpha.2...v0.1.0-alpha.3](https://github.com/openlayer-ai/openlayer-python/compare/v0.1.0-alpha.2...v0.1.0-alpha.3)
+
+### Features
+
+* **api:** OpenAPI spec update via Stainless API ([#207](https://github.com/openlayer-ai/openlayer-python/issues/207)) ([0a806f1](https://github.com/openlayer-ai/openlayer-python/commit/0a806f1be1042caeefcebb2bf17636190abb4685))
+* **api:** OpenAPI spec update via Stainless API ([#209](https://github.com/openlayer-ai/openlayer-python/issues/209)) ([da14f38](https://github.com/openlayer-ai/openlayer-python/commit/da14f383fd48523a7e79431dd50ff7c6baac370b))
+* **api:** OpenAPI spec update via Stainless API ([#210](https://github.com/openlayer-ai/openlayer-python/issues/210)) ([9a261c6](https://github.com/openlayer-ai/openlayer-python/commit/9a261c6b3bdada872bd221d5bbd311d5e3d12fcf))
+
 ## 0.1.0-alpha.2 (2024-05-20)
 
 Full Changelog: [v0.1.0-alpha.1...v0.1.0-alpha.2](https://github.com/openlayer-ai/openlayer-python/compare/v0.1.0-alpha.1...v0.1.0-alpha.2)

diff --git a/README.md b/README.md
@@ -1,6 +1,6 @@
 # Openlayer Python API library
 
-[![PyPI version](https://img.shields.io/pypi/v/openlayer-test.svg)](https://pypi.org/project/openlayer-test/)
+[![PyPI version](https://img.shields.io/pypi/v/openlayer.svg)](https://pypi.org/project/openlayer/)
 
 The Openlayer Python library provides convenient access to the Openlayer REST API from any Python 3.7+
 application. The library includes type definitions for all request params and response fields,
@@ -10,13 +10,13 @@ It is generated with [Stainless](https://www.stainlessapi.com/).
 
 ## Documentation
 
-The REST API documentation can be found [on openlayer.com](https://openlayer.com/docs/api-reference/rest). The full API of this library can be found in [api.md](api.md).
+The REST API documentation can be found [on openlayer.com](https://openlayer.com/docs/api-reference/rest/overview). The full API of this library can be found in [api.md](api.md).
 
 ## Installation
 
 ```sh
 # install from PyPI
-pip install --pre openlayer-test
+pip install --pre openlayer
 ```
 
 ## Usage

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
-name = "openlayer-test"
-version = "0.1.0-alpha.2"
+name = "openlayer"
+version = "0.1.0-alpha.3"
 description = "The official Python library for the openlayer API"
 dynamic = ["readme"]
 license = "Apache-2.0"

diff --git a/requirements-dev.lock b/requirements-dev.lock
@@ -12,7 +12,7 @@ annotated-types==0.6.0
     # via pydantic
 anyio==4.1.0
     # via httpx
-    # via openlayer-test
+    # via openlayer
 argcomplete==3.1.2
     # via nox
 attrs==23.1.0
@@ -26,7 +26,7 @@ dirty-equals==0.6.0
 distlib==0.3.7
     # via virtualenv
 distro==1.8.0
-    # via openlayer-test
+    # via openlayer
 exceptiongroup==1.1.3
     # via anyio
 filelock==3.12.4
@@ -36,7 +36,7 @@ h11==0.14.0
 httpcore==1.0.2
     # via httpx
 httpx==0.25.2
-    # via openlayer-test
+    # via openlayer
     # via respx
 idna==3.4
     # via anyio
@@ -60,7 +60,7 @@ pluggy==1.3.0
 py==1.11.0
     # via pytest
 pydantic==2.7.1
-    # via openlayer-test
+    # via openlayer
 pydantic-core==2.18.2
     # via pydantic
 pyright==1.1.359
@@ -80,14 +80,14 @@ six==1.16.0
 sniffio==1.3.0
     # via anyio
     # via httpx
-    # via openlayer-test
+    # via openlayer
 time-machine==2.9.0
 tomli==2.0.1
     # via mypy
     # via pytest
 typing-extensions==4.8.0
     # via mypy
-    # via openlayer-test
+    # via openlayer
     # via pydantic
     # via pydantic-core
 virtualenv==20.24.5

diff --git a/requirements.lock b/requirements.lock
@@ -12,32 +12,32 @@ annotated-types==0.6.0
     # via pydantic
 anyio==4.1.0
     # via httpx
-    # via openlayer-test
+    # via openlayer
 certifi==2023.7.22
     # via httpcore
     # via httpx
 distro==1.8.0
-    # via openlayer-test
+    # via openlayer
 exceptiongroup==1.1.3
     # via anyio
 h11==0.14.0
     # via httpcore
 httpcore==1.0.2
     # via httpx
 httpx==0.25.2
-    # via openlayer-test
+    # via openlayer
 idna==3.4
     # via anyio
     # via httpx
 pydantic==2.7.1
-    # via openlayer-test
+    # via openlayer
 pydantic-core==2.18.2
     # via pydantic
 sniffio==1.3.0
     # via anyio
     # via httpx
-    # via openlayer-test
+    # via openlayer
 typing-extensions==4.8.0
-    # via openlayer-test
+    # via openlayer
     # via pydantic
     # via pydantic-core
diff --git a/src/openlayer/_base_client.py b/src/openlayer/_base_client.py
@@ -361,7 +361,7 @@ def __init__(
 
         if max_retries is None:  # pyright: ignore[reportUnnecessaryComparison]
             raise TypeError(
-                "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `openlayer-test.DEFAULT_MAX_RETRIES`"
+                "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `openlayer.DEFAULT_MAX_RETRIES`"
             )
 
     def _enforce_trailing_slash(self, url: URL) -> URL:

diff --git a/src/openlayer/_version.py b/src/openlayer/_version.py
@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "openlayer"
-__version__ = "0.1.0-alpha.2"  # x-release-please-version
+__version__ = "0.1.0-alpha.3"  # x-release-please-version
diff --git a/src/openlayer/types/inference_pipelines/data_stream_params.py b/src/openlayer/types/inference_pipelines/data_stream_params.py
@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable
+from typing import Dict, List, Union, Iterable, Optional
 from typing_extensions import Required, Annotated, TypedDict
 
 from ..._utils import PropertyInfo
@@ -69,6 +69,9 @@ class ConfigLlmData(TypedDict, total=False):
     metadata: object
     """Object with metadata."""
 
+    num_of_token_column_name: Annotated[Optional[str], PropertyInfo(alias="numOfTokenColumnName")]
+    """Name of the column with the total number of tokens."""
+
     prompt: Iterable[ConfigLlmDataPrompt]
     """Prompt for the LLM."""
 

diff --git a/tests/api_resources/inference_pipelines/test_data.py b/tests/api_resources/inference_pipelines/test_data.py
@@ -39,6 +39,7 @@ def test_method_stream_with_all_params(self, client: Openlayer) -> None:
         data = client.inference_pipelines.data.stream(
             "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             config={
+                "num_of_token_column_name": "tokens",
                 "context_column_name": "context",
                 "cost_column_name": "cost",
                 "ground_truth_column_name": "ground_truth",
@@ -155,6 +156,7 @@ async def test_method_stream_with_all_params(self, async_client: AsyncOpenlayer)
         data = await async_client.inference_pipelines.data.stream(
             "182bd5e5-6e1a-4fe4-a799-aa6d9a6ab26e",
             config={
+                "num_of_token_column_name": "tokens",
                 "context_column_name": "context",
                 "cost_column_name": "cost",
                 "ground_truth_column_name": "ground_truth",