openlayer 0.11.2 → 0.12.0

data/lib/openlayer/models/inference_pipeline_retrieve_users_params.rb

@@ -12,6 +12,12 @@ module Openlayer
       #   @return [String]
       required :inference_pipeline_id, String
 
+      # @!attribute asc
+      #   Whether or not to sort on the sortColumn in ascending order.
+      #
+      #   @return [Boolean, nil]
+      optional :asc, Openlayer::Internal::Type::Boolean
+
       # @!attribute page
       #   The page to return in a paginated query.
       #
@@ -24,14 +30,249 @@ module Openlayer
       #   @return [Integer, nil]
       optional :per_page, Integer
 
-      # @!method initialize(inference_pipeline_id:, page: nil, per_page: nil, request_options: {})
+      # @!attribute sort_column
+      #   Name of the column to sort on
+      #
+      #   @return [String, nil]
+      optional :sort_column, String
+
+      # @!attribute column_filters
+      #
+      #   @return [Array<Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter>, nil]
+      optional :column_filters,
+               -> {
+                 Openlayer::Internal::Type::ArrayOf[union: Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter]
+               },
+               api_name: :columnFilters,
+               nil?: true
+
+      # @!attribute exclude_row_id_list
+      #
+      #   @return [Array<Integer>, nil]
+      optional :exclude_row_id_list,
+               Openlayer::Internal::Type::ArrayOf[Integer],
+               api_name: :excludeRowIdList,
+               nil?: true
+
+      # @!attribute not_search_query_and
+      #
+      #   @return [Array<String>, nil]
+      optional :not_search_query_and,
+               Openlayer::Internal::Type::ArrayOf[String],
+               api_name: :notSearchQueryAnd,
+               nil?: true
+
+      # @!attribute not_search_query_or
+      #
+      #   @return [Array<String>, nil]
+      optional :not_search_query_or,
+               Openlayer::Internal::Type::ArrayOf[String],
+               api_name: :notSearchQueryOr,
+               nil?: true
+
+      # @!attribute row_id_list
+      #
+      #   @return [Array<Integer>, nil]
+      optional :row_id_list, Openlayer::Internal::Type::ArrayOf[Integer], api_name: :rowIdList, nil?: true
+
+      # @!attribute search_query_and
+      #
+      #   @return [Array<String>, nil]
+      optional :search_query_and,
+               Openlayer::Internal::Type::ArrayOf[String],
+               api_name: :searchQueryAnd,
+               nil?: true
+
+      # @!attribute search_query_or
+      #
+      #   @return [Array<String>, nil]
+      optional :search_query_or,
+               Openlayer::Internal::Type::ArrayOf[String],
+               api_name: :searchQueryOr,
+               nil?: true
+
+      # @!method initialize(inference_pipeline_id:, asc: nil, page: nil, per_page: nil, sort_column: nil, column_filters: nil, exclude_row_id_list: nil, not_search_query_and: nil, not_search_query_or: nil, row_id_list: nil, search_query_and: nil, search_query_or: nil, request_options: {})
       #   @param inference_pipeline_id [String]
       #
+      #   @param asc [Boolean] Whether or not to sort on the sortColumn in ascending order.
+      #
       #   @param page [Integer] The page to return in a paginated query.
       #
       #   @param per_page [Integer] Maximum number of items to return per page.
       #
+      #   @param sort_column [String] Name of the column to sort on
+      #
+      #   @param column_filters [Array<Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter>, nil]
+      #
+      #   @param exclude_row_id_list [Array<Integer>, nil]
+      #
+      #   @param not_search_query_and [Array<String>, nil]
+      #
+      #   @param not_search_query_or [Array<String>, nil]
+      #
+      #   @param row_id_list [Array<Integer>, nil]
+      #
+      #   @param search_query_and [Array<String>, nil]
+      #
+      #   @param search_query_or [Array<String>, nil]
+      #
       #   @param request_options [Openlayer::RequestOptions, Hash{Symbol=>Object}]
+
+      module ColumnFilter
+        extend Openlayer::Internal::Type::Union
+
+        variant -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter }
+
+        variant -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter }
+
+        variant -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter }
+
+        class SetColumnFilter < Openlayer::Internal::Type::BaseModel
+          # @!attribute measurement
+          #   The name of the column.
+          #
+          #   @return [String]
+          required :measurement, String
+
+          # @!attribute operator
+          #
+          #   @return [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter::Operator]
+          required :operator,
+                   enum: -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter::Operator }
+
+          # @!attribute value
+          #
+          #   @return [Array<String, Float>]
+          required :value,
+                   -> { Openlayer::Internal::Type::ArrayOf[union: Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter::Value] }
+
+          # @!method initialize(measurement:, operator:, value:)
+          #   @param measurement [String] The name of the column.
+          #
+          #   @param operator [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter::Operator]
+          #
+          #   @param value [Array<String, Float>]
+
+          # @see Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter#operator
+          module Operator
+            extend Openlayer::Internal::Type::Enum
+
+            CONTAINS_NONE = :contains_none
+            CONTAINS_ANY = :contains_any
+            CONTAINS_ALL = :contains_all
+            ONE_OF = :one_of
+            NONE_OF = :none_of
+
+            # @!method self.values
+            #   @return [Array<Symbol>]
+          end
+
+          module Value
+            extend Openlayer::Internal::Type::Union
+
+            variant String
+
+            variant Float
+
+            # @!method self.variants
+            #   @return [Array(String, Float)]
+          end
+        end
+
+        class NumericColumnFilter < Openlayer::Internal::Type::BaseModel
+          # @!attribute measurement
+          #   The name of the column.
+          #
+          #   @return [String]
+          required :measurement, String
+
+          # @!attribute operator
+          #
+          #   @return [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter::Operator]
+          required :operator,
+                   enum: -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter::Operator }
+
+          # @!attribute value
+          #
+          #   @return [Float, nil]
+          required :value, Float, nil?: true
+
+          # @!method initialize(measurement:, operator:, value:)
+          #   @param measurement [String] The name of the column.
+          #
+          #   @param operator [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter::Operator]
+          #
+          #   @param value [Float, nil]
+
+          # @see Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter#operator
+          module Operator
+            extend Openlayer::Internal::Type::Enum
+
+            GREATER = :>
+            GREATER_OR_EQUALS = :">="
+            IS = :is
+            LESS = :<
+            LESS_OR_EQUALS = :"<="
+            NOT_EQUALS = :"!="
+
+            # @!method self.values
+            #   @return [Array<Symbol>]
+          end
+        end
+
+        class StringColumnFilter < Openlayer::Internal::Type::BaseModel
+          # @!attribute measurement
+          #   The name of the column.
+          #
+          #   @return [String]
+          required :measurement, String
+
+          # @!attribute operator
+          #
+          #   @return [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter::Operator]
+          required :operator,
+                   enum: -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter::Operator }
+
+          # @!attribute value
+          #
+          #   @return [String, Boolean]
+          required :value,
+                   union: -> { Openlayer::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter::Value }
+
+          # @!method initialize(measurement:, operator:, value:)
+          #   @param measurement [String] The name of the column.
+          #
+          #   @param operator [Symbol, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter::Operator]
+          #
+          #   @param value [String, Boolean]
+
+          # @see Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter#operator
+          module Operator
+            extend Openlayer::Internal::Type::Enum
+
+            IS = :is
+            NOT_EQUALS = :"!="
+
+            # @!method self.values
+            #   @return [Array<Symbol>]
+          end
+
+          # @see Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter#value
+          module Value
+            extend Openlayer::Internal::Type::Union
+
+            variant String
+
+            variant Openlayer::Internal::Type::Boolean
+
+            # @!method self.variants
+            #   @return [Array(String, Boolean)]
+          end
+        end
+
+        # @!method self.variants
+        #   @return [Array(Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter)]
+      end
     end
   end
 end

data/lib/openlayer/models.rb

@@ -47,6 +47,8 @@ module Openlayer
 
   InferencePipelineRetrieveParams = Openlayer::Models::InferencePipelineRetrieveParams
 
+  InferencePipelineRetrieveSessionsParams = Openlayer::Models::InferencePipelineRetrieveSessionsParams
+
   InferencePipelineRetrieveUsersParams = Openlayer::Models::InferencePipelineRetrieveUsersParams
 
   InferencePipelines = Openlayer::Models::InferencePipelines

data/lib/openlayer/resources/inference_pipelines.rb

@@ -88,19 +88,89 @@ module Openlayer
         )
       end
 
+      # Get aggregated session data for an inference pipeline with pagination and
+      # metadata.
+      #
+      # Returns a list of sessions for the inference pipeline, including activity
+      # statistics such as record counts, token usage, cost, latency, and the first and
+      # last records.
+      #
+      # @overload retrieve_sessions(inference_pipeline_id, asc: nil, page: nil, per_page: nil, sort_column: nil, column_filters: nil, exclude_row_id_list: nil, not_search_query_and: nil, not_search_query_or: nil, row_id_list: nil, search_query_and: nil, search_query_or: nil, request_options: {})
+      #
+      # @param inference_pipeline_id [String] Path param: The inference pipeline id (a UUID).
+      #
+      # @param asc [Boolean] Query param: Whether or not to sort on the sortColumn in ascending order.
+      #
+      # @param page [Integer] Query param: The page to return in a paginated query.
+      #
+      # @param per_page [Integer] Query param: Maximum number of items to return per page.
+      #
+      # @param sort_column [String] Query param: Name of the column to sort on
+      #
+      # @param column_filters [Array<Openlayer::Models::InferencePipelineRetrieveSessionsParams::ColumnFilter::SetColumnFilter, Openlayer::Models::InferencePipelineRetrieveSessionsParams::ColumnFilter::NumericColumnFilter, Openlayer::Models::InferencePipelineRetrieveSessionsParams::ColumnFilter::StringColumnFilter>, nil] Body param
+      #
+      # @param exclude_row_id_list [Array<Integer>, nil] Body param
+      #
+      # @param not_search_query_and [Array<String>, nil] Body param
+      #
+      # @param not_search_query_or [Array<String>, nil] Body param
+      #
+      # @param row_id_list [Array<Integer>, nil] Body param
+      #
+      # @param search_query_and [Array<String>, nil] Body param
+      #
+      # @param search_query_or [Array<String>, nil] Body param
+      #
+      # @param request_options [Openlayer::RequestOptions, Hash{Symbol=>Object}, nil]
+      #
+      # @return [Openlayer::Models::InferencePipelineRetrieveSessionsResponse]
+      #
+      # @see Openlayer::Models::InferencePipelineRetrieveSessionsParams
+      def retrieve_sessions(inference_pipeline_id, params = {})
+        query_params = [:asc, :page, :per_page, :sort_column]
+        parsed, options = Openlayer::InferencePipelineRetrieveSessionsParams.dump_request(params)
+        query = Openlayer::Internal::Util.encode_query_params(parsed.slice(*query_params))
+        @client.request(
+          method: :post,
+          path: ["inference-pipelines/%1$s/sessions", inference_pipeline_id],
+          query: query.transform_keys(per_page: "perPage", sort_column: "sortColumn"),
+          body: parsed.except(*query_params),
+          model: Openlayer::Models::InferencePipelineRetrieveSessionsResponse,
+          options: options
+        )
+      end
+
       # Get aggregated user data for an inference pipeline with pagination and metadata.
       #
       # Returns a list of users who have interacted with the inference pipeline,
       # including their activity statistics such as session counts, record counts, token
       # usage, and costs.
       #
-      # @overload retrieve_users(inference_pipeline_id, page: nil, per_page: nil, request_options: {})
+      # @overload retrieve_users(inference_pipeline_id, asc: nil, page: nil, per_page: nil, sort_column: nil, column_filters: nil, exclude_row_id_list: nil, not_search_query_and: nil, not_search_query_or: nil, row_id_list: nil, search_query_and: nil, search_query_or: nil, request_options: {})
       #
-      # @param inference_pipeline_id [String] The inference pipeline id (a UUID).
+      # @param inference_pipeline_id [String] Path param: The inference pipeline id (a UUID).
+      #
+      # @param asc [Boolean] Query param: Whether or not to sort on the sortColumn in ascending order.
+      #
+      # @param page [Integer] Query param: The page to return in a paginated query.
+      #
+      # @param per_page [Integer] Query param: Maximum number of items to return per page.
       #
-      # @param page [Integer] The page to return in a paginated query.
+      # @param sort_column [String] Query param: Name of the column to sort on
       #
-      # @param per_page [Integer] Maximum number of items to return per page.
+      # @param column_filters [Array<Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::SetColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::NumericColumnFilter, Openlayer::Models::InferencePipelineRetrieveUsersParams::ColumnFilter::StringColumnFilter>, nil] Body param
+      #
+      # @param exclude_row_id_list [Array<Integer>, nil] Body param
+      #
+      # @param not_search_query_and [Array<String>, nil] Body param
+      #
+      # @param not_search_query_or [Array<String>, nil] Body param
+      #
+      # @param row_id_list [Array<Integer>, nil] Body param
+      #
+      # @param search_query_and [Array<String>, nil] Body param
+      #
+      # @param search_query_or [Array<String>, nil] Body param
       #
       # @param request_options [Openlayer::RequestOptions, Hash{Symbol=>Object}, nil]
       #
@@ -108,12 +178,14 @@ module Openlayer
       #
       # @see Openlayer::Models::InferencePipelineRetrieveUsersParams
       def retrieve_users(inference_pipeline_id, params = {})
+        query_params = [:asc, :page, :per_page, :sort_column]
         parsed, options = Openlayer::InferencePipelineRetrieveUsersParams.dump_request(params)
-        query = Openlayer::Internal::Util.encode_query_params(parsed)
+        query = Openlayer::Internal::Util.encode_query_params(parsed.slice(*query_params))
         @client.request(
-          method: :get,
+          method: :post,
           path: ["inference-pipelines/%1$s/users", inference_pipeline_id],
-          query: query.transform_keys(per_page: "perPage"),
+          query: query.transform_keys(per_page: "perPage", sort_column: "sortColumn"),
+          body: parsed.except(*query_params),
           model: Openlayer::Models::InferencePipelineRetrieveUsersResponse,
           options: options
         )

data/lib/openlayer/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Openlayer
-  VERSION = "0.11.2"
+  VERSION = "0.12.0"
 end

data/lib/openlayer.rb

@@ -59,6 +59,8 @@ require_relative "openlayer/models/commits/test_result_list_response"
 require_relative "openlayer/models/inference_pipeline_delete_params"
 require_relative "openlayer/models/inference_pipeline_retrieve_params"
 require_relative "openlayer/models/inference_pipeline_retrieve_response"
+require_relative "openlayer/models/inference_pipeline_retrieve_sessions_params"
+require_relative "openlayer/models/inference_pipeline_retrieve_sessions_response"
 require_relative "openlayer/models/inference_pipeline_retrieve_users_params"
 require_relative "openlayer/models/inference_pipeline_retrieve_users_response"
 require_relative "openlayer/models/inference_pipelines/data_stream_params"