deeprails 0.10.0 → 0.12.0

data/rbi/deeprails/models/defend_retrieve_workflow_params.rbi

@@ -14,15 +14,33 @@ module Deeprails
           )
         end
 
+      # Limit the number of returned events associated with this workflow. Defaults
+      # to 10.
+      sig { returns(T.nilable(Integer)) }
+      attr_reader :limit
+
+      sig { params(limit: Integer).void }
+      attr_writer :limit
+
       sig do
-        params(request_options: Deeprails::RequestOptions::OrHash).returns(
-          T.attached_class
-        )
+        params(
+          limit: Integer,
+          request_options: Deeprails::RequestOptions::OrHash
+        ).returns(T.attached_class)
       end
-      def self.new(request_options: {})
+      def self.new(
+        # Limit the number of returned events associated with this workflow. Defaults
+        # to 10.
+        limit: nil,
+        request_options: {}
+      )
       end
 
-      sig { override.returns({ request_options: Deeprails::RequestOptions }) }
+      sig do
+        override.returns(
+          { limit: Integer, request_options: Deeprails::RequestOptions }
+        )
+      end
       def to_hash
       end
     end

data/rbi/deeprails/models/defend_update_response.rbi

@@ -0,0 +1,83 @@
+# typed: strong
+
+module Deeprails
+  module Models
+    class DefendUpdateResponse < Deeprails::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(Deeprails::DefendUpdateResponse, Deeprails::Internal::AnyHash)
+        end
+
+      # The time the workflow was last modified in UTC.
+      sig { returns(Time) }
+      attr_accessor :modified_at
+
+      # Status of the selected workflow. May be `inactive` or `active`. Inactive
+      # workflows will not accept events.
+      sig { returns(Deeprails::DefendUpdateResponse::Status::TaggedSymbol) }
+      attr_accessor :status
+
+      # A unique workflow ID.
+      sig { returns(String) }
+      attr_accessor :workflow_id
+
+      sig do
+        params(
+          modified_at: Time,
+          status: Deeprails::DefendUpdateResponse::Status::OrSymbol,
+          workflow_id: String
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        # The time the workflow was last modified in UTC.
+        modified_at:,
+        # Status of the selected workflow. May be `inactive` or `active`. Inactive
+        # workflows will not accept events.
+        status:,
+        # A unique workflow ID.
+        workflow_id:
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            modified_at: Time,
+            status: Deeprails::DefendUpdateResponse::Status::TaggedSymbol,
+            workflow_id: String
+          }
+        )
+      end
+      def to_hash
+      end
+
+      # Status of the selected workflow. May be `inactive` or `active`. Inactive
+      # workflows will not accept events.
+      module Status
+        extend Deeprails::Internal::Type::Enum
+
+        TaggedSymbol =
+          T.type_alias do
+            T.all(Symbol, Deeprails::DefendUpdateResponse::Status)
+          end
+        OrSymbol = T.type_alias { T.any(Symbol, String) }
+
+        INACTIVE =
+          T.let(
+            :inactive,
+            Deeprails::DefendUpdateResponse::Status::TaggedSymbol
+          )
+        ACTIVE =
+          T.let(:active, Deeprails::DefendUpdateResponse::Status::TaggedSymbol)
+
+        sig do
+          override.returns(
+            T::Array[Deeprails::DefendUpdateResponse::Status::TaggedSymbol]
+          )
+        end
+        def self.values
+        end
+      end
+    end
+  end
+end

data/rbi/deeprails/models/file_response.rbi

@@ -8,13 +8,6 @@ module Deeprails
           T.any(Deeprails::FileResponse, Deeprails::Internal::AnyHash)
         end
 
-      # The time the file was created in UTC.
-      sig { returns(T.nilable(Time)) }
-      attr_reader :created_at
-
-      sig { params(created_at: Time).void }
-      attr_writer :created_at
-
       # A unique file ID.
       sig { returns(T.nilable(String)) }
       attr_reader :file_id
@@ -29,52 +22,31 @@ module Deeprails
       sig { params(file_name: String).void }
       attr_writer :file_name
 
-      # Path to the s3 bucket where the file is stored.
-      sig { returns(T.nilable(String)) }
-      attr_reader :file_path
-
-      sig { params(file_path: String).void }
-      attr_writer :file_path
-
-      # The most recent time the file was modified in UTC.
-      sig { returns(T.nilable(Time)) }
-      attr_reader :updated_at
+      # The size of the file in bytes.
+      sig { returns(T.nilable(Integer)) }
+      attr_reader :file_size
 
-      sig { params(updated_at: Time).void }
-      attr_writer :updated_at
+      sig { params(file_size: Integer).void }
+      attr_writer :file_size
 
       sig do
-        params(
-          created_at: Time,
-          file_id: String,
-          file_name: String,
-          file_path: String,
-          updated_at: Time
-        ).returns(T.attached_class)
+        params(file_id: String, file_name: String, file_size: Integer).returns(
+          T.attached_class
+        )
       end
       def self.new(
-        # The time the file was created in UTC.
-        created_at: nil,
         # A unique file ID.
         file_id: nil,
         # Name of the file.
         file_name: nil,
-        # Path to the s3 bucket where the file is stored.
-        file_path: nil,
-        # The most recent time the file was modified in UTC.
-        updated_at: nil
+        # The size of the file in bytes.
+        file_size: nil
       )
       end
 
       sig do
         override.returns(
-          {
-            created_at: Time,
-            file_id: String,
-            file_name: String,
-            file_path: String,
-            updated_at: Time
-          }
+          { file_id: String, file_name: String, file_size: Integer }
         )
       end
       def to_hash

data/rbi/deeprails/models/file_upload_params.rbi

@@ -12,12 +12,12 @@ module Deeprails
         end
 
       # The contents of the file to upload.
-      sig { returns(Deeprails::Internal::FileInput) }
+      sig { returns(T::Array[String]) }
       attr_accessor :file
 
       sig do
         params(
-          file: Deeprails::Internal::FileInput,
+          file: T::Array[String],
           request_options: Deeprails::RequestOptions::OrHash
         ).returns(T.attached_class)
       end
@@ -30,10 +30,7 @@ module Deeprails
 
       sig do
         override.returns(
-          {
-            file: Deeprails::Internal::FileInput,
-            request_options: Deeprails::RequestOptions
-          }
+          { file: T::Array[String], request_options: Deeprails::RequestOptions }
         )
       end
       def to_hash

data/rbi/deeprails/models/monitor_create_params.rbi

@@ -11,6 +11,17 @@ module Deeprails
           T.any(Deeprails::MonitorCreateParams, Deeprails::Internal::AnyHash)
         end
 
+      # An array of guardrail metrics that the model input and output pair will be
+      # evaluated on. For non-enterprise users, these will be limited to `correctness`,
+      # `completeness`, `instruction_adherence`, `context_adherence`,
+      # `ground_truth_adherence`, and/or `comprehensive_safety`.
+      sig do
+        returns(
+          T::Array[Deeprails::MonitorCreateParams::GuardrailMetric::OrSymbol]
+        )
+      end
+      attr_accessor :guardrail_metrics
+
       # Name of the new monitor.
       sig { returns(String) }
       attr_accessor :name
@@ -22,18 +33,47 @@ module Deeprails
       sig { params(description: String).void }
       attr_writer :description
 
+      # An array of file IDs to search in the monitor's evaluations. Files must be
+      # uploaded via the DeepRails API first.
+      sig { returns(T.nilable(T::Array[String])) }
+      attr_reader :file_search
+
+      sig { params(file_search: T::Array[String]).void }
+      attr_writer :file_search
+
+      # Whether to enable web search for this monitor's evaluations. Defaults to false.
+      sig { returns(T.nilable(T::Boolean)) }
+      attr_reader :web_search
+
+      sig { params(web_search: T::Boolean).void }
+      attr_writer :web_search
+
       sig do
         params(
+          guardrail_metrics:
+            T::Array[Deeprails::MonitorCreateParams::GuardrailMetric::OrSymbol],
           name: String,
           description: String,
+          file_search: T::Array[String],
+          web_search: T::Boolean,
           request_options: Deeprails::RequestOptions::OrHash
         ).returns(T.attached_class)
       end
       def self.new(
+        # An array of guardrail metrics that the model input and output pair will be
+        # evaluated on. For non-enterprise users, these will be limited to `correctness`,
+        # `completeness`, `instruction_adherence`, `context_adherence`,
+        # `ground_truth_adherence`, and/or `comprehensive_safety`.
+        guardrail_metrics:,
         # Name of the new monitor.
         name:,
         # Description of the new monitor.
         description: nil,
+        # An array of file IDs to search in the monitor's evaluations. Files must be
+        # uploaded via the DeepRails API first.
+        file_search: nil,
+        # Whether to enable web search for this monitor's evaluations. Defaults to false.
+        web_search: nil,
         request_options: {}
       )
       end
@@ -41,14 +81,71 @@ module Deeprails
       sig do
         override.returns(
           {
+            guardrail_metrics:
+              T::Array[
+                Deeprails::MonitorCreateParams::GuardrailMetric::OrSymbol
+              ],
             name: String,
             description: String,
+            file_search: T::Array[String],
+            web_search: T::Boolean,
             request_options: Deeprails::RequestOptions
           }
         )
       end
       def to_hash
       end
+
+      module GuardrailMetric
+        extend Deeprails::Internal::Type::Enum
+
+        TaggedSymbol =
+          T.type_alias do
+            T.all(Symbol, Deeprails::MonitorCreateParams::GuardrailMetric)
+          end
+        OrSymbol = T.type_alias { T.any(Symbol, String) }
+
+        CORRECTNESS =
+          T.let(
+            :correctness,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+        COMPLETENESS =
+          T.let(
+            :completeness,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+        INSTRUCTION_ADHERENCE =
+          T.let(
+            :instruction_adherence,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+        CONTEXT_ADHERENCE =
+          T.let(
+            :context_adherence,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+        GROUND_TRUTH_ADHERENCE =
+          T.let(
+            :ground_truth_adherence,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+        COMPREHENSIVE_SAFETY =
+          T.let(
+            :comprehensive_safety,
+            Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+          )
+
+        sig do
+          override.returns(
+            T::Array[
+              Deeprails::MonitorCreateParams::GuardrailMetric::TaggedSymbol
+            ]
+          )
+        end
+        def self.values
+        end
+      end
     end
   end
 end

data/rbi/deeprails/models/monitor_create_response.rbi

@@ -0,0 +1,83 @@
+# typed: strong
+
+module Deeprails
+  module Models
+    class MonitorCreateResponse < Deeprails::Internal::Type::BaseModel
+      OrHash =
+        T.type_alias do
+          T.any(Deeprails::MonitorCreateResponse, Deeprails::Internal::AnyHash)
+        end
+
+      # The time the monitor was created in UTC.
+      sig { returns(Time) }
+      attr_accessor :created_at
+
+      # A unique monitor ID.
+      sig { returns(String) }
+      attr_accessor :monitor_id
+
+      # Status of the monitor. Can be `active` or `inactive`. Inactive monitors no
+      # longer record and evaluate events.
+      sig { returns(Deeprails::MonitorCreateResponse::Status::TaggedSymbol) }
+      attr_accessor :status
+
+      sig do
+        params(
+          created_at: Time,
+          monitor_id: String,
+          status: Deeprails::MonitorCreateResponse::Status::OrSymbol
+        ).returns(T.attached_class)
+      end
+      def self.new(
+        # The time the monitor was created in UTC.
+        created_at:,
+        # A unique monitor ID.
+        monitor_id:,
+        # Status of the monitor. Can be `active` or `inactive`. Inactive monitors no
+        # longer record and evaluate events.
+        status:
+      )
+      end
+
+      sig do
+        override.returns(
+          {
+            created_at: Time,
+            monitor_id: String,
+            status: Deeprails::MonitorCreateResponse::Status::TaggedSymbol
+          }
+        )
+      end
+      def to_hash
+      end
+
+      # Status of the monitor. Can be `active` or `inactive`. Inactive monitors no
+      # longer record and evaluate events.
+      module Status
+        extend Deeprails::Internal::Type::Enum
+
+        TaggedSymbol =
+          T.type_alias do
+            T.all(Symbol, Deeprails::MonitorCreateResponse::Status)
+          end
+        OrSymbol = T.type_alias { T.any(Symbol, String) }
+
+        ACTIVE =
+          T.let(:active, Deeprails::MonitorCreateResponse::Status::TaggedSymbol)
+        INACTIVE =
+          T.let(
+            :inactive,
+            Deeprails::MonitorCreateResponse::Status::TaggedSymbol
+          )
+
+        sig do
+          override.returns(
+            T::Array[Deeprails::MonitorCreateResponse::Status::TaggedSymbol]
+          )
+        end
+        def self.values
+        end
+      end
+    end
+  end
+end