deeprails 0.18.0 → 0.23.0

data/sig/deeprails/internal/type/base_stream.rbs

@@ -0,0 +1,38 @@
+module Deeprails
+  module Internal
+    module Type
+      module BaseStream[Message, Elem]
+        include Enumerable[Elem]
+
+        attr_reader status: Integer
+
+        attr_reader headers: ::Hash[String, String]
+
+        def close: -> void
+
+        private def iterator: -> Enumerable[Elem]
+
+        def each: { (Elem arg0) -> void } -> void
+
+        def to_enum: -> Enumerator[Elem]
+
+        alias enum_for to_enum
+
+        def initialize: (
+          model: Class | Deeprails::Internal::Type::Converter,
+          url: URI::Generic,
+          status: Integer,
+          headers: ::Hash[String, String],
+          response: top,
+          unwrap: Symbol
+          | Integer
+          | ::Array[Symbol | Integer]
+          | ^(top arg0) -> top,
+          stream: Enumerable[Message]
+        ) -> void
+
+        def inspect: -> String
+      end
+    end
+  end
+end

data/sig/deeprails/models/defend_create_workflow_params.rbs

@@ -6,6 +6,7 @@ module Deeprails
         name: String,
         threshold_type: Deeprails::Models::DefendCreateWorkflowParams::threshold_type,
         automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendCreateWorkflowParams::automatic_hallucination_tolerance_level],
+        context_awareness: bool,
         custom_hallucination_threshold_values: ::Hash[Symbol, Float],
         description: String,
         file_search: ::Array[String],
@@ -30,6 +31,10 @@ module Deeprails
         ::Hash[Symbol, Deeprails::Models::DefendCreateWorkflowParams::automatic_hallucination_tolerance_level]
       ) -> ::Hash[Symbol, Deeprails::Models::DefendCreateWorkflowParams::automatic_hallucination_tolerance_level]
 
+      attr_reader context_awareness: bool?
+
+      def context_awareness=: (bool) -> bool
+
       attr_reader custom_hallucination_threshold_values: ::Hash[Symbol, Float]?
 
       def custom_hallucination_threshold_values=: (
@@ -57,6 +62,7 @@ module Deeprails
         name: String,
         threshold_type: Deeprails::Models::DefendCreateWorkflowParams::threshold_type,
         ?automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendCreateWorkflowParams::automatic_hallucination_tolerance_level],
+        ?context_awareness: bool,
         ?custom_hallucination_threshold_values: ::Hash[Symbol, Float],
         ?description: String,
         ?file_search: ::Array[String],
@@ -70,6 +76,7 @@ module Deeprails
         name: String,
         threshold_type: Deeprails::Models::DefendCreateWorkflowParams::threshold_type,
         automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendCreateWorkflowParams::automatic_hallucination_tolerance_level],
+        context_awareness: bool,
         custom_hallucination_threshold_values: ::Hash[Symbol, Float],
         description: String,
         file_search: ::Array[String],

data/sig/deeprails/models/defend_response.rbs

@@ -115,13 +115,19 @@ module Deeprails
 
       type event =
         {
+          billing_request_id: String,
           evaluations: ::Array[Deeprails::DefendResponse::Event::Evaluation],
           event_id: String,
           improved_model_output: String,
-          improvement_tool_status: String
+          improvement_tool_status: Deeprails::Models::DefendResponse::Event::improvement_tool_status,
+          status: Deeprails::Models::DefendResponse::Event::status
         }
 
       class Event < Deeprails::Internal::Type::BaseModel
+        attr_reader billing_request_id: String?
+
+        def billing_request_id=: (String) -> String
+
         attr_reader evaluations: ::Array[Deeprails::DefendResponse::Event::Evaluation]?
 
         def evaluations=: (
@@ -136,26 +142,39 @@ module Deeprails
 
         def improved_model_output=: (String) -> String
 
-        attr_reader improvement_tool_status: String?
+        attr_reader improvement_tool_status: Deeprails::Models::DefendResponse::Event::improvement_tool_status?
+
+        def improvement_tool_status=: (
+          Deeprails::Models::DefendResponse::Event::improvement_tool_status
+        ) -> Deeprails::Models::DefendResponse::Event::improvement_tool_status
+
+        attr_reader status: Deeprails::Models::DefendResponse::Event::status?
 
-        def improvement_tool_status=: (String) -> String
+        def status=: (
+          Deeprails::Models::DefendResponse::Event::status
+        ) -> Deeprails::Models::DefendResponse::Event::status
 
         def initialize: (
+          ?billing_request_id: String,
           ?evaluations: ::Array[Deeprails::DefendResponse::Event::Evaluation],
           ?event_id: String,
           ?improved_model_output: String,
-          ?improvement_tool_status: String
+          ?improvement_tool_status: Deeprails::Models::DefendResponse::Event::improvement_tool_status,
+          ?status: Deeprails::Models::DefendResponse::Event::status
         ) -> void
 
         def to_hash: -> {
+          billing_request_id: String,
           evaluations: ::Array[Deeprails::DefendResponse::Event::Evaluation],
           event_id: String,
           improved_model_output: String,
-          improvement_tool_status: String
+          improvement_tool_status: Deeprails::Models::DefendResponse::Event::improvement_tool_status,
+          status: Deeprails::Models::DefendResponse::Event::status
         }
 
         type evaluation =
           {
+            analysis_of_failures: String,
             attempt: String,
             created_at: Time,
             error_message: String,
@@ -163,6 +182,8 @@ module Deeprails
             evaluation_status: String,
             evaluation_total_cost: Float,
             guardrail_metrics: ::Array[String],
+            improvement_tool_status: Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status,
+            key_improvements: ::Array[String],
             model_input: ::Hash[Symbol, top],
             model_output: String,
             modified_at: Time,
@@ -172,6 +193,10 @@ module Deeprails
           }
 
         class Evaluation < Deeprails::Internal::Type::BaseModel
+          attr_reader analysis_of_failures: String?
+
+          def analysis_of_failures=: (String) -> String
+
           attr_reader attempt: String?
 
           def attempt=: (String) -> String
@@ -200,6 +225,16 @@ module Deeprails
 
           def guardrail_metrics=: (::Array[String]) -> ::Array[String]
 
+          attr_reader improvement_tool_status: Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status?
+
+          def improvement_tool_status=: (
+            Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status
+          ) -> Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status
+
+          attr_reader key_improvements: ::Array[String]?
+
+          def key_improvements=: (::Array[String]) -> ::Array[String]
+
           attr_reader model_input: ::Hash[Symbol, top]?
 
           def model_input=: (::Hash[Symbol, top]) -> ::Hash[Symbol, top]
@@ -225,6 +260,7 @@ module Deeprails
           def run_mode=: (String) -> String
 
           def initialize: (
+            ?analysis_of_failures: String,
             ?attempt: String,
             ?created_at: Time,
             ?error_message: String,
@@ -232,6 +268,8 @@ module Deeprails
             ?evaluation_status: String,
             ?evaluation_total_cost: Float,
             ?guardrail_metrics: ::Array[String],
+            ?improvement_tool_status: Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status,
+            ?key_improvements: ::Array[String],
             ?model_input: ::Hash[Symbol, top],
             ?model_output: String,
             ?modified_at: Time,
@@ -241,6 +279,7 @@ module Deeprails
           ) -> void
 
           def to_hash: -> {
+            analysis_of_failures: String,
             attempt: String,
             created_at: Time,
             error_message: String,
@@ -248,6 +287,8 @@ module Deeprails
             evaluation_status: String,
             evaluation_total_cost: Float,
             guardrail_metrics: ::Array[String],
+            improvement_tool_status: Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status,
+            key_improvements: ::Array[String],
             model_input: ::Hash[Symbol, top],
             model_output: String,
             modified_at: Time,
@@ -255,10 +296,63 @@ module Deeprails
             progress: Integer,
             run_mode: String
           }
+
+          type improvement_tool_status =
+            :improved
+            | :improvement_failed
+            | :no_improvement_required
+            | :improvement_required
+
+          module ImprovementToolStatus
+            extend Deeprails::Internal::Type::Enum
+
+            IMPROVED: :improved
+            IMPROVEMENT_FAILED: :improvement_failed
+            NO_IMPROVEMENT_REQUIRED: :no_improvement_required
+            IMPROVEMENT_REQUIRED: :improvement_required
+
+            def self?.values: -> ::Array[Deeprails::Models::DefendResponse::Event::Evaluation::improvement_tool_status]
+          end
+        end
+
+        type improvement_tool_status =
+          :improved
+          | :improvement_failed
+          | :no_improvement_required
+          | :improvement_required
+
+        module ImprovementToolStatus
+          extend Deeprails::Internal::Type::Enum
+
+          IMPROVED: :improved
+          IMPROVEMENT_FAILED: :improvement_failed
+          NO_IMPROVEMENT_REQUIRED: :no_improvement_required
+          IMPROVEMENT_REQUIRED: :improvement_required
+
+          def self?.values: -> ::Array[Deeprails::Models::DefendResponse::Event::improvement_tool_status]
+        end
+
+        type status = :completed | :failed | :in_progress
+
+        module Status
+          extend Deeprails::Internal::Type::Enum
+
+          COMPLETED: :completed
+          FAILED: :failed
+          IN_PROGRESS: :in_progress
+
+          def self?.values: -> ::Array[Deeprails::Models::DefendResponse::Event::status]
         end
       end
 
-      type file = { file_id: String, file_name: String, file_size: Integer }
+      type file =
+        {
+          file_id: String,
+          file_name: String,
+          file_size: Integer,
+          presigned_url: String,
+          presigned_url_expires_at: Time
+        }
 
       class File < Deeprails::Internal::Type::BaseModel
         attr_reader file_id: String?
@@ -273,16 +367,28 @@ module Deeprails
 
         def file_size=: (Integer) -> Integer
 
+        attr_reader presigned_url: String?
+
+        def presigned_url=: (String) -> String
+
+        attr_reader presigned_url_expires_at: Time?
+
+        def presigned_url_expires_at=: (Time) -> Time
+
         def initialize: (
           ?file_id: String,
           ?file_name: String,
-          ?file_size: Integer
+          ?file_size: Integer,
+          ?presigned_url: String,
+          ?presigned_url_expires_at: Time
         ) -> void
 
         def to_hash: -> {
           file_id: String,
           file_name: String,
-          file_size: Integer
+          file_size: Integer,
+          presigned_url: String,
+          presigned_url_expires_at: Time
         }
       end
 

data/sig/deeprails/models/defend_submit_and_stream_event_params.rbs

@@ -0,0 +1,74 @@
+module Deeprails
+  module Models
+    type defend_submit_and_stream_event_params =
+      {
+        model_input: ::Hash[Symbol, top],
+        model_output: String,
+        model_used: String,
+        run_mode: Deeprails::Models::DefendSubmitAndStreamEventParams::run_mode,
+        stream: bool,
+        nametag: String
+      }
+      & Deeprails::Internal::Type::request_parameters
+
+    class DefendSubmitAndStreamEventParams < Deeprails::Internal::Type::BaseModel
+      extend Deeprails::Internal::Type::RequestParameters::Converter
+      include Deeprails::Internal::Type::RequestParameters
+
+      attr_accessor model_input: ::Hash[Symbol, top]
+
+      attr_accessor model_output: String
+
+      attr_accessor model_used: String
+
+      attr_accessor run_mode: Deeprails::Models::DefendSubmitAndStreamEventParams::run_mode
+
+      attr_reader stream: bool?
+
+      def stream=: (bool) -> bool
+
+      attr_reader nametag: String?
+
+      def nametag=: (String) -> String
+
+      def initialize: (
+        model_input: ::Hash[Symbol, top],
+        model_output: String,
+        model_used: String,
+        run_mode: Deeprails::Models::DefendSubmitAndStreamEventParams::run_mode,
+        ?stream: bool,
+        ?nametag: String,
+        ?request_options: Deeprails::request_opts
+      ) -> void
+
+      def to_hash: -> {
+        model_input: ::Hash[Symbol, top],
+        model_output: String,
+        model_used: String,
+        run_mode: Deeprails::Models::DefendSubmitAndStreamEventParams::run_mode,
+        stream: bool,
+        nametag: String,
+        request_options: Deeprails::RequestOptions
+      }
+
+      type run_mode =
+        :fast
+        | :precision
+        | :precision_codex
+        | :precision_max
+        | :precision_max_codex
+
+      module RunMode
+        extend Deeprails::Internal::Type::Enum
+
+        FAST: :fast
+        PRECISION: :precision
+        PRECISION_CODEX: :precision_codex
+        PRECISION_MAX: :precision_max
+        PRECISION_MAX_CODEX: :precision_max_codex
+
+        def self?.values: -> ::Array[Deeprails::Models::DefendSubmitAndStreamEventParams::run_mode]
+      end
+    end
+  end
+end

data/sig/deeprails/models/defend_submit_and_stream_event_response.rbs

@@ -0,0 +1,5 @@
+module Deeprails
+  module Models
+    class DefendSubmitAndStreamEventResponse = String
+  end
+end

data/sig/deeprails/models/defend_submit_event_params.rbs

@@ -45,9 +45,22 @@ module Deeprails
       }
 
       type model_input =
-        { ground_truth: String, system_prompt: String, user_prompt: String }
+        {
+          user_prompt: String,
+          context: ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context],
+          ground_truth: String,
+          system_prompt: String
+        }
 
       class ModelInput < Deeprails::Internal::Type::BaseModel
+        attr_accessor user_prompt: String
+
+        attr_reader context: ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context]?
+
+        def context=: (
+          ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context]
+        ) -> ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context]
+
         attr_reader ground_truth: String?
 
         def ground_truth=: (String) -> String
@@ -56,28 +69,44 @@ module Deeprails
 
         def system_prompt=: (String) -> String
 
-        attr_reader user_prompt: String?
-
-        def user_prompt=: (String) -> String
-
         def initialize: (
+          user_prompt: String,
+          ?context: ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context],
           ?ground_truth: String,
-          ?system_prompt: String,
-          ?user_prompt: String
+          ?system_prompt: String
         ) -> void
 
         def to_hash: -> {
+          user_prompt: String,
+          context: ::Array[Deeprails::DefendSubmitEventParams::ModelInput::Context],
           ground_truth: String,
-          system_prompt: String,
-          user_prompt: String
+          system_prompt: String
         }
+
+        type context = { content: String, role: String }
+
+        class Context < Deeprails::Internal::Type::BaseModel
+          attr_reader content: String?
+
+          def content=: (String) -> String
+
+          attr_reader role: String?
+
+          def role=: (String) -> String
+
+          def initialize: (?content: String, ?role: String) -> void
+
+          def to_hash: -> { content: String, role: String }
+        end
       end
 
-      type run_mode = :precision_plus | :precision | :smart | :economy
+      type run_mode =
+        :precision_plus_codex | :precision_plus | :precision | :smart | :economy
 
       module RunMode
         extend Deeprails::Internal::Type::Enum
 
+        PRECISION_PLUS_CODEX: :precision_plus_codex
         PRECISION_PLUS: :precision_plus
         PRECISION: :precision
         SMART: :smart

data/sig/deeprails/models/defend_update_response.rbs

@@ -4,7 +4,8 @@ module Deeprails
       {
         modified_at: Time,
         status: Deeprails::Models::DefendUpdateResponse::status,
-        workflow_id: String
+        workflow_id: String,
+        name: String
       }
 
     class DefendUpdateResponse < Deeprails::Internal::Type::BaseModel
@@ -14,16 +15,22 @@ module Deeprails
 
       attr_accessor workflow_id: String
 
+      attr_reader name: String?
+
+      def name=: (String) -> String
+
       def initialize: (
         modified_at: Time,
         status: Deeprails::Models::DefendUpdateResponse::status,
-        workflow_id: String
+        workflow_id: String,
+        ?name: String
       ) -> void
 
       def to_hash: -> {
         modified_at: Time,
         status: Deeprails::Models::DefendUpdateResponse::status,
-        workflow_id: String
+        workflow_id: String,
+        name: String
       }
 
       type status = :inactive | :active

data/sig/deeprails/models/defend_update_workflow_params.rbs

@@ -1,32 +1,134 @@
 module Deeprails
   module Models
     type defend_update_workflow_params =
-      { description: String, name: String }
+      {
+        automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level],
+        context_awareness: bool,
+        custom_hallucination_threshold_values: ::Hash[Symbol, Float],
+        description: String,
+        file_search: ::Array[String],
+        improvement_action: Deeprails::Models::DefendUpdateWorkflowParams::improvement_action,
+        max_improvement_attempts: Integer,
+        name: String,
+        threshold_type: Deeprails::Models::DefendUpdateWorkflowParams::threshold_type,
+        web_search: bool
+      }
       & Deeprails::Internal::Type::request_parameters
 
     class DefendUpdateWorkflowParams < Deeprails::Internal::Type::BaseModel
       extend Deeprails::Internal::Type::RequestParameters::Converter
       include Deeprails::Internal::Type::RequestParameters
 
+      attr_reader automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level]?
+
+      def automatic_hallucination_tolerance_levels=: (
+        ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level]
+      ) -> ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level]
+
+      attr_reader context_awareness: bool?
+
+      def context_awareness=: (bool) -> bool
+
+      attr_reader custom_hallucination_threshold_values: ::Hash[Symbol, Float]?
+
+      def custom_hallucination_threshold_values=: (
+        ::Hash[Symbol, Float]
+      ) -> ::Hash[Symbol, Float]
+
       attr_reader description: String?
 
       def description=: (String) -> String
 
+      attr_reader file_search: ::Array[String]?
+
+      def file_search=: (::Array[String]) -> ::Array[String]
+
+      attr_reader improvement_action: Deeprails::Models::DefendUpdateWorkflowParams::improvement_action?
+
+      def improvement_action=: (
+        Deeprails::Models::DefendUpdateWorkflowParams::improvement_action
+      ) -> Deeprails::Models::DefendUpdateWorkflowParams::improvement_action
+
+      attr_reader max_improvement_attempts: Integer?
+
+      def max_improvement_attempts=: (Integer) -> Integer
+
       attr_reader name: String?
 
       def name=: (String) -> String
 
+      attr_reader threshold_type: Deeprails::Models::DefendUpdateWorkflowParams::threshold_type?
+
+      def threshold_type=: (
+        Deeprails::Models::DefendUpdateWorkflowParams::threshold_type
+      ) -> Deeprails::Models::DefendUpdateWorkflowParams::threshold_type
+
+      attr_reader web_search: bool?
+
+      def web_search=: (bool) -> bool
+
       def initialize: (
+        ?automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level],
+        ?context_awareness: bool,
+        ?custom_hallucination_threshold_values: ::Hash[Symbol, Float],
         ?description: String,
+        ?file_search: ::Array[String],
+        ?improvement_action: Deeprails::Models::DefendUpdateWorkflowParams::improvement_action,
+        ?max_improvement_attempts: Integer,
         ?name: String,
+        ?threshold_type: Deeprails::Models::DefendUpdateWorkflowParams::threshold_type,
+        ?web_search: bool,
         ?request_options: Deeprails::request_opts
       ) -> void
 
       def to_hash: -> {
+        automatic_hallucination_tolerance_levels: ::Hash[Symbol, Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level],
+        context_awareness: bool,
+        custom_hallucination_threshold_values: ::Hash[Symbol, Float],
         description: String,
+        file_search: ::Array[String],
+        improvement_action: Deeprails::Models::DefendUpdateWorkflowParams::improvement_action,
+        max_improvement_attempts: Integer,
         name: String,
+        threshold_type: Deeprails::Models::DefendUpdateWorkflowParams::threshold_type,
+        web_search: bool,
         request_options: Deeprails::RequestOptions
       }
+
+      type automatic_hallucination_tolerance_level = :low | :medium | :high
+
+      module AutomaticHallucinationToleranceLevel
+        extend Deeprails::Internal::Type::Enum
+
+        LOW: :low
+        MEDIUM: :medium
+        HIGH: :high
+
+        def self?.values: -> ::Array[Deeprails::Models::DefendUpdateWorkflowParams::automatic_hallucination_tolerance_level]
+      end
+
+      type improvement_action = :regen | :fixit | :do_nothing
+
+      module ImprovementAction
+        extend Deeprails::Internal::Type::Enum
+
+        REGEN: :regen
+        FIXIT: :fixit
+        DO_NOTHING: :do_nothing
+
+        def self?.values: -> ::Array[Deeprails::Models::DefendUpdateWorkflowParams::improvement_action]
+      end
+
+      type threshold_type = :automatic | :custom
+
+      module ThresholdType
+        extend Deeprails::Internal::Type::Enum
+
+        AUTOMATIC: :automatic
+        CUSTOM: :custom
+
+        def self?.values: -> ::Array[Deeprails::Models::DefendUpdateWorkflowParams::threshold_type]
+      end
     end
   end
 end

data/sig/deeprails/models/monitor_create_params.rbs

@@ -4,6 +4,7 @@ module Deeprails
       {
         guardrail_metrics: ::Array[Deeprails::Models::MonitorCreateParams::guardrail_metric],
         name: String,
+        context_awareness: bool,
         description: String,
         file_search: ::Array[String],
         web_search: bool
@@ -18,6 +19,10 @@ module Deeprails
 
       attr_accessor name: String
 
+      attr_reader context_awareness: bool?
+
+      def context_awareness=: (bool) -> bool
+
       attr_reader description: String?
 
       def description=: (String) -> String
@@ -33,6 +38,7 @@ module Deeprails
       def initialize: (
         guardrail_metrics: ::Array[Deeprails::Models::MonitorCreateParams::guardrail_metric],
         name: String,
+        ?context_awareness: bool,
         ?description: String,
         ?file_search: ::Array[String],
         ?web_search: bool,
@@ -42,6 +48,7 @@ module Deeprails
       def to_hash: -> {
         guardrail_metrics: ::Array[Deeprails::Models::MonitorCreateParams::guardrail_metric],
         name: String,
+        context_awareness: bool,
         description: String,
         file_search: ::Array[String],
         web_search: bool,