feat(api): OpenAPI spec update via Stainless API (#210)

Author: stainless-app[bot]

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."""
 

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",