vellum_ai 0.3.6 → 0.3.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 182d24bd91fdd87ff3a318f7461005f46d29728410b1c2fc5db8304def338ff6
-  data.tar.gz: 291e636ee7663835359bec22c9e6cc34aba510b683e34c009bcccfdbae4fb617
+  metadata.gz: f8ab21719c1b255f61fdfec3f4b59311f8ff6abac1fd53084179ad36741f3d1d
+  data.tar.gz: 56727296184e00efcb37b0277e78b15d2fbb4666c767e6d0519c04f781da4d22
 SHA512:
-  metadata.gz: c88379aa69acc559f718f130fab218cd243dc93ba0dddc9b3b260ea152b8da8508b1c4fffdeec602738ddfc4a5828dbb186912b4f934bc9fe296bb2bd3abcfdc
-  data.tar.gz: 4caecedbcf0b68fffb1d1439bf8cbb444eefac11bedfb144f9fdb97cefbde163e3533b7654651528a7cee072a095ad83a635281b4ed5dbc1a470641f6141493b
+  metadata.gz: ed3a4f2da49c992af2c410a3b170ddc23da6b9e150707d88948739714d244a068aaf202c60f625850d5608251cdae77652733352677ea1c5c6cafadd918d71c5
+  data.tar.gz: '06927b37c0254a198d2928e32de4d2f40620dc4e8c1317ddad49dd676fae8008f8ccb13f4486cac478773f4543a561c977eb1e9639691d8d369d033dfa5b43c9'

data/lib/requests.rb

@@ -20,7 +20,7 @@ module Vellum
       @headers = {
         "X-Fern-Language": "Ruby",
         "X-Fern-SDK-Name": "Vellum",
-        "X-Fern-SDK-Version": "0.3.6",
+        "X-Fern-SDK-Version": "0.3.7",
         "X_API_KEY": api_key.to_s
       }
       @conn = Faraday.new(headers: @headers) do |faraday|
@@ -46,7 +46,7 @@ module Vellum
       @headers = {
         "X-Fern-Language": "Ruby",
         "X-Fern-SDK-Name": "Vellum",
-        "X-Fern-SDK-Version": "0.3.6",
+        "X-Fern-SDK-Version": "0.3.7",
         "X_API_KEY": api_key.to_s
       }
       @conn = Faraday.new(headers: @headers) do |faraday|

data/lib/vellum_ai/folder_entities/client.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require_relative "../../requests"
+require "async"
+
+module Vellum
+  class FolderEntitiesClient
+    attr_reader :request_client
+
+    # @param request_client [RequestClient]
+    # @return [FolderEntitiesClient]
+    def initialize(request_client:)
+      # @type [RequestClient]
+      @request_client = request_client
+    end
+
+    # Add an entity to a specific folder or root directory.
+    #
+    # Adding an entity to a folder will remove it from any other folders it might have been a member of.
+    #
+    # @param folder_id [String] The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root directory (e.g. "PROMPT_SANDBOX").
+    # @param entity_id [String] The ID of the entity you would like to move.
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def add_entity_to_folder(folder_id:, entity_id:, request_options: nil)
+      @request_client.conn.post do |req|
+        req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+        req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+        req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+        req.body = { **(request_options&.additional_body_parameters || {}), entity_id: entity_id }.compact
+        req.url "#{@request_client.default_environment[:Default]}/v1/folders/#{folder_id}/add-entity"
+      end
+    end
+  end
+
+  class AsyncFolderEntitiesClient
+    attr_reader :request_client
+
+    # @param request_client [AsyncRequestClient]
+    # @return [AsyncFolderEntitiesClient]
+    def initialize(request_client:)
+      # @type [AsyncRequestClient]
+      @request_client = request_client
+    end
+
+    # Add an entity to a specific folder or root directory.
+    #
+    # Adding an entity to a folder will remove it from any other folders it might have been a member of.
+    #
+    # @param folder_id [String] The ID of the folder to which the entity should be added. This can be a UUID of a folder, or the name of a root directory (e.g. "PROMPT_SANDBOX").
+    # @param entity_id [String] The ID of the entity you would like to move.
+    # @param request_options [RequestOptions]
+    # @return [Void]
+    def add_entity_to_folder(folder_id:, entity_id:, request_options: nil)
+      Async do
+        @request_client.conn.post do |req|
+          req.options.timeout = request_options.timeout_in_seconds unless request_options&.timeout_in_seconds.nil?
+          req.headers["X_API_KEY"] = request_options.api_key unless request_options&.api_key.nil?
+          req.headers = { **req.headers, **(request_options&.additional_headers || {}) }.compact
+          req.body = { **(request_options&.additional_body_parameters || {}), entity_id: entity_id }.compact
+          req.url "#{@request_client.default_environment[:Default]}/v1/folders/#{folder_id}/add-entity"
+        end
+      end
+    end
+  end
+end

data/lib/vellum_ai.rb

@@ -6,6 +6,7 @@ require_relative "requests"
 require_relative "vellum_ai/deployments/client"
 require_relative "vellum_ai/document_indexes/client"
 require_relative "vellum_ai/documents/client"
+require_relative "vellum_ai/folder_entities/client"
 require_relative "vellum_ai/model_versions/client"
 require_relative "vellum_ai/registered_prompts/client"
 require_relative "vellum_ai/sandboxes/client"
@@ -27,8 +28,8 @@ require_relative "vellum_ai/types/submit_workflow_execution_actual_request"
 
 module Vellum
   class Client
-    attr_reader :deployments, :document_indexes, :documents, :model_versions, :registered_prompts, :sandboxes,
-                :test_suites, :workflow_deployments
+    attr_reader :deployments, :document_indexes, :documents, :folder_entities, :model_versions, :registered_prompts,
+                :sandboxes, :test_suites, :workflow_deployments
 
     # @param environment [Environment]
     # @param max_retries [Long] The number of times to retry a failed request, defaults to 2.
@@ -41,6 +42,7 @@ module Vellum
       @deployments = DeploymentsClient.new(request_client: @request_client)
       @document_indexes = DocumentIndexesClient.new(request_client: @request_client)
       @documents = DocumentsClient.new(request_client: @request_client)
+      @folder_entities = FolderEntitiesClient.new(request_client: @request_client)
       @model_versions = ModelVersionsClient.new(request_client: @request_client)
       @registered_prompts = RegisteredPromptsClient.new(request_client: @request_client)
       @sandboxes = SandboxesClient.new(request_client: @request_client)
@@ -248,8 +250,8 @@ module Vellum
   end
 
   class AsyncClient
-    attr_reader :deployments, :document_indexes, :documents, :model_versions, :registered_prompts, :sandboxes,
-                :test_suites, :workflow_deployments
+    attr_reader :deployments, :document_indexes, :documents, :folder_entities, :model_versions, :registered_prompts,
+                :sandboxes, :test_suites, :workflow_deployments
 
     # @param environment [Environment]
     # @param max_retries [Long] The number of times to retry a failed request, defaults to 2.
@@ -262,6 +264,7 @@ module Vellum
       @deployments = AsyncDeploymentsClient.new(request_client: @async_request_client)
       @document_indexes = AsyncDocumentIndexesClient.new(request_client: @async_request_client)
       @documents = AsyncDocumentsClient.new(request_client: @async_request_client)
+      @folder_entities = AsyncFolderEntitiesClient.new(request_client: @async_request_client)
       @model_versions = AsyncModelVersionsClient.new(request_client: @async_request_client)
       @registered_prompts = AsyncRegisteredPromptsClient.new(request_client: @async_request_client)
       @sandboxes = AsyncSandboxesClient.new(request_client: @async_request_client)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: vellum_ai
 version: !ruby/object:Gem::Version
-  version: 0.3.6
+  version: 0.3.7
 platform: ruby
 authors:
 - Vellum
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-02-21 00:00:00.000000000 Z
+date: 2024-02-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: async-http-faraday
@@ -96,6 +96,7 @@ files:
 - lib/vellum_ai/deployments/types/deployments_list_request_status.rb
 - lib/vellum_ai/document_indexes/client.rb
 - lib/vellum_ai/documents/client.rb
+- lib/vellum_ai/folder_entities/client.rb
 - lib/vellum_ai/model_versions/client.rb
 - lib/vellum_ai/registered_prompts/client.rb
 - lib/vellum_ai/sandboxes/client.rb