diff --git a/.release-please-manifest.json b/.release-please-manifest.json index 3e2bf49..2aca35a 100644 --- a/.release-please-manifest.json +++ b/.release-please-manifest.json @@ -1,3 +1,3 @@ { - ".": "0.4.1" + ".": "0.5.0" } \ No newline at end of file diff --git a/.stats.yml b/.stats.yml index 21a1b4e..ffcf5e3 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,3 +1,3 @@ configured_endpoints: 18 -openapi_spec_hash: 153617b7252b1b12f21043b2a1246f8b +openapi_spec_hash: 539798fac79a1eeebf9ac4faa0492455 config_hash: 6dcf08c4324405f152d1da9fc11ab04a diff --git a/CHANGELOG.md b/CHANGELOG.md index b637cb7..8b559dc 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,20 @@ # Changelog +## 0.5.0 (2025-10-28) + +Full Changelog: [v0.4.1...v0.5.0](https://github.com/openlayer-ai/openlayer-ruby/compare/v0.4.1...v0.5.0) + +### Features + +* **api:** api update ([e686c41](https://github.com/openlayer-ai/openlayer-ruby/commit/e686c41ce85b71f1d36e6e98c57fd1e15b0dba50)) +* **api:** api update ([062f345](https://github.com/openlayer-ai/openlayer-ruby/commit/062f34578e996d2f947d6175656e29657bd6d325)) +* handle thread interrupts in the core HTTP client ([ec67b44](https://github.com/openlayer-ai/openlayer-ruby/commit/ec67b44714f9f117f129ace368b40e47e9ec3f3c)) + + +### Bug Fixes + +* absolutely qualified uris should always override the default ([e245d11](https://github.com/openlayer-ai/openlayer-ruby/commit/e245d1191c6f5e73a5f4431a8bfff1989d828648)) + ## 0.4.1 (2025-10-15) Full Changelog: [v0.4.0...v0.4.1](https://github.com/openlayer-ai/openlayer-ruby/compare/v0.4.0...v0.4.1) diff --git a/Gemfile.lock b/Gemfile.lock index 6fbb472..b6e6c2f 100644 --- a/Gemfile.lock +++ b/Gemfile.lock @@ -11,7 +11,7 @@ GIT PATH remote: . specs: - openlayer (0.4.1) + openlayer (0.5.0) connection_pool GEM diff --git a/README.md b/README.md index fa6e6b7..91d14c5 100644 --- a/README.md +++ b/README.md @@ -17,7 +17,7 @@ To use this gem, install via Bundler by adding the following to your application ```ruby -gem "openlayer", "~> 0.4.1" +gem "openlayer", "~> 0.5.0" ``` diff --git a/lib/openlayer/internal/transport/pooled_net_requester.rb b/lib/openlayer/internal/transport/pooled_net_requester.rb index 94f640f..d6ca2d8 100644 --- a/lib/openlayer/internal/transport/pooled_net_requester.rb +++ b/lib/openlayer/internal/transport/pooled_net_requester.rb @@ -128,40 +128,48 @@ def execute(request) url, deadline = request.fetch_values(:url, :deadline) req = nil - eof = false finished = false - closing = nil # rubocop:disable Metrics/BlockLength enum = Enumerator.new do |y| next if finished with_pool(url, deadline: deadline) do |conn| - req, closing = self.class.build_request(request) do - self.class.calibrate_socket_timeout(conn, deadline) - end - - self.class.calibrate_socket_timeout(conn, deadline) - unless conn.started? - conn.keep_alive_timeout = self.class::KEEP_ALIVE_TIMEOUT - conn.start - end + eof = false + closing = nil + ::Thread.handle_interrupt(Object => :never) do + ::Thread.handle_interrupt(Object => :immediate) do + req, closing = self.class.build_request(request) do + self.class.calibrate_socket_timeout(conn, deadline) + end - self.class.calibrate_socket_timeout(conn, deadline) - conn.request(req) do |rsp| - y << [req, rsp] - break if finished - - rsp.read_body do |bytes| - y << bytes.force_encoding(Encoding::BINARY) - break if finished + self.class.calibrate_socket_timeout(conn, deadline) + unless conn.started? + conn.keep_alive_timeout = self.class::KEEP_ALIVE_TIMEOUT + conn.start + end self.class.calibrate_socket_timeout(conn, deadline) + conn.request(req) do |rsp| + y << [req, rsp] + break if finished + + rsp.read_body do |bytes| + y << bytes.force_encoding(Encoding::BINARY) + break if finished + + self.class.calibrate_socket_timeout(conn, deadline) + end + eof = true + end + end + ensure + begin + conn.finish if !eof && conn&.started? + ensure + closing&.call end - eof = true end - ensure - conn.finish if !eof && conn&.started? end rescue Timeout::Error raise Openlayer::Errors::APITimeoutError.new(url: url, request: req) @@ -174,8 +182,6 @@ def execute(request) body = Openlayer::Internal::Util.fused_enum(enum, external: true) do finished = true loop { enum.next } - ensure - closing&.call end [Integer(response.code), response, body] end diff --git a/lib/openlayer/internal/util.rb b/lib/openlayer/internal/util.rb index 738fd19..add608b 100644 --- a/lib/openlayer/internal/util.rb +++ b/lib/openlayer/internal/util.rb @@ -346,8 +346,9 @@ def join_parsed_uri(lhs, rhs) base_path, base_query = lhs.fetch_values(:path, :query) slashed = base_path.end_with?("/") ? base_path : "#{base_path}/" - parsed_path, parsed_query = parse_uri(rhs.fetch(:path)).fetch_values(:path, :query) - override = URI::Generic.build(**rhs.slice(:scheme, :host, :port), path: parsed_path) + merged = {**parse_uri(rhs.fetch(:path)), **rhs.except(:path, :query)} + parsed_path, parsed_query = merged.fetch_values(:path, :query) + override = URI::Generic.build(**merged.slice(:scheme, :host, :port), path: parsed_path) joined = URI.join(URI::Generic.build(lhs.except(:path, :query)), slashed, override) query = deep_merge( diff --git a/lib/openlayer/models/inference_pipeline_retrieve_response.rb b/lib/openlayer/models/inference_pipeline_retrieve_response.rb index cdd031a..8cceb10 100644 --- a/lib/openlayer/models/inference_pipeline_retrieve_response.rb +++ b/lib/openlayer/models/inference_pipeline_retrieve_response.rb @@ -16,6 +16,14 @@ class InferencePipelineRetrieveResponse < Openlayer::Internal::Type::BaseModel # @return [String] required :name, String + # @!attribute data_backend + # + # @return [Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5, nil] + optional :data_backend, + union: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend }, + api_name: :dataBackend, + nil?: true + # @!attribute project # # @return [Openlayer::Models::InferencePipelineRetrieveResponse::Project, nil] @@ -104,6 +112,18 @@ class InferencePipelineRetrieveResponse < Openlayer::Internal::Type::BaseModel # @return [Integer] required :total_goal_count, Integer, api_name: :totalGoalCount + # @!attribute date_last_polled + # The last time the data was polled. + # + # @return [Time, nil] + optional :date_last_polled, Time, api_name: :dateLastPolled, nil?: true + + # @!attribute total_records_count + # The total number of records in the data backend. + # + # @return [Integer, nil] + optional :total_records_count, Integer, api_name: :totalRecordsCount, nil?: true + # @!attribute workspace_id # The workspace id. # @@ -111,7 +131,7 @@ class InferencePipelineRetrieveResponse < Openlayer::Internal::Type::BaseModel optional :workspace_id, String, api_name: :workspaceId end - # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, project: nil, workspace: nil, workspace_id: nil) + # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, data_backend: nil, date_last_polled: nil, project: nil, total_records_count: nil, workspace: nil, workspace_id: nil) # @param id [String] The inference pipeline id. # # @param date_created [Time] The creation date. @@ -142,8 +162,14 @@ class InferencePipelineRetrieveResponse < Openlayer::Internal::Type::BaseModel # # @param total_goal_count [Integer] The total number of tests. # + # @param data_backend [Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5, nil] + # + # @param date_last_polled [Time, nil] The last time the data was polled. + # # @param project [Openlayer::Models::InferencePipelineRetrieveResponse::Project, nil] # + # @param total_records_count [Integer, nil] The total number of records in the data backend. + # # @param workspace [Openlayer::Models::InferencePipelineRetrieveResponse::Workspace, nil] # # @param workspace_id [String] The workspace id. @@ -176,6 +202,486 @@ module Status # @return [Array] end + # @see Openlayer::Models::InferencePipelineRetrieveResponse#data_backend + module DataBackend + extend Openlayer::Internal::Type::Union + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0 } + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType } + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2 } + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3 } + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4 } + + variant -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5 } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType }, + api_name: :backendType + + # @!attribute bigquery_connection_id + # + # @return [String, nil] + required :bigquery_connection_id, String, api_name: :bigqueryConnectionId, nil?: true + + # @!attribute dataset_id + # + # @return [String] + required :dataset_id, String, api_name: :datasetId + + # @!attribute project_id + # + # @return [String] + required :project_id, String, api_name: :projectId + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!attribute partition_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType, nil] + optional :partition_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType }, + api_name: :partitionType, + nil?: true + + # @!method initialize(backend_type:, bigquery_connection_id:, dataset_id:, project_id:, table_id:, partition_type: nil) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType] + # @param bigquery_connection_id [String, nil] + # @param dataset_id [String] + # @param project_id [String] + # @param table_id [String, nil] + # @param partition_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType, nil] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY = :bigquery + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0#partition_type + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY = :DAY + MONTH = :MONTH + YEAR = :YEAR + + # @!method self.values + # @return [Array] + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType }, + api_name: :backendType + + # @!method initialize(backend_type:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT = :default + + # @!method self.values + # @return [Array] + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute snowflake_connection_id + # + # @return [String, nil] + required :snowflake_connection_id, String, api_name: :snowflakeConnectionId, nil?: true + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, schema:, snowflake_connection_id:, table:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType] + # @param database [String] + # @param schema [String] + # @param snowflake_connection_id [String, nil] + # @param table [String, nil] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE = :snowflake + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType }, + api_name: :backendType + + # @!attribute databricks_dtl_connection_id + # + # @return [String, nil] + required :databricks_dtl_connection_id, String, api_name: :databricksDtlConnectionId, nil?: true + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!method initialize(backend_type:, databricks_dtl_connection_id:, table_id:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType] + # @param databricks_dtl_connection_id [String, nil] + # @param table_id [String, nil] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL = :databricks_dtl + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType }, + api_name: :backendType + + # @!attribute redshift_connection_id + # + # @return [String, nil] + required :redshift_connection_id, String, api_name: :redshiftConnectionId, nil?: true + + # @!attribute schema_name + # + # @return [String] + required :schema_name, String, api_name: :schemaName + + # @!attribute table_name + # + # @return [String] + required :table_name, String, api_name: :tableName + + # @!method initialize(backend_type:, redshift_connection_id:, schema_name:, table_name:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType] + # @param redshift_connection_id [String, nil] + # @param schema_name [String] + # @param table_name [String] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT = :redshift + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute postgres_connection_id + # + # @return [String, nil] + required :postgres_connection_id, String, api_name: :postgresConnectionId, nil?: true + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, postgres_connection_id:, schema:, table:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType] + # @param database [String] + # @param postgres_connection_id [String, nil] + # @param schema [String] + # @param table [String, nil] + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES = :postgres + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + # @!method self.variants + # @return [Array(Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5)] + end + # @see Openlayer::Models::InferencePipelineRetrieveResponse#project class Project < Openlayer::Internal::Type::BaseModel # @!attribute name diff --git a/lib/openlayer/models/inference_pipeline_update_response.rb b/lib/openlayer/models/inference_pipeline_update_response.rb index 82856bf..75a0eb7 100644 --- a/lib/openlayer/models/inference_pipeline_update_response.rb +++ b/lib/openlayer/models/inference_pipeline_update_response.rb @@ -16,6 +16,14 @@ class InferencePipelineUpdateResponse < Openlayer::Internal::Type::BaseModel # @return [String] required :name, String + # @!attribute data_backend + # + # @return [Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5, nil] + optional :data_backend, + union: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend }, + api_name: :dataBackend, + nil?: true + # @!attribute project # # @return [Openlayer::Models::InferencePipelineUpdateResponse::Project, nil] @@ -104,6 +112,18 @@ class InferencePipelineUpdateResponse < Openlayer::Internal::Type::BaseModel # @return [Integer] required :total_goal_count, Integer, api_name: :totalGoalCount + # @!attribute date_last_polled + # The last time the data was polled. + # + # @return [Time, nil] + optional :date_last_polled, Time, api_name: :dateLastPolled, nil?: true + + # @!attribute total_records_count + # The total number of records in the data backend. + # + # @return [Integer, nil] + optional :total_records_count, Integer, api_name: :totalRecordsCount, nil?: true + # @!attribute workspace_id # The workspace id. # @@ -111,7 +131,7 @@ class InferencePipelineUpdateResponse < Openlayer::Internal::Type::BaseModel optional :workspace_id, String, api_name: :workspaceId end - # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, project: nil, workspace: nil, workspace_id: nil) + # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, data_backend: nil, date_last_polled: nil, project: nil, total_records_count: nil, workspace: nil, workspace_id: nil) # @param id [String] The inference pipeline id. # # @param date_created [Time] The creation date. @@ -142,8 +162,14 @@ class InferencePipelineUpdateResponse < Openlayer::Internal::Type::BaseModel # # @param total_goal_count [Integer] The total number of tests. # + # @param data_backend [Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5, nil] + # + # @param date_last_polled [Time, nil] The last time the data was polled. + # # @param project [Openlayer::Models::InferencePipelineUpdateResponse::Project, nil] # + # @param total_records_count [Integer, nil] The total number of records in the data backend. + # # @param workspace [Openlayer::Models::InferencePipelineUpdateResponse::Workspace, nil] # # @param workspace_id [String] The workspace id. @@ -176,6 +202,486 @@ module Status # @return [Array] end + # @see Openlayer::Models::InferencePipelineUpdateResponse#data_backend + module DataBackend + extend Openlayer::Internal::Type::Union + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0 } + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType } + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2 } + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3 } + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4 } + + variant -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5 } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType }, + api_name: :backendType + + # @!attribute bigquery_connection_id + # + # @return [String, nil] + required :bigquery_connection_id, String, api_name: :bigqueryConnectionId, nil?: true + + # @!attribute dataset_id + # + # @return [String] + required :dataset_id, String, api_name: :datasetId + + # @!attribute project_id + # + # @return [String] + required :project_id, String, api_name: :projectId + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!attribute partition_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType, nil] + optional :partition_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType }, + api_name: :partitionType, + nil?: true + + # @!method initialize(backend_type:, bigquery_connection_id:, dataset_id:, project_id:, table_id:, partition_type: nil) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType] + # @param bigquery_connection_id [String, nil] + # @param dataset_id [String] + # @param project_id [String] + # @param table_id [String, nil] + # @param partition_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType, nil] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY = :bigquery + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0#partition_type + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY = :DAY + MONTH = :MONTH + YEAR = :YEAR + + # @!method self.values + # @return [Array] + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType }, + api_name: :backendType + + # @!method initialize(backend_type:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT = :default + + # @!method self.values + # @return [Array] + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute snowflake_connection_id + # + # @return [String, nil] + required :snowflake_connection_id, String, api_name: :snowflakeConnectionId, nil?: true + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, schema:, snowflake_connection_id:, table:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType] + # @param database [String] + # @param schema [String] + # @param snowflake_connection_id [String, nil] + # @param table [String, nil] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE = :snowflake + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType }, + api_name: :backendType + + # @!attribute databricks_dtl_connection_id + # + # @return [String, nil] + required :databricks_dtl_connection_id, String, api_name: :databricksDtlConnectionId, nil?: true + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!method initialize(backend_type:, databricks_dtl_connection_id:, table_id:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType] + # @param databricks_dtl_connection_id [String, nil] + # @param table_id [String, nil] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL = :databricks_dtl + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType }, + api_name: :backendType + + # @!attribute redshift_connection_id + # + # @return [String, nil] + required :redshift_connection_id, String, api_name: :redshiftConnectionId, nil?: true + + # @!attribute schema_name + # + # @return [String] + required :schema_name, String, api_name: :schemaName + + # @!attribute table_name + # + # @return [String] + required :table_name, String, api_name: :tableName + + # @!method initialize(backend_type:, redshift_connection_id:, schema_name:, table_name:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType] + # @param redshift_connection_id [String, nil] + # @param schema_name [String] + # @param table_name [String] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT = :redshift + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute postgres_connection_id + # + # @return [String, nil] + required :postgres_connection_id, String, api_name: :postgresConnectionId, nil?: true + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, postgres_connection_id:, schema:, table:) + # @param backend_type [Symbol, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType] + # @param database [String] + # @param postgres_connection_id [String, nil] + # @param schema [String] + # @param table [String, nil] + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES = :postgres + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + # @!method self.variants + # @return [Array(Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4, Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5)] + end + # @see Openlayer::Models::InferencePipelineUpdateResponse#project class Project < Openlayer::Internal::Type::BaseModel # @!attribute name diff --git a/lib/openlayer/models/projects/inference_pipeline_create_params.rb b/lib/openlayer/models/projects/inference_pipeline_create_params.rb index 68da1df..5e481aa 100644 --- a/lib/openlayer/models/projects/inference_pipeline_create_params.rb +++ b/lib/openlayer/models/projects/inference_pipeline_create_params.rb @@ -20,6 +20,14 @@ class InferencePipelineCreateParams < Openlayer::Internal::Type::BaseModel # @return [String] required :name, String + # @!attribute data_backend + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5, nil] + optional :data_backend, + union: -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend }, + api_name: :dataBackend, + nil?: true + # @!attribute project # # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::Project, nil] @@ -30,17 +38,617 @@ class InferencePipelineCreateParams < Openlayer::Internal::Type::BaseModel # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::Workspace, nil] optional :workspace, -> { Openlayer::Projects::InferencePipelineCreateParams::Workspace }, nil?: true - # @!method initialize(description:, name:, project: nil, workspace: nil, request_options: {}) + # @!method initialize(description:, name:, data_backend: nil, project: nil, workspace: nil, request_options: {}) # @param description [String, nil] The inference pipeline description. # # @param name [String] The inference pipeline name. # + # @param data_backend [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5, nil] + # # @param project [Openlayer::Models::Projects::InferencePipelineCreateParams::Project, nil] # # @param workspace [Openlayer::Models::Projects::InferencePipelineCreateParams::Workspace, nil] # # @param request_options [Openlayer::RequestOptions, Hash{Symbol=>Object}] + module DataBackend + extend Openlayer::Internal::Type::Union + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0 } + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType } + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2 } + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3 } + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4 } + + variant -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5 } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType + }, + api_name: :backendType + + # @!attribute bigquery_connection_id + # + # @return [String, nil] + required :bigquery_connection_id, String, api_name: :bigqueryConnectionId, nil?: true + + # @!attribute dataset_id + # + # @return [String] + required :dataset_id, String, api_name: :datasetId + + # @!attribute project_id + # + # @return [String] + required :project_id, String, api_name: :projectId + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!attribute partition_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType, nil] + optional :partition_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType + }, + api_name: :partitionType, + nil?: true + + request_only do + # @!attribute config + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config] + required :config, + -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config } + end + + # @!method initialize(backend_type:, bigquery_connection_id:, config:, dataset_id:, project_id:, table_id:, partition_type: nil) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType] + # @param bigquery_connection_id [String, nil] + # @param config [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config] + # @param dataset_id [String] + # @param project_id [String] + # @param table_id [String, nil] + # @param partition_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY = :bigquery + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + request_only do + # @!attribute inference_id_column_name + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + # + # @return [String, nil] + optional :inference_id_column_name, String, api_name: :inferenceIdColumnName, nil?: true + end + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, inference_id_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param inference_id_column_name [String, nil] Name of the column with the inference ids. This is useful if you want to update + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0#partition_type + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY = :DAY + MONTH = :MONTH + YEAR = :YEAR + + # @!method self.values + # @return [Array] + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType + }, + api_name: :backendType + + # @!method initialize(backend_type:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT = :default + + # @!method self.values + # @return [Array] + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType + }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute snowflake_connection_id + # + # @return [String, nil] + required :snowflake_connection_id, String, api_name: :snowflakeConnectionId, nil?: true + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + request_only do + # @!attribute config + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config] + required :config, + -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config } + end + + # @!method initialize(backend_type:, config:, database:, schema:, snowflake_connection_id:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType] + # @param config [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config] + # @param database [String] + # @param schema [String] + # @param snowflake_connection_id [String, nil] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE = :snowflake + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + request_only do + # @!attribute inference_id_column_name + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + # + # @return [String, nil] + optional :inference_id_column_name, String, api_name: :inferenceIdColumnName, nil?: true + end + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, inference_id_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param inference_id_column_name [String, nil] Name of the column with the inference ids. This is useful if you want to update + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType + }, + api_name: :backendType + + # @!attribute databricks_dtl_connection_id + # + # @return [String, nil] + required :databricks_dtl_connection_id, String, api_name: :databricksDtlConnectionId, nil?: true + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + request_only do + # @!attribute config + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config] + required :config, + -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config } + end + + # @!method initialize(backend_type:, config:, databricks_dtl_connection_id:, table_id:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType] + # @param config [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config] + # @param databricks_dtl_connection_id [String, nil] + # @param table_id [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL = :databricks_dtl + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + request_only do + # @!attribute inference_id_column_name + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + # + # @return [String, nil] + optional :inference_id_column_name, String, api_name: :inferenceIdColumnName, nil?: true + end + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, inference_id_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param inference_id_column_name [String, nil] Name of the column with the inference ids. This is useful if you want to update + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType + }, + api_name: :backendType + + # @!attribute redshift_connection_id + # + # @return [String, nil] + required :redshift_connection_id, String, api_name: :redshiftConnectionId, nil?: true + + # @!attribute schema_name + # + # @return [String] + required :schema_name, String, api_name: :schemaName + + # @!attribute table_name + # + # @return [String] + required :table_name, String, api_name: :tableName + + request_only do + # @!attribute config + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config] + required :config, + -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config } + end + + # @!method initialize(backend_type:, config:, redshift_connection_id:, schema_name:, table_name:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType] + # @param config [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config] + # @param redshift_connection_id [String, nil] + # @param schema_name [String] + # @param table_name [String] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT = :redshift + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + request_only do + # @!attribute inference_id_column_name + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + # + # @return [String, nil] + optional :inference_id_column_name, String, api_name: :inferenceIdColumnName, nil?: true + end + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, inference_id_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param inference_id_column_name [String, nil] Name of the column with the inference ids. This is useful if you want to update + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType] + required :backend_type, + enum: -> { + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType + }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute postgres_connection_id + # + # @return [String, nil] + required :postgres_connection_id, String, api_name: :postgresConnectionId, nil?: true + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + request_only do + # @!attribute config + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config] + required :config, + -> { Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config } + end + + # @!method initialize(backend_type:, config:, database:, postgres_connection_id:, schema:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType] + # @param config [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config] + # @param database [String] + # @param postgres_connection_id [String, nil] + # @param schema [String] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES = :postgres + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + request_only do + # @!attribute inference_id_column_name + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + # + # @return [String, nil] + optional :inference_id_column_name, String, api_name: :inferenceIdColumnName, nil?: true + end + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, inference_id_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param inference_id_column_name [String, nil] Name of the column with the inference ids. This is useful if you want to update + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + # @!method self.variants + # @return [Array(Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5)] + end + class Project < Openlayer::Internal::Type::BaseModel # @!attribute name # The project name. diff --git a/lib/openlayer/models/projects/inference_pipeline_create_response.rb b/lib/openlayer/models/projects/inference_pipeline_create_response.rb index 5019e9a..46f9dc3 100644 --- a/lib/openlayer/models/projects/inference_pipeline_create_response.rb +++ b/lib/openlayer/models/projects/inference_pipeline_create_response.rb @@ -17,6 +17,14 @@ class InferencePipelineCreateResponse < Openlayer::Internal::Type::BaseModel # @return [String] required :name, String + # @!attribute data_backend + # + # @return [Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5, nil] + optional :data_backend, + union: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend }, + api_name: :dataBackend, + nil?: true + # @!attribute project # # @return [Openlayer::Models::Projects::InferencePipelineCreateResponse::Project, nil] @@ -109,6 +117,18 @@ class InferencePipelineCreateResponse < Openlayer::Internal::Type::BaseModel # @return [Integer] required :total_goal_count, Integer, api_name: :totalGoalCount + # @!attribute date_last_polled + # The last time the data was polled. + # + # @return [Time, nil] + optional :date_last_polled, Time, api_name: :dateLastPolled, nil?: true + + # @!attribute total_records_count + # The total number of records in the data backend. + # + # @return [Integer, nil] + optional :total_records_count, Integer, api_name: :totalRecordsCount, nil?: true + # @!attribute workspace_id # The workspace id. # @@ -116,7 +136,7 @@ class InferencePipelineCreateResponse < Openlayer::Internal::Type::BaseModel optional :workspace_id, String, api_name: :workspaceId end - # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, project: nil, workspace: nil, workspace_id: nil) + # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, data_backend: nil, date_last_polled: nil, project: nil, total_records_count: nil, workspace: nil, workspace_id: nil) # @param id [String] The inference pipeline id. # # @param date_created [Time] The creation date. @@ -147,8 +167,14 @@ class InferencePipelineCreateResponse < Openlayer::Internal::Type::BaseModel # # @param total_goal_count [Integer] The total number of tests. # + # @param data_backend [Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5, nil] + # + # @param date_last_polled [Time, nil] The last time the data was polled. + # # @param project [Openlayer::Models::Projects::InferencePipelineCreateResponse::Project, nil] # + # @param total_records_count [Integer, nil] The total number of records in the data backend. + # # @param workspace [Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace, nil] # # @param workspace_id [String] The workspace id. @@ -181,6 +207,486 @@ module Status # @return [Array] end + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse#data_backend + module DataBackend + extend Openlayer::Internal::Type::Union + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0 } + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType } + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2 } + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3 } + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4 } + + variant -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5 } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType }, + api_name: :backendType + + # @!attribute bigquery_connection_id + # + # @return [String, nil] + required :bigquery_connection_id, String, api_name: :bigqueryConnectionId, nil?: true + + # @!attribute dataset_id + # + # @return [String] + required :dataset_id, String, api_name: :datasetId + + # @!attribute project_id + # + # @return [String] + required :project_id, String, api_name: :projectId + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!attribute partition_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType, nil] + optional :partition_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType }, + api_name: :partitionType, + nil?: true + + # @!method initialize(backend_type:, bigquery_connection_id:, dataset_id:, project_id:, table_id:, partition_type: nil) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType] + # @param bigquery_connection_id [String, nil] + # @param dataset_id [String] + # @param project_id [String] + # @param table_id [String, nil] + # @param partition_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY = :bigquery + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0#partition_type + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY = :DAY + MONTH = :MONTH + YEAR = :YEAR + + # @!method self.values + # @return [Array] + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType }, + api_name: :backendType + + # @!method initialize(backend_type:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT = :default + + # @!method self.values + # @return [Array] + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute snowflake_connection_id + # + # @return [String, nil] + required :snowflake_connection_id, String, api_name: :snowflakeConnectionId, nil?: true + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, schema:, snowflake_connection_id:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType] + # @param database [String] + # @param schema [String] + # @param snowflake_connection_id [String, nil] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE = :snowflake + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType }, + api_name: :backendType + + # @!attribute databricks_dtl_connection_id + # + # @return [String, nil] + required :databricks_dtl_connection_id, String, api_name: :databricksDtlConnectionId, nil?: true + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!method initialize(backend_type:, databricks_dtl_connection_id:, table_id:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType] + # @param databricks_dtl_connection_id [String, nil] + # @param table_id [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL = :databricks_dtl + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType }, + api_name: :backendType + + # @!attribute redshift_connection_id + # + # @return [String, nil] + required :redshift_connection_id, String, api_name: :redshiftConnectionId, nil?: true + + # @!attribute schema_name + # + # @return [String] + required :schema_name, String, api_name: :schemaName + + # @!attribute table_name + # + # @return [String] + required :table_name, String, api_name: :tableName + + # @!method initialize(backend_type:, redshift_connection_id:, schema_name:, table_name:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType] + # @param redshift_connection_id [String, nil] + # @param schema_name [String] + # @param table_name [String] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT = :redshift + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute postgres_connection_id + # + # @return [String, nil] + required :postgres_connection_id, String, api_name: :postgresConnectionId, nil?: true + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, postgres_connection_id:, schema:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType] + # @param database [String] + # @param postgres_connection_id [String, nil] + # @param schema [String] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES = :postgres + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + # @!method self.variants + # @return [Array(Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5)] + end + # @see Openlayer::Models::Projects::InferencePipelineCreateResponse#project class Project < Openlayer::Internal::Type::BaseModel # @!attribute name diff --git a/lib/openlayer/models/projects/inference_pipeline_list_response.rb b/lib/openlayer/models/projects/inference_pipeline_list_response.rb index 963a897..853d77e 100644 --- a/lib/openlayer/models/projects/inference_pipeline_list_response.rb +++ b/lib/openlayer/models/projects/inference_pipeline_list_response.rb @@ -27,6 +27,14 @@ class Item < Openlayer::Internal::Type::BaseModel # @return [String] required :name, String + # @!attribute data_backend + # + # @return [Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5, nil] + optional :data_backend, + union: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend }, + api_name: :dataBackend, + nil?: true + # @!attribute project # # @return [Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project, nil] @@ -119,6 +127,18 @@ class Item < Openlayer::Internal::Type::BaseModel # @return [Integer] required :total_goal_count, Integer, api_name: :totalGoalCount + # @!attribute date_last_polled + # The last time the data was polled. + # + # @return [Time, nil] + optional :date_last_polled, Time, api_name: :dateLastPolled, nil?: true + + # @!attribute total_records_count + # The total number of records in the data backend. + # + # @return [Integer, nil] + optional :total_records_count, Integer, api_name: :totalRecordsCount, nil?: true + # @!attribute workspace_id # The workspace id. # @@ -126,7 +146,7 @@ class Item < Openlayer::Internal::Type::BaseModel optional :workspace_id, String, api_name: :workspaceId end - # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, project: nil, workspace: nil, workspace_id: nil) + # @!method initialize(id:, date_created:, date_last_evaluated:, date_last_sample_received:, date_of_next_evaluation:, date_updated:, description:, failing_goal_count:, links:, name:, passing_goal_count:, project_id:, status:, status_message:, total_goal_count:, data_backend: nil, date_last_polled: nil, project: nil, total_records_count: nil, workspace: nil, workspace_id: nil) # @param id [String] The inference pipeline id. # # @param date_created [Time] The creation date. @@ -157,8 +177,14 @@ class Item < Openlayer::Internal::Type::BaseModel # # @param total_goal_count [Integer] The total number of tests. # + # @param data_backend [Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5, nil] + # + # @param date_last_polled [Time, nil] The last time the data was polled. + # # @param project [Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project, nil] # + # @param total_records_count [Integer, nil] The total number of records in the data backend. + # # @param workspace [Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace, nil] # # @param workspace_id [String] The workspace id. @@ -191,6 +217,486 @@ module Status # @return [Array] end + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item#data_backend + module DataBackend + extend Openlayer::Internal::Type::Union + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0 } + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType } + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2 } + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3 } + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4 } + + variant -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5 } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType }, + api_name: :backendType + + # @!attribute bigquery_connection_id + # + # @return [String, nil] + required :bigquery_connection_id, String, api_name: :bigqueryConnectionId, nil?: true + + # @!attribute dataset_id + # + # @return [String] + required :dataset_id, String, api_name: :datasetId + + # @!attribute project_id + # + # @return [String] + required :project_id, String, api_name: :projectId + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!attribute partition_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType, nil] + optional :partition_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType }, + api_name: :partitionType, + nil?: true + + # @!method initialize(backend_type:, bigquery_connection_id:, dataset_id:, project_id:, table_id:, partition_type: nil) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType] + # @param bigquery_connection_id [String, nil] + # @param dataset_id [String] + # @param project_id [String] + # @param table_id [String, nil] + # @param partition_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType, nil] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY = :bigquery + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0#partition_type + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY = :DAY + MONTH = :MONTH + YEAR = :YEAR + + # @!method self.values + # @return [Array] + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType }, + api_name: :backendType + + # @!method initialize(backend_type:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT = :default + + # @!method self.values + # @return [Array] + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute snowflake_connection_id + # + # @return [String, nil] + required :snowflake_connection_id, String, api_name: :snowflakeConnectionId, nil?: true + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, schema:, snowflake_connection_id:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType] + # @param database [String] + # @param schema [String] + # @param snowflake_connection_id [String, nil] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE = :snowflake + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType }, + api_name: :backendType + + # @!attribute databricks_dtl_connection_id + # + # @return [String, nil] + required :databricks_dtl_connection_id, String, api_name: :databricksDtlConnectionId, nil?: true + + # @!attribute table_id + # + # @return [String, nil] + required :table_id, String, api_name: :tableId, nil?: true + + # @!method initialize(backend_type:, databricks_dtl_connection_id:, table_id:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType] + # @param databricks_dtl_connection_id [String, nil] + # @param table_id [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL = :databricks_dtl + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType }, + api_name: :backendType + + # @!attribute redshift_connection_id + # + # @return [String, nil] + required :redshift_connection_id, String, api_name: :redshiftConnectionId, nil?: true + + # @!attribute schema_name + # + # @return [String] + required :schema_name, String, api_name: :schemaName + + # @!attribute table_name + # + # @return [String] + required :table_name, String, api_name: :tableName + + # @!method initialize(backend_type:, redshift_connection_id:, schema_name:, table_name:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType] + # @param redshift_connection_id [String, nil] + # @param schema_name [String] + # @param table_name [String] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT = :redshift + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + # @!attribute backend_type + # + # @return [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType] + required :backend_type, + enum: -> { Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType }, + api_name: :backendType + + # @!attribute database + # + # @return [String] + required :database, String + + # @!attribute postgres_connection_id + # + # @return [String, nil] + required :postgres_connection_id, String, api_name: :postgresConnectionId, nil?: true + + # @!attribute schema + # + # @return [String] + required :schema, String + + # @!attribute table + # + # @return [String, nil] + required :table, String, nil?: true + + # @!method initialize(backend_type:, database:, postgres_connection_id:, schema:, table:) + # @param backend_type [Symbol, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType] + # @param database [String] + # @param postgres_connection_id [String, nil] + # @param schema [String] + # @param table [String, nil] + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5#backend_type + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES = :postgres + + # @!method self.values + # @return [Array] + end + + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5#config + class Config < Openlayer::Internal::Type::BaseModel + # @!attribute ground_truth_column_name + # Name of the column with the ground truths. + # + # @return [String, nil] + optional :ground_truth_column_name, String, api_name: :groundTruthColumnName, nil?: true + + # @!attribute human_feedback_column_name + # Name of the column with human feedback. + # + # @return [String, nil] + optional :human_feedback_column_name, String, api_name: :humanFeedbackColumnName, nil?: true + + # @!attribute latency_column_name + # Name of the column with the latencies. + # + # @return [String, nil] + optional :latency_column_name, String, api_name: :latencyColumnName, nil?: true + + # @!attribute timestamp_column_name + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + # + # @return [String, nil] + optional :timestamp_column_name, String, api_name: :timestampColumnName, nil?: true + + # @!method initialize(ground_truth_column_name: nil, human_feedback_column_name: nil, latency_column_name: nil, timestamp_column_name: nil) + # Some parameter documentations has been truncated, see + # {Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::Config} + # for more details. + # + # @param ground_truth_column_name [String, nil] Name of the column with the ground truths. + # + # @param human_feedback_column_name [String, nil] Name of the column with human feedback. + # + # @param latency_column_name [String, nil] Name of the column with the latencies. + # + # @param timestamp_column_name [String, nil] Name of the column with the timestamps. Timestamps must be in UNIX sec format. I + end + end + + # @!method self.variants + # @return [Array(Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5)] + end + # @see Openlayer::Models::Projects::InferencePipelineListResponse::Item#project class Project < Openlayer::Internal::Type::BaseModel # @!attribute name diff --git a/lib/openlayer/models/projects/test_create_params.rb b/lib/openlayer/models/projects/test_create_params.rb index 13032d6..a5fc9d3 100644 --- a/lib/openlayer/models/projects/test_create_params.rb +++ b/lib/openlayer/models/projects/test_create_params.rb @@ -44,6 +44,16 @@ class TestCreateParams < Openlayer::Internal::Type::BaseModel # @return [Boolean, nil] optional :archived, Openlayer::Internal::Type::Boolean + # @!attribute default_to_all_pipelines + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + # + # @return [Boolean, nil] + optional :default_to_all_pipelines, + Openlayer::Internal::Type::Boolean, + api_name: :defaultToAllPipelines, + nil?: true + # @!attribute delay_window # The delay window in seconds. Only applies to tests that use production data. # @@ -57,6 +67,36 @@ class TestCreateParams < Openlayer::Internal::Type::BaseModel # @return [Float, nil] optional :evaluation_window, Float, api_name: :evaluationWindow, nil?: true + # @!attribute exclude_pipelines + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :exclude_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :excludePipelines, + nil?: true + + # @!attribute include_historical_data + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + # + # @return [Boolean, nil] + optional :include_historical_data, + Openlayer::Internal::Type::Boolean, + api_name: :includeHistoricalData, + nil?: true + + # @!attribute include_pipelines + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :include_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :includePipelines, + nil?: true + # @!attribute uses_ml_model # Whether the test uses an ML model. # @@ -89,7 +129,7 @@ class TestCreateParams < Openlayer::Internal::Type::BaseModel Openlayer::Internal::Type::Boolean, api_name: :usesValidationDataset - # @!method initialize(description:, name:, subtype:, thresholds:, type:, archived: nil, delay_window: nil, evaluation_window: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil, request_options: {}) + # @!method initialize(description:, name:, subtype:, thresholds:, type:, archived: nil, default_to_all_pipelines: nil, delay_window: nil, evaluation_window: nil, exclude_pipelines: nil, include_historical_data: nil, include_pipelines: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil, request_options: {}) # Some parameter documentations has been truncated, see # {Openlayer::Models::Projects::TestCreateParams} for more details. # @@ -105,10 +145,18 @@ class TestCreateParams < Openlayer::Internal::Type::BaseModel # # @param archived [Boolean] Whether the test is archived. # + # @param default_to_all_pipelines [Boolean, nil] Whether to apply the test to all pipelines (data sources) or to a specific set o + # # @param delay_window [Float, nil] The delay window in seconds. Only applies to tests that use production data. # # @param evaluation_window [Float, nil] The evaluation window in seconds. Only applies to tests that use production data # + # @param exclude_pipelines [Array, nil] Array of pipelines (data sources) to which the test should not be applied. Only + # + # @param include_historical_data [Boolean, nil] Whether to include historical data in the test result. Only applies to tests tha + # + # @param include_pipelines [Array, nil] Array of pipelines (data sources) to which the test should be applied. Only appl + # # @param uses_ml_model [Boolean] Whether the test uses an ML model. # # @param uses_production_data [Boolean] Whether the test uses production data (monitoring mode only). diff --git a/lib/openlayer/models/projects/test_create_response.rb b/lib/openlayer/models/projects/test_create_response.rb index b89d1b6..8f7f470 100644 --- a/lib/openlayer/models/projects/test_create_response.rb +++ b/lib/openlayer/models/projects/test_create_response.rb @@ -41,6 +41,16 @@ class TestCreateResponse < Openlayer::Internal::Type::BaseModel # @return [Boolean, nil] optional :archived, Openlayer::Internal::Type::Boolean + # @!attribute default_to_all_pipelines + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + # + # @return [Boolean, nil] + optional :default_to_all_pipelines, + Openlayer::Internal::Type::Boolean, + api_name: :defaultToAllPipelines, + nil?: true + # @!attribute delay_window # The delay window in seconds. Only applies to tests that use production data. # @@ -54,6 +64,36 @@ class TestCreateResponse < Openlayer::Internal::Type::BaseModel # @return [Float, nil] optional :evaluation_window, Float, api_name: :evaluationWindow, nil?: true + # @!attribute exclude_pipelines + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :exclude_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :excludePipelines, + nil?: true + + # @!attribute include_historical_data + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + # + # @return [Boolean, nil] + optional :include_historical_data, + Openlayer::Internal::Type::Boolean, + api_name: :includeHistoricalData, + nil?: true + + # @!attribute include_pipelines + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :include_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :includePipelines, + nil?: true + # @!attribute uses_ml_model # Whether the test uses an ML model. # @@ -142,7 +182,7 @@ class TestCreateResponse < Openlayer::Internal::Type::BaseModel required :suggested, Openlayer::Internal::Type::Boolean end - # @!method initialize(id:, comment_count:, creator_id:, date_archived:, date_created:, date_updated:, description:, name:, number:, origin_project_version_id:, subtype:, suggested:, thresholds:, type:, archived: nil, delay_window: nil, evaluation_window: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil) + # @!method initialize(id:, comment_count:, creator_id:, date_archived:, date_created:, date_updated:, description:, name:, number:, origin_project_version_id:, subtype:, suggested:, thresholds:, type:, archived: nil, default_to_all_pipelines: nil, delay_window: nil, evaluation_window: nil, exclude_pipelines: nil, include_historical_data: nil, include_pipelines: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil) # Some parameter documentations has been truncated, see # {Openlayer::Models::Projects::TestCreateResponse} for more details. # @@ -176,10 +216,18 @@ class TestCreateResponse < Openlayer::Internal::Type::BaseModel # # @param archived [Boolean] Whether the test is archived. # + # @param default_to_all_pipelines [Boolean, nil] Whether to apply the test to all pipelines (data sources) or to a specific set o + # # @param delay_window [Float, nil] The delay window in seconds. Only applies to tests that use production data. # # @param evaluation_window [Float, nil] The evaluation window in seconds. Only applies to tests that use production data # + # @param exclude_pipelines [Array, nil] Array of pipelines (data sources) to which the test should not be applied. Only + # + # @param include_historical_data [Boolean, nil] Whether to include historical data in the test result. Only applies to tests tha + # + # @param include_pipelines [Array, nil] Array of pipelines (data sources) to which the test should be applied. Only appl + # # @param uses_ml_model [Boolean] Whether the test uses an ML model. # # @param uses_production_data [Boolean] Whether the test uses production data (monitoring mode only). diff --git a/lib/openlayer/models/projects/test_list_response.rb b/lib/openlayer/models/projects/test_list_response.rb index bb28ce0..3339944 100644 --- a/lib/openlayer/models/projects/test_list_response.rb +++ b/lib/openlayer/models/projects/test_list_response.rb @@ -51,6 +51,16 @@ class Item < Openlayer::Internal::Type::BaseModel # @return [Boolean, nil] optional :archived, Openlayer::Internal::Type::Boolean + # @!attribute default_to_all_pipelines + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + # + # @return [Boolean, nil] + optional :default_to_all_pipelines, + Openlayer::Internal::Type::Boolean, + api_name: :defaultToAllPipelines, + nil?: true + # @!attribute delay_window # The delay window in seconds. Only applies to tests that use production data. # @@ -64,6 +74,36 @@ class Item < Openlayer::Internal::Type::BaseModel # @return [Float, nil] optional :evaluation_window, Float, api_name: :evaluationWindow, nil?: true + # @!attribute exclude_pipelines + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :exclude_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :excludePipelines, + nil?: true + + # @!attribute include_historical_data + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + # + # @return [Boolean, nil] + optional :include_historical_data, + Openlayer::Internal::Type::Boolean, + api_name: :includeHistoricalData, + nil?: true + + # @!attribute include_pipelines + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + # + # @return [Array, nil] + optional :include_pipelines, + Openlayer::Internal::Type::ArrayOf[String], + api_name: :includePipelines, + nil?: true + # @!attribute uses_ml_model # Whether the test uses an ML model. # @@ -154,7 +194,7 @@ class Item < Openlayer::Internal::Type::BaseModel required :suggested, Openlayer::Internal::Type::Boolean end - # @!method initialize(id:, comment_count:, creator_id:, date_archived:, date_created:, date_updated:, description:, name:, number:, origin_project_version_id:, subtype:, suggested:, thresholds:, type:, archived: nil, delay_window: nil, evaluation_window: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil) + # @!method initialize(id:, comment_count:, creator_id:, date_archived:, date_created:, date_updated:, description:, name:, number:, origin_project_version_id:, subtype:, suggested:, thresholds:, type:, archived: nil, default_to_all_pipelines: nil, delay_window: nil, evaluation_window: nil, exclude_pipelines: nil, include_historical_data: nil, include_pipelines: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil) # Some parameter documentations has been truncated, see # {Openlayer::Models::Projects::TestListResponse::Item} for more details. # @@ -188,10 +228,18 @@ class Item < Openlayer::Internal::Type::BaseModel # # @param archived [Boolean] Whether the test is archived. # + # @param default_to_all_pipelines [Boolean, nil] Whether to apply the test to all pipelines (data sources) or to a specific set o + # # @param delay_window [Float, nil] The delay window in seconds. Only applies to tests that use production data. # # @param evaluation_window [Float, nil] The evaluation window in seconds. Only applies to tests that use production data # + # @param exclude_pipelines [Array, nil] Array of pipelines (data sources) to which the test should not be applied. Only + # + # @param include_historical_data [Boolean, nil] Whether to include historical data in the test result. Only applies to tests tha + # + # @param include_pipelines [Array, nil] Array of pipelines (data sources) to which the test should be applied. Only appl + # # @param uses_ml_model [Boolean] Whether the test uses an ML model. # # @param uses_production_data [Boolean] Whether the test uses production data (monitoring mode only). diff --git a/lib/openlayer/resources/projects/inference_pipelines.rb b/lib/openlayer/resources/projects/inference_pipelines.rb index 41ec052..8435746 100644 --- a/lib/openlayer/resources/projects/inference_pipelines.rb +++ b/lib/openlayer/resources/projects/inference_pipelines.rb @@ -6,7 +6,7 @@ class Projects class InferencePipelines # Create an inference pipeline in a project. # - # @overload create(project_id, description:, name:, project: nil, workspace: nil, request_options: {}) + # @overload create(project_id, description:, name:, data_backend: nil, project: nil, workspace: nil, request_options: {}) # # @param project_id [String] The project id. # @@ -14,6 +14,8 @@ class InferencePipelines # # @param name [String] The inference pipeline name. # + # @param data_backend [Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5, nil] + # # @param project [Openlayer::Models::Projects::InferencePipelineCreateParams::Project, nil] # # @param workspace [Openlayer::Models::Projects::InferencePipelineCreateParams::Workspace, nil] diff --git a/lib/openlayer/resources/projects/tests.rb b/lib/openlayer/resources/projects/tests.rb index d5792a8..c11e4ee 100644 --- a/lib/openlayer/resources/projects/tests.rb +++ b/lib/openlayer/resources/projects/tests.rb @@ -9,7 +9,7 @@ class Tests # # Create a test. # - # @overload create(project_id, description:, name:, subtype:, thresholds:, type:, archived: nil, delay_window: nil, evaluation_window: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil, request_options: {}) + # @overload create(project_id, description:, name:, subtype:, thresholds:, type:, archived: nil, default_to_all_pipelines: nil, delay_window: nil, evaluation_window: nil, exclude_pipelines: nil, include_historical_data: nil, include_pipelines: nil, uses_ml_model: nil, uses_production_data: nil, uses_reference_dataset: nil, uses_training_dataset: nil, uses_validation_dataset: nil, request_options: {}) # # @param project_id [String] The project id. # @@ -25,10 +25,18 @@ class Tests # # @param archived [Boolean] Whether the test is archived. # + # @param default_to_all_pipelines [Boolean, nil] Whether to apply the test to all pipelines (data sources) or to a specific set o + # # @param delay_window [Float, nil] The delay window in seconds. Only applies to tests that use production data. # # @param evaluation_window [Float, nil] The evaluation window in seconds. Only applies to tests that use production data # + # @param exclude_pipelines [Array, nil] Array of pipelines (data sources) to which the test should not be applied. Only + # + # @param include_historical_data [Boolean, nil] Whether to include historical data in the test result. Only applies to tests tha + # + # @param include_pipelines [Array, nil] Array of pipelines (data sources) to which the test should be applied. Only appl + # # @param uses_ml_model [Boolean] Whether the test uses an ML model. # # @param uses_production_data [Boolean] Whether the test uses production data (monitoring mode only). diff --git a/lib/openlayer/version.rb b/lib/openlayer/version.rb index 9bdc821..6934c4f 100644 --- a/lib/openlayer/version.rb +++ b/lib/openlayer/version.rb @@ -1,5 +1,5 @@ # frozen_string_literal: true module Openlayer - VERSION = "0.4.1" + VERSION = "0.5.0" end diff --git a/rbi/openlayer/models/inference_pipeline_retrieve_response.rbi b/rbi/openlayer/models/inference_pipeline_retrieve_response.rbi index 2b36c93..79da578 100644 --- a/rbi/openlayer/models/inference_pipeline_retrieve_response.rbi +++ b/rbi/openlayer/models/inference_pipeline_retrieve_response.rbi @@ -19,6 +19,15 @@ module Openlayer sig { returns(String) } attr_accessor :name + sig do + returns( + T.nilable( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::Variants + ) + ) + end + attr_accessor :data_backend + sig do returns( T.nilable( @@ -122,6 +131,14 @@ module Openlayer sig { returns(Integer) } attr_accessor :total_goal_count + # The last time the data was polled. + sig { returns(T.nilable(Time)) } + attr_accessor :date_last_polled + + # The total number of records in the data backend. + sig { returns(T.nilable(Integer)) } + attr_accessor :total_records_count + # The workspace id. sig { returns(T.nilable(String)) } attr_reader :workspace_id @@ -148,10 +165,23 @@ module Openlayer Openlayer::Models::InferencePipelineRetrieveResponse::Status::OrSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::OrHash, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::OrHash, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::OrHash, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::OrHash, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::OrHash, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::OrHash + ) + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::InferencePipelineRetrieveResponse::Project::OrHash ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::InferencePipelineRetrieveResponse::Workspace::OrHash @@ -189,7 +219,12 @@ module Openlayer status_message:, # The total number of tests. total_goal_count:, + data_backend: nil, + # The last time the data was polled. + date_last_polled: nil, project: nil, + # The total number of records in the data backend. + total_records_count: nil, workspace: nil, # The workspace id. workspace_id: nil @@ -215,10 +250,16 @@ module Openlayer Openlayer::Models::InferencePipelineRetrieveResponse::Status::TaggedSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::Variants + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::InferencePipelineRetrieveResponse::Project ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::InferencePipelineRetrieveResponse::Workspace @@ -306,6 +347,913 @@ module Openlayer end end + module DataBackend + extend Openlayer::Internal::Type::Union + + Variants = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5 + ) + end + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :bigquery_connection_id + + sig { returns(String) } + attr_accessor :dataset_id + + sig { returns(String) } + attr_accessor :project_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + T.nilable( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + ) + end + attr_accessor :partition_type + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + bigquery_connection_id:, + dataset_id:, + project_id:, + table_id:, + partition_type: nil + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + BIGQUERY = + T.let( + :bigquery, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + + module PartitionType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DAY = + T.let( + :DAY, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + MONTH = + T.let( + :MONTH, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + YEAR = + T.let( + :YEAR, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType::OrSymbol + ).returns(T.attached_class) + end + def self.new(backend_type:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType::TaggedSymbol + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DEFAULT = + T.let( + :default, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :snowflake_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType::OrSymbol, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + schema:, + snowflake_connection_id:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + SNOWFLAKE = + T.let( + :snowflake, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :databricks_dtl_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType::OrSymbol, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + ).returns(T.attached_class) + end + def self.new(backend_type:, databricks_dtl_connection_id:, table_id:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DATABRICKS_DTL = + T.let( + :databricks_dtl, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :redshift_connection_id + + sig { returns(String) } + attr_accessor :schema_name + + sig { returns(String) } + attr_accessor :table_name + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType::OrSymbol, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + ).returns(T.attached_class) + end + def self.new( + backend_type:, + redshift_connection_id:, + schema_name:, + table_name: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + REDSHIFT = + T.let( + :redshift, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(T.nilable(String)) } + attr_accessor :postgres_connection_id + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType::OrSymbol, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + postgres_connection_id:, + schema:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + POSTGRES = + T.let( + :postgres, + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::Variants + ] + ) + end + def self.variants + end + end + class Project < Openlayer::Internal::Type::BaseModel OrHash = T.type_alias do diff --git a/rbi/openlayer/models/inference_pipeline_update_response.rbi b/rbi/openlayer/models/inference_pipeline_update_response.rbi index 00a02af..3b73471 100644 --- a/rbi/openlayer/models/inference_pipeline_update_response.rbi +++ b/rbi/openlayer/models/inference_pipeline_update_response.rbi @@ -19,6 +19,15 @@ module Openlayer sig { returns(String) } attr_accessor :name + sig do + returns( + T.nilable( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::Variants + ) + ) + end + attr_accessor :data_backend + sig do returns( T.nilable(Openlayer::Models::InferencePipelineUpdateResponse::Project) @@ -118,6 +127,14 @@ module Openlayer sig { returns(Integer) } attr_accessor :total_goal_count + # The last time the data was polled. + sig { returns(T.nilable(Time)) } + attr_accessor :date_last_polled + + # The total number of records in the data backend. + sig { returns(T.nilable(Integer)) } + attr_accessor :total_records_count + # The workspace id. sig { returns(T.nilable(String)) } attr_reader :workspace_id @@ -144,10 +161,23 @@ module Openlayer Openlayer::Models::InferencePipelineUpdateResponse::Status::OrSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::OrHash, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::OrHash, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::OrHash, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::OrHash, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::OrHash, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::OrHash + ) + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::InferencePipelineUpdateResponse::Project::OrHash ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::InferencePipelineUpdateResponse::Workspace::OrHash @@ -185,7 +215,12 @@ module Openlayer status_message:, # The total number of tests. total_goal_count:, + data_backend: nil, + # The last time the data was polled. + date_last_polled: nil, project: nil, + # The total number of records in the data backend. + total_records_count: nil, workspace: nil, # The workspace id. workspace_id: nil @@ -211,10 +246,16 @@ module Openlayer Openlayer::Models::InferencePipelineUpdateResponse::Status::TaggedSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::Variants + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::InferencePipelineUpdateResponse::Project ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::InferencePipelineUpdateResponse::Workspace @@ -302,6 +343,913 @@ module Openlayer end end + module DataBackend + extend Openlayer::Internal::Type::Union + + Variants = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5 + ) + end + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :bigquery_connection_id + + sig { returns(String) } + attr_accessor :dataset_id + + sig { returns(String) } + attr_accessor :project_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + T.nilable( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + ) + end + attr_accessor :partition_type + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + bigquery_connection_id:, + dataset_id:, + project_id:, + table_id:, + partition_type: nil + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + BIGQUERY = + T.let( + :bigquery, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + + module PartitionType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DAY = + T.let( + :DAY, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + MONTH = + T.let( + :MONTH, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + YEAR = + T.let( + :YEAR, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType::OrSymbol + ).returns(T.attached_class) + end + def self.new(backend_type:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DEFAULT = + T.let( + :default, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :snowflake_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType::OrSymbol, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + schema:, + snowflake_connection_id:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + SNOWFLAKE = + T.let( + :snowflake, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :databricks_dtl_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType::OrSymbol, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + ).returns(T.attached_class) + end + def self.new(backend_type:, databricks_dtl_connection_id:, table_id:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DATABRICKS_DTL = + T.let( + :databricks_dtl, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :redshift_connection_id + + sig { returns(String) } + attr_accessor :schema_name + + sig { returns(String) } + attr_accessor :table_name + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType::OrSymbol, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + ).returns(T.attached_class) + end + def self.new( + backend_type:, + redshift_connection_id:, + schema_name:, + table_name: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + REDSHIFT = + T.let( + :redshift, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(T.nilable(String)) } + attr_accessor :postgres_connection_id + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType::OrSymbol, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + postgres_connection_id:, + schema:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol, + config: + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + POSTGRES = + T.let( + :postgres, + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + sig do + override.returns( + T::Array[ + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::Variants + ] + ) + end + def self.variants + end + end + class Project < Openlayer::Internal::Type::BaseModel OrHash = T.type_alias do diff --git a/rbi/openlayer/models/projects/inference_pipeline_create_params.rbi b/rbi/openlayer/models/projects/inference_pipeline_create_params.rbi index cafdd34..27d0310 100644 --- a/rbi/openlayer/models/projects/inference_pipeline_create_params.rbi +++ b/rbi/openlayer/models/projects/inference_pipeline_create_params.rbi @@ -23,6 +23,22 @@ module Openlayer sig { returns(String) } attr_accessor :name + sig do + returns( + T.nilable( + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5 + ) + ) + ) + end + attr_accessor :data_backend + sig do returns( T.nilable( @@ -65,6 +81,17 @@ module Openlayer params( description: T.nilable(String), name: String, + data_backend: + T.nilable( + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::OrHash + ) + ), project: T.nilable( Openlayer::Projects::InferencePipelineCreateParams::Project::OrHash @@ -81,6 +108,7 @@ module Openlayer description:, # The inference pipeline name. name:, + data_backend: nil, project: nil, workspace: nil, request_options: {} @@ -92,6 +120,17 @@ module Openlayer { description: T.nilable(String), name: String, + data_backend: + T.nilable( + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5 + ) + ), project: T.nilable( Openlayer::Projects::InferencePipelineCreateParams::Project @@ -107,6 +146,1062 @@ module Openlayer def to_hash end + module DataBackend + extend Openlayer::Internal::Type::Union + + Variants = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5 + ) + end + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :bigquery_connection_id + + sig { returns(String) } + attr_accessor :dataset_id + + sig { returns(String) } + attr_accessor :project_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + T.nilable( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ) + end + attr_accessor :partition_type + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config + ) + end + attr_reader :config + + sig do + params( + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config::OrHash + ).void + end + attr_writer :config + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config::OrHash, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + bigquery_connection_id:, + config:, + dataset_id:, + project_id:, + table_id:, + partition_type: nil + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + BIGQUERY = + T.let( + :bigquery, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + sig { returns(T.nilable(String)) } + attr_accessor :inference_id_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + inference_id_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + + module PartitionType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DAY = + T.let( + :DAY, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + MONTH = + T.let( + :MONTH, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + YEAR = + T.let( + :YEAR, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType::OrSymbol + ).returns(T.attached_class) + end + def self.new(backend_type:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType::OrSymbol + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DEFAULT = + T.let( + :default, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :snowflake_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config + ) + end + attr_reader :config + + sig do + params( + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config::OrHash + ).void + end + attr_writer :config + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config::OrHash, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + config:, + database:, + schema:, + snowflake_connection_id:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + SNOWFLAKE = + T.let( + :snowflake, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + sig { returns(T.nilable(String)) } + attr_accessor :inference_id_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + inference_id_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :databricks_dtl_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config + ) + end + attr_reader :config + + sig do + params( + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config::OrHash + ).void + end + attr_writer :config + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config::OrHash, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + config:, + databricks_dtl_connection_id:, + table_id: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DATABRICKS_DTL = + T.let( + :databricks_dtl, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + sig { returns(T.nilable(String)) } + attr_accessor :inference_id_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + inference_id_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :redshift_connection_id + + sig { returns(String) } + attr_accessor :schema_name + + sig { returns(String) } + attr_accessor :table_name + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config + ) + end + attr_reader :config + + sig do + params( + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config::OrHash + ).void + end + attr_writer :config + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config::OrHash, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + ).returns(T.attached_class) + end + def self.new( + backend_type:, + config:, + redshift_connection_id:, + schema_name:, + table_name: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + REDSHIFT = + T.let( + :redshift, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + sig { returns(T.nilable(String)) } + attr_accessor :inference_id_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + inference_id_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType::OrSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(T.nilable(String)) } + attr_accessor :postgres_connection_id + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + returns( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config + ) + end + attr_reader :config + + sig do + params( + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config::OrHash + ).void + end + attr_writer :config + + sig do + params( + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config::OrHash, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + config:, + database:, + postgres_connection_id:, + schema:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType::OrSymbol, + config: + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + POSTGRES = + T.let( + :postgres, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + sig { returns(T.nilable(String)) } + attr_accessor :inference_id_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the inference ids. This is useful if you want to update + # rows at a later point in time. If not provided, a unique id is generated by + # Openlayer. + inference_id_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + sig do + override.returns( + T::Array[ + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::Variants + ] + ) + end + def self.variants + end + end + class Project < Openlayer::Internal::Type::BaseModel OrHash = T.type_alias do diff --git a/rbi/openlayer/models/projects/inference_pipeline_create_response.rbi b/rbi/openlayer/models/projects/inference_pipeline_create_response.rbi index 1b1a14a..8317f6f 100644 --- a/rbi/openlayer/models/projects/inference_pipeline_create_response.rbi +++ b/rbi/openlayer/models/projects/inference_pipeline_create_response.rbi @@ -20,6 +20,15 @@ module Openlayer sig { returns(String) } attr_accessor :name + sig do + returns( + T.nilable( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::Variants + ) + ) + end + attr_accessor :data_backend + sig do returns( T.nilable( @@ -125,6 +134,14 @@ module Openlayer sig { returns(Integer) } attr_accessor :total_goal_count + # The last time the data was polled. + sig { returns(T.nilable(Time)) } + attr_accessor :date_last_polled + + # The total number of records in the data backend. + sig { returns(T.nilable(Integer)) } + attr_accessor :total_records_count + # The workspace id. sig { returns(T.nilable(String)) } attr_reader :workspace_id @@ -151,10 +168,23 @@ module Openlayer Openlayer::Models::Projects::InferencePipelineCreateResponse::Status::OrSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::OrHash, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::OrHash, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::OrHash, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::OrHash, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::OrHash, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::OrHash + ) + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::Projects::InferencePipelineCreateResponse::Project::OrHash ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace::OrHash @@ -192,7 +222,12 @@ module Openlayer status_message:, # The total number of tests. total_goal_count:, + data_backend: nil, + # The last time the data was polled. + date_last_polled: nil, project: nil, + # The total number of records in the data backend. + total_records_count: nil, workspace: nil, # The workspace id. workspace_id: nil @@ -219,10 +254,16 @@ module Openlayer Openlayer::Models::Projects::InferencePipelineCreateResponse::Status::TaggedSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::Variants + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::Projects::InferencePipelineCreateResponse::Project ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace @@ -310,6 +351,917 @@ module Openlayer end end + module DataBackend + extend Openlayer::Internal::Type::Union + + Variants = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5 + ) + end + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :bigquery_connection_id + + sig { returns(String) } + attr_accessor :dataset_id + + sig { returns(String) } + attr_accessor :project_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + T.nilable( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + ) + end + attr_accessor :partition_type + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + bigquery_connection_id:, + dataset_id:, + project_id:, + table_id:, + partition_type: nil + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + BIGQUERY = + T.let( + :bigquery, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + + module PartitionType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DAY = + T.let( + :DAY, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + MONTH = + T.let( + :MONTH, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + YEAR = + T.let( + :YEAR, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType::OrSymbol + ).returns(T.attached_class) + end + def self.new(backend_type:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DEFAULT = + T.let( + :default, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :snowflake_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType::OrSymbol, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + schema:, + snowflake_connection_id:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + SNOWFLAKE = + T.let( + :snowflake, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :databricks_dtl_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType::OrSymbol, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + databricks_dtl_connection_id:, + table_id: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DATABRICKS_DTL = + T.let( + :databricks_dtl, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :redshift_connection_id + + sig { returns(String) } + attr_accessor :schema_name + + sig { returns(String) } + attr_accessor :table_name + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType::OrSymbol, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + ).returns(T.attached_class) + end + def self.new( + backend_type:, + redshift_connection_id:, + schema_name:, + table_name: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + REDSHIFT = + T.let( + :redshift, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(T.nilable(String)) } + attr_accessor :postgres_connection_id + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType::OrSymbol, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + postgres_connection_id:, + schema:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + POSTGRES = + T.let( + :postgres, + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::Variants + ] + ) + end + def self.variants + end + end + class Project < Openlayer::Internal::Type::BaseModel OrHash = T.type_alias do diff --git a/rbi/openlayer/models/projects/inference_pipeline_list_response.rbi b/rbi/openlayer/models/projects/inference_pipeline_list_response.rbi index 9fdc20a..aa0a726 100644 --- a/rbi/openlayer/models/projects/inference_pipeline_list_response.rbi +++ b/rbi/openlayer/models/projects/inference_pipeline_list_response.rbi @@ -62,6 +62,15 @@ module Openlayer sig { returns(String) } attr_accessor :name + sig do + returns( + T.nilable( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::Variants + ) + ) + end + attr_accessor :data_backend + sig do returns( T.nilable( @@ -167,6 +176,14 @@ module Openlayer sig { returns(Integer) } attr_accessor :total_goal_count + # The last time the data was polled. + sig { returns(T.nilable(Time)) } + attr_accessor :date_last_polled + + # The total number of records in the data backend. + sig { returns(T.nilable(Integer)) } + attr_accessor :total_records_count + # The workspace id. sig { returns(T.nilable(String)) } attr_reader :workspace_id @@ -193,10 +210,23 @@ module Openlayer Openlayer::Models::Projects::InferencePipelineListResponse::Item::Status::OrSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::OrHash, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::OrHash, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::OrHash, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::OrHash, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::OrHash, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::OrHash + ) + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project::OrHash ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace::OrHash @@ -234,7 +264,12 @@ module Openlayer status_message:, # The total number of tests. total_goal_count:, + data_backend: nil, + # The last time the data was polled. + date_last_polled: nil, project: nil, + # The total number of records in the data backend. + total_records_count: nil, workspace: nil, # The workspace id. workspace_id: nil @@ -261,10 +296,16 @@ module Openlayer Openlayer::Models::Projects::InferencePipelineListResponse::Item::Status::TaggedSymbol, status_message: T.nilable(String), total_goal_count: Integer, + data_backend: + T.nilable( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::Variants + ), + date_last_polled: T.nilable(Time), project: T.nilable( Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project ), + total_records_count: T.nilable(Integer), workspace: T.nilable( Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace @@ -352,6 +393,917 @@ module Openlayer end end + module DataBackend + extend Openlayer::Internal::Type::Union + + Variants = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5 + ) + end + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :bigquery_connection_id + + sig { returns(String) } + attr_accessor :dataset_id + + sig { returns(String) } + attr_accessor :project_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + returns( + T.nilable( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + ) + end + attr_accessor :partition_type + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType::OrSymbol, + bigquery_connection_id: T.nilable(String), + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::OrSymbol + ) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + bigquery_connection_id:, + dataset_id:, + project_id:, + table_id:, + partition_type: nil + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType::TaggedSymbol, + bigquery_connection_id: T.nilable(String), + config: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: T.nilable(String), + partition_type: + T.nilable( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + BIGQUERY = + T.let( + :bigquery, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + + module PartitionType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DAY = + T.let( + :DAY, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + MONTH = + T.let( + :MONTH, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + YEAR = + T.let( + :YEAR, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::PartitionType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class BackendType < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType::OrSymbol + ).returns(T.attached_class) + end + def self.new(backend_type:) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType::TaggedSymbol + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DEFAULT = + T.let( + :default, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + end + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :snowflake_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType::OrSymbol, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + schema:, + snowflake_connection_id:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: T.nilable(String), + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + SNOWFLAKE = + T.let( + :snowflake, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :databricks_dtl_connection_id + + sig { returns(T.nilable(String)) } + attr_accessor :table_id + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType::OrSymbol, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + databricks_dtl_connection_id:, + table_id: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: T.nilable(String), + table_id: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + DATABRICKS_DTL = + T.let( + :databricks_dtl, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(T.nilable(String)) } + attr_accessor :redshift_connection_id + + sig { returns(String) } + attr_accessor :schema_name + + sig { returns(String) } + attr_accessor :table_name + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType::OrSymbol, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + ).returns(T.attached_class) + end + def self.new( + backend_type:, + redshift_connection_id:, + schema_name:, + table_name: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::Config, + redshift_connection_id: T.nilable(String), + schema_name: String, + table_name: String + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + REDSHIFT = + T.let( + :redshift, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5, + Openlayer::Internal::AnyHash + ) + end + + sig do + returns( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + end + attr_accessor :backend_type + + sig { returns(String) } + attr_accessor :database + + sig { returns(T.nilable(String)) } + attr_accessor :postgres_connection_id + + sig { returns(String) } + attr_accessor :schema + + sig { returns(T.nilable(String)) } + attr_accessor :table + + sig do + params( + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType::OrSymbol, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + backend_type:, + database:, + postgres_connection_id:, + schema:, + table: + ) + end + + sig do + override.returns( + { + backend_type: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType::TaggedSymbol, + config: + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: T.nilable(String), + schema: String, + table: T.nilable(String) + } + ) + end + def to_hash + end + + module BackendType + extend Openlayer::Internal::Type::Enum + + TaggedSymbol = + T.type_alias do + T.all( + Symbol, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType + ) + end + OrSymbol = T.type_alias { T.any(Symbol, String) } + + POSTGRES = + T.let( + :postgres, + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType::TaggedSymbol + ) + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::BackendType::TaggedSymbol + ] + ) + end + def self.values + end + end + + class Config < Openlayer::Internal::Type::BaseModel + OrHash = + T.type_alias do + T.any( + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::Config, + Openlayer::Internal::AnyHash + ) + end + + # Name of the column with the ground truths. + sig { returns(T.nilable(String)) } + attr_accessor :ground_truth_column_name + + # Name of the column with human feedback. + sig { returns(T.nilable(String)) } + attr_accessor :human_feedback_column_name + + # Name of the column with the latencies. + sig { returns(T.nilable(String)) } + attr_accessor :latency_column_name + + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + sig { returns(T.nilable(String)) } + attr_accessor :timestamp_column_name + + sig do + params( + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + ).returns(T.attached_class) + end + def self.new( + # Name of the column with the ground truths. + ground_truth_column_name: nil, + # Name of the column with human feedback. + human_feedback_column_name: nil, + # Name of the column with the latencies. + latency_column_name: nil, + # Name of the column with the timestamps. Timestamps must be in UNIX sec format. + # If not provided, the upload timestamp is used. + timestamp_column_name: nil + ) + end + + sig do + override.returns( + { + ground_truth_column_name: T.nilable(String), + human_feedback_column_name: T.nilable(String), + inference_id_column_name: T.nilable(String), + latency_column_name: T.nilable(String), + timestamp_column_name: T.nilable(String) + } + ) + end + def to_hash + end + end + end + + sig do + override.returns( + T::Array[ + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::Variants + ] + ) + end + def self.variants + end + end + class Project < Openlayer::Internal::Type::BaseModel OrHash = T.type_alias do diff --git a/rbi/openlayer/models/projects/test_create_params.rbi b/rbi/openlayer/models/projects/test_create_params.rbi index 84edec2..9b917a2 100644 --- a/rbi/openlayer/models/projects/test_create_params.rbi +++ b/rbi/openlayer/models/projects/test_create_params.rbi @@ -45,6 +45,11 @@ module Openlayer sig { params(archived: T::Boolean).void } attr_writer :archived + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :default_to_all_pipelines + # The delay window in seconds. Only applies to tests that use production data. sig { returns(T.nilable(Float)) } attr_accessor :delay_window @@ -54,6 +59,21 @@ module Openlayer sig { returns(T.nilable(Float)) } attr_accessor :evaluation_window + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :exclude_pipelines + + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :include_historical_data + + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :include_pipelines + # Whether the test uses an ML model. sig { returns(T.nilable(T::Boolean)) } attr_reader :uses_ml_model @@ -100,8 +120,12 @@ module Openlayer ], type: Openlayer::Projects::TestCreateParams::Type::OrSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, @@ -122,11 +146,23 @@ module Openlayer type:, # Whether the test is archived. archived: nil, + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + default_to_all_pipelines: nil, # The delay window in seconds. Only applies to tests that use production data. delay_window: nil, # The evaluation window in seconds. Only applies to tests that use production # data. evaluation_window: nil, + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + exclude_pipelines: nil, + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + include_historical_data: nil, + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + include_pipelines: nil, # Whether the test uses an ML model. uses_ml_model: nil, # Whether the test uses production data (monitoring mode only). @@ -151,8 +187,12 @@ module Openlayer T::Array[Openlayer::Projects::TestCreateParams::Threshold], type: Openlayer::Projects::TestCreateParams::Type::OrSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, diff --git a/rbi/openlayer/models/projects/test_create_response.rbi b/rbi/openlayer/models/projects/test_create_response.rbi index 954a6ae..8a0a95f 100644 --- a/rbi/openlayer/models/projects/test_create_response.rbi +++ b/rbi/openlayer/models/projects/test_create_response.rbi @@ -50,6 +50,11 @@ module Openlayer sig { params(archived: T::Boolean).void } attr_writer :archived + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :default_to_all_pipelines + # The delay window in seconds. Only applies to tests that use production data. sig { returns(T.nilable(Float)) } attr_accessor :delay_window @@ -59,6 +64,21 @@ module Openlayer sig { returns(T.nilable(Float)) } attr_accessor :evaluation_window + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :exclude_pipelines + + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :include_historical_data + + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :include_pipelines + # Whether the test uses an ML model. sig { returns(T.nilable(T::Boolean)) } attr_reader :uses_ml_model @@ -152,8 +172,12 @@ module Openlayer type: Openlayer::Models::Projects::TestCreateResponse::Type::OrSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, @@ -191,11 +215,23 @@ module Openlayer type:, # Whether the test is archived. archived: nil, + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + default_to_all_pipelines: nil, # The delay window in seconds. Only applies to tests that use production data. delay_window: nil, # The evaluation window in seconds. Only applies to tests that use production # data. evaluation_window: nil, + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + exclude_pipelines: nil, + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + include_historical_data: nil, + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + include_pipelines: nil, # Whether the test uses an ML model. uses_ml_model: nil, # Whether the test uses production data (monitoring mode only). @@ -232,8 +268,12 @@ module Openlayer type: Openlayer::Models::Projects::TestCreateResponse::Type::TaggedSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, diff --git a/rbi/openlayer/models/projects/test_list_response.rbi b/rbi/openlayer/models/projects/test_list_response.rbi index c77189e..5ba97ac 100644 --- a/rbi/openlayer/models/projects/test_list_response.rbi +++ b/rbi/openlayer/models/projects/test_list_response.rbi @@ -88,6 +88,11 @@ module Openlayer sig { params(archived: T::Boolean).void } attr_writer :archived + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :default_to_all_pipelines + # The delay window in seconds. Only applies to tests that use production data. sig { returns(T.nilable(Float)) } attr_accessor :delay_window @@ -97,6 +102,21 @@ module Openlayer sig { returns(T.nilable(Float)) } attr_accessor :evaluation_window + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :exclude_pipelines + + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + sig { returns(T.nilable(T::Boolean)) } + attr_accessor :include_historical_data + + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + sig { returns(T.nilable(T::Array[String])) } + attr_accessor :include_pipelines + # Whether the test uses an ML model. sig { returns(T.nilable(T::Boolean)) } attr_reader :uses_ml_model @@ -190,8 +210,12 @@ module Openlayer type: Openlayer::Models::Projects::TestListResponse::Item::Type::OrSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, @@ -229,11 +253,23 @@ module Openlayer type:, # Whether the test is archived. archived: nil, + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + default_to_all_pipelines: nil, # The delay window in seconds. Only applies to tests that use production data. delay_window: nil, # The evaluation window in seconds. Only applies to tests that use production # data. evaluation_window: nil, + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + exclude_pipelines: nil, + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + include_historical_data: nil, + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + include_pipelines: nil, # Whether the test uses an ML model. uses_ml_model: nil, # Whether the test uses production data (monitoring mode only). @@ -270,8 +306,12 @@ module Openlayer type: Openlayer::Models::Projects::TestListResponse::Item::Type::TaggedSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, diff --git a/rbi/openlayer/resources/projects/inference_pipelines.rbi b/rbi/openlayer/resources/projects/inference_pipelines.rbi index ee21bc4..0c30331 100644 --- a/rbi/openlayer/resources/projects/inference_pipelines.rbi +++ b/rbi/openlayer/resources/projects/inference_pipelines.rbi @@ -10,6 +10,17 @@ module Openlayer project_id: String, description: T.nilable(String), name: String, + data_backend: + T.nilable( + T.any( + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::OrHash, + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::OrHash + ) + ), project: T.nilable( Openlayer::Projects::InferencePipelineCreateParams::Project::OrHash @@ -30,6 +41,7 @@ module Openlayer description:, # The inference pipeline name. name:, + data_backend: nil, project: nil, workspace: nil, request_options: {} diff --git a/rbi/openlayer/resources/projects/tests.rbi b/rbi/openlayer/resources/projects/tests.rbi index dc479de..8162d16 100644 --- a/rbi/openlayer/resources/projects/tests.rbi +++ b/rbi/openlayer/resources/projects/tests.rbi @@ -17,8 +17,12 @@ module Openlayer ], type: Openlayer::Projects::TestCreateParams::Type::OrSymbol, archived: T::Boolean, + default_to_all_pipelines: T.nilable(T::Boolean), delay_window: T.nilable(Float), evaluation_window: T.nilable(Float), + exclude_pipelines: T.nilable(T::Array[String]), + include_historical_data: T.nilable(T::Boolean), + include_pipelines: T.nilable(T::Array[String]), uses_ml_model: T::Boolean, uses_production_data: T::Boolean, uses_reference_dataset: T::Boolean, @@ -41,11 +45,23 @@ module Openlayer type:, # Whether the test is archived. archived: nil, + # Whether to apply the test to all pipelines (data sources) or to a specific set + # of pipelines. Only applies to tests that use production data. + default_to_all_pipelines: nil, # The delay window in seconds. Only applies to tests that use production data. delay_window: nil, # The evaluation window in seconds. Only applies to tests that use production # data. evaluation_window: nil, + # Array of pipelines (data sources) to which the test should not be applied. Only + # applies to tests that use production data. + exclude_pipelines: nil, + # Whether to include historical data in the test result. Only applies to tests + # that use production data. + include_historical_data: nil, + # Array of pipelines (data sources) to which the test should be applied. Only + # applies to tests that use production data. + include_pipelines: nil, # Whether the test uses an ML model. uses_ml_model: nil, # Whether the test uses production data (monitoring mode only). diff --git a/sig/openlayer/models/inference_pipeline_retrieve_response.rbs b/sig/openlayer/models/inference_pipeline_retrieve_response.rbs index 468762c..68d21a2 100644 --- a/sig/openlayer/models/inference_pipeline_retrieve_response.rbs +++ b/sig/openlayer/models/inference_pipeline_retrieve_response.rbs @@ -17,7 +17,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineRetrieveResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineRetrieveResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::InferencePipelineRetrieveResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::InferencePipelineRetrieveResponse::Workspace?, workspace_id: String } @@ -27,6 +30,8 @@ module Openlayer attr_accessor name: String + attr_accessor data_backend: Openlayer::Models::InferencePipelineRetrieveResponse::data_backend? + attr_accessor project: Openlayer::Models::InferencePipelineRetrieveResponse::Project? attr_accessor workspace: Openlayer::Models::InferencePipelineRetrieveResponse::Workspace? @@ -57,6 +62,10 @@ module Openlayer attr_accessor total_goal_count: Integer + attr_accessor date_last_polled: Time? + + attr_accessor total_records_count: Integer? + attr_reader workspace_id: String? def workspace_id=: (String) -> String @@ -77,7 +86,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineRetrieveResponse::status, status_message: String?, total_goal_count: Integer, + ?data_backend: Openlayer::Models::InferencePipelineRetrieveResponse::data_backend?, + ?date_last_polled: Time?, ?project: Openlayer::Models::InferencePipelineRetrieveResponse::Project?, + ?total_records_count: Integer?, ?workspace: Openlayer::Models::InferencePipelineRetrieveResponse::Workspace?, ?workspace_id: String ) -> void @@ -98,7 +110,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineRetrieveResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineRetrieveResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::InferencePipelineRetrieveResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::InferencePipelineRetrieveResponse::Workspace?, workspace_id: String } @@ -129,6 +144,464 @@ module Openlayer def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::status] end + type data_backend = + Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0 + | Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType + | Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2 + | Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3 + | Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4 + | Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5 + + module DataBackend + extend Openlayer::Internal::Type::Union + + type union_member0 = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::partition_type? + } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::backend_type + + attr_accessor bigquery_connection_id: String? + + attr_accessor dataset_id: String + + attr_accessor project_id: String + + attr_accessor table_id: String? + + attr_accessor partition_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::partition_type? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + dataset_id: String, + project_id: String, + table_id: String?, + ?partition_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::partition_type? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::partition_type? + } + + type backend_type = :bigquery + + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY: :bigquery + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + + type partition_type = :DAY | :MONTH | :YEAR + + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY: :DAY + MONTH: :MONTH + YEAR: :YEAR + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember0::partition_type] + end + end + + type backend_type = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::backend_type + } + + class BackendType < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::backend_type + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::backend_type + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::backend_type + } + + type backend_type = :default + + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT: :default + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::BackendType::backend_type] + end + end + + type union_member2 = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::backend_type + + attr_accessor database: String + + attr_accessor schema: String + + attr_accessor snowflake_connection_id: String? + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::backend_type, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + type backend_type = :snowflake + + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE: :snowflake + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember2::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member3 = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::backend_type + + attr_accessor databricks_dtl_connection_id: String? + + attr_accessor table_id: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::backend_type, + databricks_dtl_connection_id: String?, + table_id: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + type backend_type = :databricks_dtl + + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL: :databricks_dtl + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember3::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member4 = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::backend_type + + attr_accessor redshift_connection_id: String? + + attr_accessor schema_name: String + + attr_accessor table_name: String + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::backend_type, + redshift_connection_id: String?, + schema_name: String, + table_name: String + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + type backend_type = :redshift + + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT: :redshift + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember4::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member5 = + { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::backend_type + + attr_accessor database: String + + attr_accessor postgres_connection_id: String? + + attr_accessor schema: String + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::backend_type, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + type backend_type = :postgres + + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES: :postgres + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend::UnionMember5::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + def self?.variants: -> ::Array[Openlayer::Models::InferencePipelineRetrieveResponse::data_backend] + end + type project = { id: String, diff --git a/sig/openlayer/models/inference_pipeline_update_response.rbs b/sig/openlayer/models/inference_pipeline_update_response.rbs index 6e914b3..b5fabf8 100644 --- a/sig/openlayer/models/inference_pipeline_update_response.rbs +++ b/sig/openlayer/models/inference_pipeline_update_response.rbs @@ -17,7 +17,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineUpdateResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineUpdateResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::InferencePipelineUpdateResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::InferencePipelineUpdateResponse::Workspace?, workspace_id: String } @@ -27,6 +30,8 @@ module Openlayer attr_accessor name: String + attr_accessor data_backend: Openlayer::Models::InferencePipelineUpdateResponse::data_backend? + attr_accessor project: Openlayer::Models::InferencePipelineUpdateResponse::Project? attr_accessor workspace: Openlayer::Models::InferencePipelineUpdateResponse::Workspace? @@ -57,6 +62,10 @@ module Openlayer attr_accessor total_goal_count: Integer + attr_accessor date_last_polled: Time? + + attr_accessor total_records_count: Integer? + attr_reader workspace_id: String? def workspace_id=: (String) -> String @@ -77,7 +86,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineUpdateResponse::status, status_message: String?, total_goal_count: Integer, + ?data_backend: Openlayer::Models::InferencePipelineUpdateResponse::data_backend?, + ?date_last_polled: Time?, ?project: Openlayer::Models::InferencePipelineUpdateResponse::Project?, + ?total_records_count: Integer?, ?workspace: Openlayer::Models::InferencePipelineUpdateResponse::Workspace?, ?workspace_id: String ) -> void @@ -98,7 +110,10 @@ module Openlayer status: Openlayer::Models::InferencePipelineUpdateResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineUpdateResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::InferencePipelineUpdateResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::InferencePipelineUpdateResponse::Workspace?, workspace_id: String } @@ -129,6 +144,464 @@ module Openlayer def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::status] end + type data_backend = + Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0 + | Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType + | Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2 + | Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3 + | Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4 + | Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5 + + module DataBackend + extend Openlayer::Internal::Type::Union + + type union_member0 = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::partition_type? + } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::backend_type + + attr_accessor bigquery_connection_id: String? + + attr_accessor dataset_id: String + + attr_accessor project_id: String + + attr_accessor table_id: String? + + attr_accessor partition_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::partition_type? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + dataset_id: String, + project_id: String, + table_id: String?, + ?partition_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::partition_type? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::partition_type? + } + + type backend_type = :bigquery + + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY: :bigquery + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + + type partition_type = :DAY | :MONTH | :YEAR + + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY: :DAY + MONTH: :MONTH + YEAR: :YEAR + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember0::partition_type] + end + end + + type backend_type = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::backend_type + } + + class BackendType < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::backend_type + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::backend_type + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::backend_type + } + + type backend_type = :default + + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT: :default + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::BackendType::backend_type] + end + end + + type union_member2 = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::backend_type + + attr_accessor database: String + + attr_accessor schema: String + + attr_accessor snowflake_connection_id: String? + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::backend_type, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + type backend_type = :snowflake + + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE: :snowflake + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember2::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member3 = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::backend_type + + attr_accessor databricks_dtl_connection_id: String? + + attr_accessor table_id: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::backend_type, + databricks_dtl_connection_id: String?, + table_id: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + type backend_type = :databricks_dtl + + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL: :databricks_dtl + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember3::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member4 = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::backend_type + + attr_accessor redshift_connection_id: String? + + attr_accessor schema_name: String + + attr_accessor table_name: String + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::backend_type, + redshift_connection_id: String?, + schema_name: String, + table_name: String + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + type backend_type = :redshift + + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT: :redshift + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember4::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member5 = + { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::backend_type + + attr_accessor database: String + + attr_accessor postgres_connection_id: String? + + attr_accessor schema: String + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::backend_type, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + type backend_type = :postgres + + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES: :postgres + + def self?.values: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::DataBackend::UnionMember5::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + def self?.variants: -> ::Array[Openlayer::Models::InferencePipelineUpdateResponse::data_backend] + end + type project = { id: String, diff --git a/sig/openlayer/models/projects/inference_pipeline_create_params.rbs b/sig/openlayer/models/projects/inference_pipeline_create_params.rbs index 16e6b57..6990728 100644 --- a/sig/openlayer/models/projects/inference_pipeline_create_params.rbs +++ b/sig/openlayer/models/projects/inference_pipeline_create_params.rbs @@ -5,6 +5,7 @@ module Openlayer { description: String?, name: String, + data_backend: Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend?, project: Openlayer::Projects::InferencePipelineCreateParams::Project?, workspace: Openlayer::Projects::InferencePipelineCreateParams::Workspace? } @@ -18,6 +19,8 @@ module Openlayer attr_accessor name: String + attr_accessor data_backend: Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend? + attr_accessor project: Openlayer::Projects::InferencePipelineCreateParams::Project? attr_accessor workspace: Openlayer::Projects::InferencePipelineCreateParams::Workspace? @@ -25,6 +28,7 @@ module Openlayer def initialize: ( description: String?, name: String, + ?data_backend: Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend?, ?project: Openlayer::Projects::InferencePipelineCreateParams::Project?, ?workspace: Openlayer::Projects::InferencePipelineCreateParams::Workspace?, ?request_options: Openlayer::request_opts @@ -33,11 +37,500 @@ module Openlayer def to_hash: -> { description: String?, name: String, + data_backend: Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend?, project: Openlayer::Projects::InferencePipelineCreateParams::Project?, workspace: Openlayer::Projects::InferencePipelineCreateParams::Workspace?, request_options: Openlayer::RequestOptions } + type data_backend = + Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0 + | Openlayer::Projects::InferencePipelineCreateParams::DataBackend::BackendType + | Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2 + | Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3 + | Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4 + | Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5 + + module DataBackend + extend Openlayer::Internal::Type::Union + + type union_member0 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::partition_type? + } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::backend_type + + attr_accessor bigquery_connection_id: String? + + attr_accessor dataset_id: String + + attr_accessor project_id: String + + attr_accessor table_id: String? + + attr_accessor partition_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::partition_type? + + attr_accessor config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + ?partition_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::partition_type? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::partition_type? + } + + type backend_type = :bigquery + + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY: :bigquery + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + attr_accessor inference_id_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?inference_id_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + + type partition_type = :DAY | :MONTH | :YEAR + + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY: :DAY + MONTH: :MONTH + YEAR: :YEAR + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember0::partition_type] + end + end + + type backend_type = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::backend_type + } + + class BackendType < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::backend_type + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::backend_type + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::backend_type + } + + type backend_type = :default + + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT: :default + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::BackendType::backend_type] + end + end + + type union_member2 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::backend_type + + attr_accessor database: String + + attr_accessor schema: String + + attr_accessor snowflake_connection_id: String? + + attr_accessor table: String? + + attr_accessor config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + type backend_type = :snowflake + + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE: :snowflake + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember2::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + attr_accessor inference_id_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?inference_id_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member3 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::backend_type + + attr_accessor databricks_dtl_connection_id: String? + + attr_accessor table_id: String? + + attr_accessor config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + type backend_type = :databricks_dtl + + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL: :databricks_dtl + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember3::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + attr_accessor inference_id_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?inference_id_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member4 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::backend_type + + attr_accessor redshift_connection_id: String? + + attr_accessor schema_name: String + + attr_accessor table_name: String + + attr_accessor config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + type backend_type = :redshift + + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT: :redshift + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember4::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + attr_accessor inference_id_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?inference_id_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member5 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::backend_type + + attr_accessor database: String + + attr_accessor postgres_connection_id: String? + + attr_accessor schema: String + + attr_accessor table: String? + + attr_accessor config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::backend_type, + config: Openlayer::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + type backend_type = :postgres + + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES: :postgres + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::DataBackend::UnionMember5::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + attr_accessor inference_id_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?inference_id_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + def self?.variants: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend] + end + type project = { id: String, diff --git a/sig/openlayer/models/projects/inference_pipeline_create_response.rbs b/sig/openlayer/models/projects/inference_pipeline_create_response.rbs index 2f7b9fe..f5983db 100644 --- a/sig/openlayer/models/projects/inference_pipeline_create_response.rbs +++ b/sig/openlayer/models/projects/inference_pipeline_create_response.rbs @@ -18,7 +18,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineCreateResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::Projects::InferencePipelineCreateResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::Projects::InferencePipelineCreateResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace?, workspace_id: String } @@ -28,6 +31,8 @@ module Openlayer attr_accessor name: String + attr_accessor data_backend: Openlayer::Models::Projects::InferencePipelineCreateResponse::data_backend? + attr_accessor project: Openlayer::Models::Projects::InferencePipelineCreateResponse::Project? attr_accessor workspace: Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace? @@ -58,6 +63,10 @@ module Openlayer attr_accessor total_goal_count: Integer + attr_accessor date_last_polled: Time? + + attr_accessor total_records_count: Integer? + attr_reader workspace_id: String? def workspace_id=: (String) -> String @@ -78,7 +87,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineCreateResponse::status, status_message: String?, total_goal_count: Integer, + ?data_backend: Openlayer::Models::Projects::InferencePipelineCreateResponse::data_backend?, + ?date_last_polled: Time?, ?project: Openlayer::Models::Projects::InferencePipelineCreateResponse::Project?, + ?total_records_count: Integer?, ?workspace: Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace?, ?workspace_id: String ) -> void @@ -99,7 +111,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineCreateResponse::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::Projects::InferencePipelineCreateResponse::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::Projects::InferencePipelineCreateResponse::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace?, workspace_id: String } @@ -130,6 +145,464 @@ module Openlayer def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::status] end + type data_backend = + Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0 + | Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType + | Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2 + | Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3 + | Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4 + | Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5 + + module DataBackend + extend Openlayer::Internal::Type::Union + + type union_member0 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::partition_type? + } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::backend_type + + attr_accessor bigquery_connection_id: String? + + attr_accessor dataset_id: String + + attr_accessor project_id: String + + attr_accessor table_id: String? + + attr_accessor partition_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::partition_type? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + dataset_id: String, + project_id: String, + table_id: String?, + ?partition_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::partition_type? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::partition_type? + } + + type backend_type = :bigquery + + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY: :bigquery + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + + type partition_type = :DAY | :MONTH | :YEAR + + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY: :DAY + MONTH: :MONTH + YEAR: :YEAR + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember0::partition_type] + end + end + + type backend_type = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::backend_type + } + + class BackendType < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::backend_type + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::backend_type + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::backend_type + } + + type backend_type = :default + + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT: :default + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::BackendType::backend_type] + end + end + + type union_member2 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::backend_type + + attr_accessor database: String + + attr_accessor schema: String + + attr_accessor snowflake_connection_id: String? + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::backend_type, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + type backend_type = :snowflake + + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE: :snowflake + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember2::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member3 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::backend_type + + attr_accessor databricks_dtl_connection_id: String? + + attr_accessor table_id: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::backend_type, + databricks_dtl_connection_id: String?, + table_id: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + type backend_type = :databricks_dtl + + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL: :databricks_dtl + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember3::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member4 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::backend_type + + attr_accessor redshift_connection_id: String? + + attr_accessor schema_name: String + + attr_accessor table_name: String + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::backend_type, + redshift_connection_id: String?, + schema_name: String, + table_name: String + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + type backend_type = :redshift + + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT: :redshift + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember4::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member5 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::backend_type + + attr_accessor database: String + + attr_accessor postgres_connection_id: String? + + attr_accessor schema: String + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::backend_type, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + type backend_type = :postgres + + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES: :postgres + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend::UnionMember5::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + def self?.variants: -> ::Array[Openlayer::Models::Projects::InferencePipelineCreateResponse::data_backend] + end + type project = { id: String, diff --git a/sig/openlayer/models/projects/inference_pipeline_list_response.rbs b/sig/openlayer/models/projects/inference_pipeline_list_response.rbs index 3394741..e6412b8 100644 --- a/sig/openlayer/models/projects/inference_pipeline_list_response.rbs +++ b/sig/openlayer/models/projects/inference_pipeline_list_response.rbs @@ -34,7 +34,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineListResponse::Item::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::Projects::InferencePipelineListResponse::Item::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace?, workspace_id: String } @@ -44,6 +47,8 @@ module Openlayer attr_accessor name: String + attr_accessor data_backend: Openlayer::Models::Projects::InferencePipelineListResponse::Item::data_backend? + attr_accessor project: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project? attr_accessor workspace: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace? @@ -74,6 +79,10 @@ module Openlayer attr_accessor total_goal_count: Integer + attr_accessor date_last_polled: Time? + + attr_accessor total_records_count: Integer? + attr_reader workspace_id: String? def workspace_id=: (String) -> String @@ -94,7 +103,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineListResponse::Item::status, status_message: String?, total_goal_count: Integer, + ?data_backend: Openlayer::Models::Projects::InferencePipelineListResponse::Item::data_backend?, + ?date_last_polled: Time?, ?project: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project?, + ?total_records_count: Integer?, ?workspace: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace?, ?workspace_id: String ) -> void @@ -115,7 +127,10 @@ module Openlayer status: Openlayer::Models::Projects::InferencePipelineListResponse::Item::status, status_message: String?, total_goal_count: Integer, + data_backend: Openlayer::Models::Projects::InferencePipelineListResponse::Item::data_backend?, + date_last_polled: Time?, project: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Project?, + total_records_count: Integer?, workspace: Openlayer::Models::Projects::InferencePipelineListResponse::Item::Workspace?, workspace_id: String } @@ -146,6 +161,464 @@ module Openlayer def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::status] end + type data_backend = + Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0 + | Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType + | Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2 + | Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3 + | Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4 + | Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5 + + module DataBackend + extend Openlayer::Internal::Type::Union + + type union_member0 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::partition_type? + } + + class UnionMember0 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::backend_type + + attr_accessor bigquery_connection_id: String? + + attr_accessor dataset_id: String + + attr_accessor project_id: String + + attr_accessor table_id: String? + + attr_accessor partition_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::partition_type? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + dataset_id: String, + project_id: String, + table_id: String?, + ?partition_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::partition_type? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::backend_type, + bigquery_connection_id: String?, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::Config, + dataset_id: String, + project_id: String, + table_id: String?, + partition_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::partition_type? + } + + type backend_type = :bigquery + + module BackendType + extend Openlayer::Internal::Type::Enum + + BIGQUERY: :bigquery + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + + type partition_type = :DAY | :MONTH | :YEAR + + module PartitionType + extend Openlayer::Internal::Type::Enum + + DAY: :DAY + MONTH: :MONTH + YEAR: :YEAR + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember0::partition_type] + end + end + + type backend_type = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::backend_type + } + + class BackendType < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::backend_type + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::backend_type + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::backend_type + } + + type backend_type = :default + + module BackendType + extend Openlayer::Internal::Type::Enum + + DEFAULT: :default + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::BackendType::backend_type] + end + end + + type union_member2 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + class UnionMember2 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::backend_type + + attr_accessor database: String + + attr_accessor schema: String + + attr_accessor snowflake_connection_id: String? + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::backend_type, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::Config, + database: String, + schema: String, + snowflake_connection_id: String?, + table: String? + } + + type backend_type = :snowflake + + module BackendType + extend Openlayer::Internal::Type::Enum + + SNOWFLAKE: :snowflake + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember2::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member3 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + class UnionMember3 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::backend_type + + attr_accessor databricks_dtl_connection_id: String? + + attr_accessor table_id: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::backend_type, + databricks_dtl_connection_id: String?, + table_id: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::Config, + databricks_dtl_connection_id: String?, + table_id: String? + } + + type backend_type = :databricks_dtl + + module BackendType + extend Openlayer::Internal::Type::Enum + + DATABRICKS_DTL: :databricks_dtl + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember3::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member4 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + class UnionMember4 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::backend_type + + attr_accessor redshift_connection_id: String? + + attr_accessor schema_name: String + + attr_accessor table_name: String + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::backend_type, + redshift_connection_id: String?, + schema_name: String, + table_name: String + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::Config, + redshift_connection_id: String?, + schema_name: String, + table_name: String + } + + type backend_type = :redshift + + module BackendType + extend Openlayer::Internal::Type::Enum + + REDSHIFT: :redshift + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember4::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + type union_member5 = + { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + class UnionMember5 < Openlayer::Internal::Type::BaseModel + attr_accessor backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::backend_type + + attr_accessor database: String + + attr_accessor postgres_connection_id: String? + + attr_accessor schema: String + + attr_accessor table: String? + + def initialize: ( + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::backend_type, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + ) -> void + + def to_hash: -> { + backend_type: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::backend_type, + config: Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::Config, + database: String, + postgres_connection_id: String?, + schema: String, + table: String? + } + + type backend_type = :postgres + + module BackendType + extend Openlayer::Internal::Type::Enum + + POSTGRES: :postgres + + def self?.values: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::DataBackend::UnionMember5::backend_type] + end + + type config = + { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + + class Config < Openlayer::Internal::Type::BaseModel + attr_accessor ground_truth_column_name: String? + + attr_accessor human_feedback_column_name: String? + + attr_accessor latency_column_name: String? + + attr_accessor timestamp_column_name: String? + + def initialize: ( + ?ground_truth_column_name: String?, + ?human_feedback_column_name: String?, + ?latency_column_name: String?, + ?timestamp_column_name: String? + ) -> void + + def to_hash: -> { + ground_truth_column_name: String?, + human_feedback_column_name: String?, + inference_id_column_name: String?, + latency_column_name: String?, + timestamp_column_name: String? + } + end + end + + def self?.variants: -> ::Array[Openlayer::Models::Projects::InferencePipelineListResponse::Item::data_backend] + end + type project = { id: String, diff --git a/sig/openlayer/models/projects/test_create_params.rbs b/sig/openlayer/models/projects/test_create_params.rbs index 8b3ba3e..5ae51d6 100644 --- a/sig/openlayer/models/projects/test_create_params.rbs +++ b/sig/openlayer/models/projects/test_create_params.rbs @@ -9,8 +9,12 @@ module Openlayer thresholds: ::Array[Openlayer::Projects::TestCreateParams::Threshold], type: Openlayer::Models::Projects::TestCreateParams::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, @@ -37,10 +41,18 @@ module Openlayer def archived=: (bool) -> bool + attr_accessor default_to_all_pipelines: bool? + attr_accessor delay_window: Float? attr_accessor evaluation_window: Float? + attr_accessor exclude_pipelines: ::Array[String]? + + attr_accessor include_historical_data: bool? + + attr_accessor include_pipelines: ::Array[String]? + attr_reader uses_ml_model: bool? def uses_ml_model=: (bool) -> bool @@ -68,8 +80,12 @@ module Openlayer thresholds: ::Array[Openlayer::Projects::TestCreateParams::Threshold], type: Openlayer::Models::Projects::TestCreateParams::type_, ?archived: bool, + ?default_to_all_pipelines: bool?, ?delay_window: Float?, ?evaluation_window: Float?, + ?exclude_pipelines: ::Array[String]?, + ?include_historical_data: bool?, + ?include_pipelines: ::Array[String]?, ?uses_ml_model: bool, ?uses_production_data: bool, ?uses_reference_dataset: bool, @@ -85,8 +101,12 @@ module Openlayer thresholds: ::Array[Openlayer::Projects::TestCreateParams::Threshold], type: Openlayer::Models::Projects::TestCreateParams::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, diff --git a/sig/openlayer/models/projects/test_create_response.rbs b/sig/openlayer/models/projects/test_create_response.rbs index 0b992ca..8c9f210 100644 --- a/sig/openlayer/models/projects/test_create_response.rbs +++ b/sig/openlayer/models/projects/test_create_response.rbs @@ -18,8 +18,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestCreateResponse::Threshold], type: Openlayer::Models::Projects::TestCreateResponse::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, @@ -42,10 +46,18 @@ module Openlayer def archived=: (bool) -> bool + attr_accessor default_to_all_pipelines: bool? + attr_accessor delay_window: Float? attr_accessor evaluation_window: Float? + attr_accessor exclude_pipelines: ::Array[String]? + + attr_accessor include_historical_data: bool? + + attr_accessor include_pipelines: ::Array[String]? + attr_reader uses_ml_model: bool? def uses_ml_model=: (bool) -> bool @@ -100,8 +112,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestCreateResponse::Threshold], type: Openlayer::Models::Projects::TestCreateResponse::type_, ?archived: bool, + ?default_to_all_pipelines: bool?, ?delay_window: Float?, ?evaluation_window: Float?, + ?exclude_pipelines: ::Array[String]?, + ?include_historical_data: bool?, + ?include_pipelines: ::Array[String]?, ?uses_ml_model: bool, ?uses_production_data: bool, ?uses_reference_dataset: bool, @@ -125,8 +141,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestCreateResponse::Threshold], type: Openlayer::Models::Projects::TestCreateResponse::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, diff --git a/sig/openlayer/models/projects/test_list_response.rbs b/sig/openlayer/models/projects/test_list_response.rbs index 441742a..30671e1 100644 --- a/sig/openlayer/models/projects/test_list_response.rbs +++ b/sig/openlayer/models/projects/test_list_response.rbs @@ -32,8 +32,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestListResponse::Item::Threshold], type: Openlayer::Models::Projects::TestListResponse::Item::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, @@ -56,10 +60,18 @@ module Openlayer def archived=: (bool) -> bool + attr_accessor default_to_all_pipelines: bool? + attr_accessor delay_window: Float? attr_accessor evaluation_window: Float? + attr_accessor exclude_pipelines: ::Array[String]? + + attr_accessor include_historical_data: bool? + + attr_accessor include_pipelines: ::Array[String]? + attr_reader uses_ml_model: bool? def uses_ml_model=: (bool) -> bool @@ -114,8 +126,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestListResponse::Item::Threshold], type: Openlayer::Models::Projects::TestListResponse::Item::type_, ?archived: bool, + ?default_to_all_pipelines: bool?, ?delay_window: Float?, ?evaluation_window: Float?, + ?exclude_pipelines: ::Array[String]?, + ?include_historical_data: bool?, + ?include_pipelines: ::Array[String]?, ?uses_ml_model: bool, ?uses_production_data: bool, ?uses_reference_dataset: bool, @@ -139,8 +155,12 @@ module Openlayer thresholds: ::Array[Openlayer::Models::Projects::TestListResponse::Item::Threshold], type: Openlayer::Models::Projects::TestListResponse::Item::type_, archived: bool, + default_to_all_pipelines: bool?, delay_window: Float?, evaluation_window: Float?, + exclude_pipelines: ::Array[String]?, + include_historical_data: bool?, + include_pipelines: ::Array[String]?, uses_ml_model: bool, uses_production_data: bool, uses_reference_dataset: bool, diff --git a/sig/openlayer/resources/projects/inference_pipelines.rbs b/sig/openlayer/resources/projects/inference_pipelines.rbs index b3e2b6e..e9f76a3 100644 --- a/sig/openlayer/resources/projects/inference_pipelines.rbs +++ b/sig/openlayer/resources/projects/inference_pipelines.rbs @@ -6,6 +6,7 @@ module Openlayer String project_id, description: String?, name: String, + ?data_backend: Openlayer::Models::Projects::InferencePipelineCreateParams::data_backend?, ?project: Openlayer::Projects::InferencePipelineCreateParams::Project?, ?workspace: Openlayer::Projects::InferencePipelineCreateParams::Workspace?, ?request_options: Openlayer::request_opts diff --git a/sig/openlayer/resources/projects/tests.rbs b/sig/openlayer/resources/projects/tests.rbs index 5546240..739797d 100644 --- a/sig/openlayer/resources/projects/tests.rbs +++ b/sig/openlayer/resources/projects/tests.rbs @@ -10,8 +10,12 @@ module Openlayer thresholds: ::Array[Openlayer::Projects::TestCreateParams::Threshold], type: Openlayer::Models::Projects::TestCreateParams::type_, ?archived: bool, + ?default_to_all_pipelines: bool?, ?delay_window: Float?, ?evaluation_window: Float?, + ?exclude_pipelines: ::Array[String]?, + ?include_historical_data: bool?, + ?include_pipelines: ::Array[String]?, ?uses_ml_model: bool, ?uses_production_data: bool, ?uses_reference_dataset: bool, diff --git a/test/openlayer/internal/util_test.rb b/test/openlayer/internal/util_test.rb index 2d15dad..6c99d2f 100644 --- a/test/openlayer/internal/util_test.rb +++ b/test/openlayer/internal/util_test.rb @@ -124,6 +124,14 @@ def test_joining path: "/c", query: {"d" => ["e"]} } + ], + [ + "h://a.b/c?d=e", + "h://nope", + { + path: "h://a.b/c", + query: {"d" => ["e"]} + } ] ] @@ -335,6 +343,29 @@ def test_rewind_closing assert_equal(0, steps) end + def test_thread_interrupts + once = 0 + que = Queue.new + enum = Enumerator.new do |y| + 10.times { y << _1 } + ensure + once = once.succ + end + + fused_1 = Openlayer::Internal::Util.fused_enum(enum, external: true) { loop { enum.next } } + fused_2 = Openlayer::Internal::Util.chain_fused(fused_1) { fused_1.each(&_1) } + fused_3 = Openlayer::Internal::Util.chain_fused(fused_2) { fused_2.each(&_1) } + + th = ::Thread.new do + que << "🐶" + fused_3.each { sleep(10) } + end + + assert_equal("🐶", que.pop) + th.kill.join + assert_equal(1, once) + end + def test_closing arr = [1, 2, 3] once = 0 diff --git a/test/openlayer/resources/inference_pipelines_test.rb b/test/openlayer/resources/inference_pipelines_test.rb index b37357e..c5f636a 100644 --- a/test/openlayer/resources/inference_pipelines_test.rb +++ b/test/openlayer/resources/inference_pipelines_test.rb @@ -27,7 +27,10 @@ def test_retrieve status: Openlayer::Models::InferencePipelineRetrieveResponse::Status, status_message: String | nil, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineRetrieveResponse::DataBackend | nil, + date_last_polled: Time | nil, project: Openlayer::Models::InferencePipelineRetrieveResponse::Project | nil, + total_records_count: Integer | nil, workspace: Openlayer::Models::InferencePipelineRetrieveResponse::Workspace | nil, workspace_id: String | nil } @@ -58,7 +61,10 @@ def test_update status: Openlayer::Models::InferencePipelineUpdateResponse::Status, status_message: String | nil, total_goal_count: Integer, + data_backend: Openlayer::Models::InferencePipelineUpdateResponse::DataBackend | nil, + date_last_polled: Time | nil, project: Openlayer::Models::InferencePipelineUpdateResponse::Project | nil, + total_records_count: Integer | nil, workspace: Openlayer::Models::InferencePipelineUpdateResponse::Workspace | nil, workspace_id: String | nil } diff --git a/test/openlayer/resources/projects/inference_pipelines_test.rb b/test/openlayer/resources/projects/inference_pipelines_test.rb index e15baef..0d40f36 100644 --- a/test/openlayer/resources/projects/inference_pipelines_test.rb +++ b/test/openlayer/resources/projects/inference_pipelines_test.rb @@ -32,7 +32,10 @@ def test_create_required_params status: Openlayer::Models::Projects::InferencePipelineCreateResponse::Status, status_message: String | nil, total_goal_count: Integer, + data_backend: Openlayer::Models::Projects::InferencePipelineCreateResponse::DataBackend | nil, + date_last_polled: Time | nil, project: Openlayer::Models::Projects::InferencePipelineCreateResponse::Project | nil, + total_records_count: Integer | nil, workspace: Openlayer::Models::Projects::InferencePipelineCreateResponse::Workspace | nil, workspace_id: String | nil } diff --git a/test/openlayer/resources/projects/tests_test.rb b/test/openlayer/resources/projects/tests_test.rb index e073673..225e1e5 100644 --- a/test/openlayer/resources/projects/tests_test.rb +++ b/test/openlayer/resources/projects/tests_test.rb @@ -35,8 +35,12 @@ def test_create_required_params thresholds: ^(Openlayer::Internal::Type::ArrayOf[Openlayer::Models::Projects::TestCreateResponse::Threshold]), type: Openlayer::Models::Projects::TestCreateResponse::Type, archived: Openlayer::Internal::Type::Boolean | nil, + default_to_all_pipelines: Openlayer::Internal::Type::Boolean | nil, delay_window: Float | nil, evaluation_window: Float | nil, + exclude_pipelines: ^(Openlayer::Internal::Type::ArrayOf[String]) | nil, + include_historical_data: Openlayer::Internal::Type::Boolean | nil, + include_pipelines: ^(Openlayer::Internal::Type::ArrayOf[String]) | nil, uses_ml_model: Openlayer::Internal::Type::Boolean | nil, uses_production_data: Openlayer::Internal::Type::Boolean | nil, uses_reference_dataset: Openlayer::Internal::Type::Boolean | nil,