omniai-openai 1.2.1 → 1.3.1

data/lib/omniai/openai/thread/runs.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    class Thread
+      # An OpenAI scope for establishing messages.
+      class Runs
+        # @param client [OmniAI::OpenAI::Client] required
+        # @param thread [OmniAI::OpenAI::Thread] required
+        def initialize(client:, thread:)
+          @client = client
+          @thread = thread
+        end
+
+        # @param limit [Integer] optional
+        # @return [Array<OmniAI::Thread::Message>]
+        def all(limit:)
+          Run.all(thread_id: @thread.id, limit:, client: @client)
+        end
+
+        # @param id [String] required
+        # @return [OmniAI::OpenAI::Thread::Message]
+        def find(id:)
+          Run.find(id:, thread_id: @thread.id, client: @client)
+        end
+
+        # @param id [String] required
+        # @return [Hash]
+        def cancel!(id:)
+          Run.cancel!(id:, thread_id: @thread.id, client: @client)
+        end
+
+        # @param assistant_id [String] required
+        # @param model [String] optional
+        # @param temperature [Float] optional
+        # @param instructions [String] optional
+        # @param tools [Array<Hash>, nil] optional
+        # @param metadata [Hash, nil] optional
+        # @param client [OmniAI::OpenAI::Client] optional
+        # @return [OmniAI::OpenAI::Thread::Message]
+        def build(assistant_id:, model: nil, temperature: nil, instructions: nil, tools: nil, metadata: {})
+          Run.new(
+            assistant_id:,
+            thread_id: @thread.id,
+            model:,
+            temperature:,
+            instructions:,
+            tools:,
+            metadata:,
+            client: @client
+          )
+        end
+      end
+    end
+  end
+end

data/lib/omniai/openai/thread.rb

@@ -0,0 +1,141 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    # An OpenAI threads implementation.
+    class Thread
+      HEADERS = { 'OpenAI-Beta': 'assistants=v2' }.freeze
+
+      # @!attribute [rw] id
+      #   @return [String, nil]
+      attr_accessor :id
+
+      # @!attribute [rw] metadata
+      #   @return [Hash]
+      attr_accessor :metadata
+
+      # @!attribute [rw] tool_resources
+      #   @return [Hash]
+      attr_accessor :tool_resources
+
+      # @!attribute [rw] deleted
+      #   @return [Boolean, nil]
+      attr_accessor :deleted
+
+      # @param client [OmniAI::OpenAI::Client] optional
+      # @param id [String]
+      # @param metadata [String]
+      def initialize(
+        client: Client.new,
+        id: nil,
+        metadata: {},
+        tool_resources: {}
+      )
+        @client = client
+        @id = id
+        @metadata = metadata
+        @tool_resources = tool_resources
+      end
+
+      # @return [String]
+      def inspect
+        "#<#{self.class.name} id=#{@id.inspect}>"
+      end
+
+      # @param id [String] required
+      # @param client [OmniAI::OpenAI::Client] optional
+      # @return [OmniAI::OpenAI::Thread]
+      def self.find(id:, client: Client.new)
+        response = client.connection
+          .accept(:json)
+          .headers(HEADERS)
+          .get("/#{OmniAI::OpenAI::Client::VERSION}/threads/#{id}")
+
+        raise HTTPError, response.flush unless response.status.ok?
+
+        parse(data: response.parse)
+      end
+
+      # @param id [String] required
+      # @param client [OmniAI::OpenAI::Client] optional
+      # @return [Hash]
+      def self.destroy!(id:, client: Client.new)
+        response = client.connection
+          .accept(:json)
+          .headers(HEADERS)
+          .delete("/#{OmniAI::OpenAI::Client::VERSION}/threads/#{id}")
+
+        raise HTTPError, response.flush unless response.status.ok?
+
+        response.parse
+      end
+
+      # @raise [HTTPError]
+      # @return [OmniAI::OpenAI::Thread]
+      def save!
+        response = @client.connection
+          .accept(:json)
+          .headers(HEADERS)
+          .post("/#{OmniAI::OpenAI::Client::VERSION}/threads#{"/#{@id}" if @id}", json: payload)
+        raise HTTPError, response.flush unless response.status.ok?
+
+        parse(data: response.parse)
+        self
+      end
+
+      # @raise [OmniAI::Error]
+      # @return [OmniAI::OpenAI::Thread]
+      def destroy!
+        raise OmniAI::Error, 'cannot destroy a non-persisted thread' unless @id
+
+        data = self.class.destroy!(id: @id, client: @client)
+        @deleted = data['deleted']
+        self
+      end
+
+      # @return [OmniAI::OpenAI::Thread::Messages]
+      def messages
+        Messages.new(client: @client, thread: self)
+      end
+
+      # @return [OmniAI::OpenAI::Thread::Runs]
+      def runs
+        Runs.new(client: @client, thread: self)
+      end
+
+      private
+
+      class << self
+        private
+
+        # @param data [Hash] required
+        # @param client [OmniAI::OpenAI::Client] required
+        # @return [OmniAI::OpenAI::Thread]
+        def parse(data:, client: Client.new)
+          new(
+            client:,
+            id: data['id'],
+            metadata: data['metadata'],
+            tool_resources: data['tool_resources']
+          )
+        end
+      end
+
+      # @param data [Hash] required
+      # @return [OmniAI::OpenAI::Thread]
+      def parse(data:)
+        @id = data['id']
+        @metadata = data['metadata']
+        @tool_resources = data['tool_resources']
+      end
+
+      # @return [Hash]
+      def payload
+        {
+          metadata: @metadata,
+          tool_resources: @tool_resources,
+        }.compact
+      end
+    end
+  end
+end

data/lib/omniai/openai/threads.rb

@@ -0,0 +1,29 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    # An OpenAI scope for establishing threads.
+    class Threads
+      # @param client [OmniAI::OpenAI::Client] required
+      def initialize(client:)
+        @client = client
+      end
+
+      # @param id [String] required
+      def find(id:)
+        Thread.find(id:, client: @client)
+      end
+
+      # @param id [String] required
+      def destroy!(id:)
+        Thread.destroy!(id:, client: @client)
+      end
+
+      # @param metadata [Hash] optional
+      # @param tool_resources [Hash] optional
+      def build(metadata: {}, tool_resources: {})
+        Thread.new(metadata:, tool_resources:, client: @client)
+      end
+    end
+  end
+end

data/lib/omniai/openai/tool.rb

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+module OmniAI
+  module OpenAI
+    # An set of tools.
+    module Tool
+      FILE_SEARCH = { type: 'file_search' }.freeze
+      CODE_INTERPRETER = { type: 'code_interpreter' }.freeze
+    end
+  end
+end

data/lib/omniai/openai/version.rb

@@ -2,6 +2,6 @@
 
 module OmniAI
   module OpenAI
-    VERSION = '1.2.1'
+    VERSION = '1.3.1'
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai-openai
 version: !ruby/object:Gem::Version
-  version: 1.2.1
+  version: 1.3.1
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-06-20 00:00:00.000000000 Z
+date: 2024-06-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser
@@ -62,10 +62,21 @@ files:
 - Gemfile
 - README.md
 - lib/omniai/openai.rb
+- lib/omniai/openai/assistant.rb
+- lib/omniai/openai/assistants.rb
 - lib/omniai/openai/chat.rb
 - lib/omniai/openai/client.rb
 - lib/omniai/openai/config.rb
+- lib/omniai/openai/file.rb
+- lib/omniai/openai/files.rb
 - lib/omniai/openai/speak.rb
+- lib/omniai/openai/thread.rb
+- lib/omniai/openai/thread/message.rb
+- lib/omniai/openai/thread/messages.rb
+- lib/omniai/openai/thread/run.rb
+- lib/omniai/openai/thread/runs.rb
+- lib/omniai/openai/threads.rb
+- lib/omniai/openai/tool.rb
 - lib/omniai/openai/transcribe.rb
 - lib/omniai/openai/version.rb
 homepage: https://github.com/ksylvest/omniai-openai