feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,3 +1,3 @@
 configured_endpoints: 18
-openapi_spec_hash: 153617b7252b1b12f21043b2a1246f8b
+openapi_spec_hash: 8cd3908ccce472dcb5480c14b290b460
 config_hash: 6dcf08c4324405f152d1da9fc11ab04a

src/openlayer/resources/projects/inference_pipelines.py

@@ -50,6 +50,7 @@ def create(
         *,
         description: Optional[str],
         name: str,
+        data_backend: Optional[inference_pipeline_create_params.DataBackend] | Omit = omit,
         project: Optional[inference_pipeline_create_params.Project] | Omit = omit,
         workspace: Optional[inference_pipeline_create_params.Workspace] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -83,6 +84,7 @@ def create(
                 {
                     "description": description,
                     "name": name,
+                    "data_backend": data_backend,
                     "project": project,
                     "workspace": workspace,
                 },
@@ -174,6 +176,7 @@ async def create(
         *,
         description: Optional[str],
         name: str,
+        data_backend: Optional[inference_pipeline_create_params.DataBackend] | Omit = omit,
         project: Optional[inference_pipeline_create_params.Project] | Omit = omit,
         workspace: Optional[inference_pipeline_create_params.Workspace] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -207,6 +210,7 @@ async def create(
                 {
                     "description": description,
                     "name": name,
+                    "data_backend": data_backend,
                     "project": project,
                     "workspace": workspace,
                 },

src/openlayer/types/inference_pipeline_retrieve_response.py

@@ -1,8 +1,8 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import List, Optional
+from typing import List, Union, Optional
 from datetime import date, datetime
-from typing_extensions import Literal
+from typing_extensions import Literal, TypeAlias
 
 from pydantic import Field as FieldInfo
 
@@ -11,6 +11,18 @@
 __all__ = [
     "InferencePipelineRetrieveResponse",
     "Links",
+    "DataBackend",
+    "DataBackendUnionMember0",
+    "DataBackendUnionMember0Config",
+    "DataBackendBackendType",
+    "DataBackendUnionMember2",
+    "DataBackendUnionMember2Config",
+    "DataBackendUnionMember3",
+    "DataBackendUnionMember3Config",
+    "DataBackendUnionMember4",
+    "DataBackendUnionMember4Config",
+    "DataBackendUnionMember5",
+    "DataBackendUnionMember5Config",
     "Project",
     "ProjectLinks",
     "ProjectGitRepo",
@@ -23,6 +35,167 @@ class Links(BaseModel):
     app: str
 
 
+class DataBackendUnionMember0Config(BaseModel):
+    ground_truth_column_name: Optional[str] = FieldInfo(alias="groundTruthColumnName", default=None)
+    """Name of the column with the ground truths."""
+
+    human_feedback_column_name: Optional[str] = FieldInfo(alias="humanFeedbackColumnName", default=None)
+    """Name of the column with human feedback."""
+
+    latency_column_name: Optional[str] = FieldInfo(alias="latencyColumnName", default=None)
+    """Name of the column with the latencies."""
+
+    timestamp_column_name: Optional[str] = FieldInfo(alias="timestampColumnName", default=None)
+    """Name of the column with the timestamps.
+
+    Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
+    used.
+    """
+
+
+class DataBackendUnionMember0(BaseModel):
+    backend_type: Literal["bigquery"] = FieldInfo(alias="backendType")
+
+    bigquery_connection_id: Optional[str] = FieldInfo(alias="bigqueryConnectionId", default=None)
+
+    dataset_id: str = FieldInfo(alias="datasetId")
+
+    project_id: str = FieldInfo(alias="projectId")
+
+    table_id: Optional[str] = FieldInfo(alias="tableId", default=None)
+
+    partition_type: Optional[Literal["DAY", "MONTH", "YEAR"]] = FieldInfo(alias="partitionType", default=None)
+
+
+class DataBackendBackendType(BaseModel):
+    backend_type: Literal["default"] = FieldInfo(alias="backendType")
+
+
+class DataBackendUnionMember2Config(BaseModel):
+    ground_truth_column_name: Optional[str] = FieldInfo(alias="groundTruthColumnName", default=None)
+    """Name of the column with the ground truths."""
+
+    human_feedback_column_name: Optional[str] = FieldInfo(alias="humanFeedbackColumnName", default=None)
+    """Name of the column with human feedback."""
+
+    latency_column_name: Optional[str] = FieldInfo(alias="latencyColumnName", default=None)
+    """Name of the column with the latencies."""
+
+    timestamp_column_name: Optional[str] = FieldInfo(alias="timestampColumnName", default=None)
+    """Name of the column with the timestamps.
+
+    Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
+    used.
+    """
+
+
+class DataBackendUnionMember2(BaseModel):
+    backend_type: Literal["snowflake"] = FieldInfo(alias="backendType")
+
+    database: str
+
+    schema_: str = FieldInfo(alias="schema")
+
+    snowflake_connection_id: Optional[str] = FieldInfo(alias="snowflakeConnectionId", default=None)
+
+    table: Optional[str] = None
+
+
+class DataBackendUnionMember3Config(BaseModel):
+    ground_truth_column_name: Optional[str] = FieldInfo(alias="groundTruthColumnName", default=None)
+    """Name of the column with the ground truths."""
+
+    human_feedback_column_name: Optional[str] = FieldInfo(alias="humanFeedbackColumnName", default=None)
+    """Name of the column with human feedback."""
+
+    latency_column_name: Optional[str] = FieldInfo(alias="latencyColumnName", default=None)
+    """Name of the column with the latencies."""
+
+    timestamp_column_name: Optional[str] = FieldInfo(alias="timestampColumnName", default=None)
+    """Name of the column with the timestamps.
+
+    Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
+    used.
+    """
+
+
+class DataBackendUnionMember3(BaseModel):
+    backend_type: Literal["databricks_dtl"] = FieldInfo(alias="backendType")
+
+    databricks_dtl_connection_id: Optional[str] = FieldInfo(alias="databricksDtlConnectionId", default=None)
+
+    table_id: Optional[str] = FieldInfo(alias="tableId", default=None)
+
+
+class DataBackendUnionMember4Config(BaseModel):
+    ground_truth_column_name: Optional[str] = FieldInfo(alias="groundTruthColumnName", default=None)
+    """Name of the column with the ground truths."""
+
+    human_feedback_column_name: Optional[str] = FieldInfo(alias="humanFeedbackColumnName", default=None)
+    """Name of the column with human feedback."""
+
+    latency_column_name: Optional[str] = FieldInfo(alias="latencyColumnName", default=None)
+    """Name of the column with the latencies."""
+
+    timestamp_column_name: Optional[str] = FieldInfo(alias="timestampColumnName", default=None)
+    """Name of the column with the timestamps.
+
+    Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
+    used.
+    """
+
+
+class DataBackendUnionMember4(BaseModel):
+    backend_type: Literal["redshift"] = FieldInfo(alias="backendType")
+
+    redshift_connection_id: Optional[str] = FieldInfo(alias="redshiftConnectionId", default=None)
+
+    schema_name: str = FieldInfo(alias="schemaName")
+
+    table_name: str = FieldInfo(alias="tableName")
+
+
+class DataBackendUnionMember5Config(BaseModel):
+    ground_truth_column_name: Optional[str] = FieldInfo(alias="groundTruthColumnName", default=None)
+    """Name of the column with the ground truths."""
+
+    human_feedback_column_name: Optional[str] = FieldInfo(alias="humanFeedbackColumnName", default=None)
+    """Name of the column with human feedback."""
+
+    latency_column_name: Optional[str] = FieldInfo(alias="latencyColumnName", default=None)
+    """Name of the column with the latencies."""
+
+    timestamp_column_name: Optional[str] = FieldInfo(alias="timestampColumnName", default=None)
+    """Name of the column with the timestamps.
+
+    Timestamps must be in UNIX sec format. If not provided, the upload timestamp is
+    used.
+    """
+
+
+class DataBackendUnionMember5(BaseModel):
+    backend_type: Literal["postgres"] = FieldInfo(alias="backendType")
+
+    database: str
+
+    postgres_connection_id: Optional[str] = FieldInfo(alias="postgresConnectionId", default=None)
+
+    schema_: str = FieldInfo(alias="schema")
+
+    table: Optional[str] = None
+
+
+DataBackend: TypeAlias = Union[
+    DataBackendUnionMember0,
+    DataBackendBackendType,
+    DataBackendUnionMember2,
+    DataBackendUnionMember3,
+    DataBackendUnionMember4,
+    DataBackendUnionMember5,
+    None,
+]
+
+
 class ProjectLinks(BaseModel):
     app: str
 
@@ -203,8 +376,16 @@ class InferencePipelineRetrieveResponse(BaseModel):
     total_goal_count: int = FieldInfo(alias="totalGoalCount")
     """The total number of tests."""
 
+    data_backend: Optional[DataBackend] = FieldInfo(alias="dataBackend", default=None)
+
+    date_last_polled: Optional[datetime] = FieldInfo(alias="dateLastPolled", default=None)
+    """The last time the data was polled."""
+
     project: Optional[Project] = None
 
+    total_records_count: Optional[int] = FieldInfo(alias="totalRecordsCount", default=None)
+    """The total number of records in the data backend."""
+
     workspace: Optional[Workspace] = None
 
     workspace_id: Optional[str] = FieldInfo(alias="workspaceId", default=None)