cadenya 1.5.0 → 1.7.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 223bf6181a9de37bd24bf02abb889e1473bb6ed030913e6bde182a41d2fa8e68
-  data.tar.gz: e57f24e581cb8015bccb16a788c9693f93c8c64cd8b1c501016e559033a0b711
+  metadata.gz: f5648f64510c4220cdd7bbe9ad1c9d1f05b49640af22106da25282673c37510d
+  data.tar.gz: 288d867f32173e26da0c449830fc32f48eaa8e610ba825439a26dc8c4cc3e60e
 SHA512:
-  metadata.gz: 3bd3b1fd7aa13a95d4d77b2e1d75328b181e4d5bb9001b32422c5737ca48335cb222817a5ebae7205573b71537ca2e88282c05ad775c5d150d9cb60418d23fbb
-  data.tar.gz: 71e1e7f8e34ec480611fa8eda3a27fc623b83d5857d2d4110750905f4fa47974dda6cf5a8d9f9f2f1bc3c5abbef7328730ea696fb545ce53786cf67c51a86499
+  metadata.gz: b77e5f294970813230c90859e0e0efbd5b6ff056c30b0a56ef56f5f77fe10d6e33038ec7bf780e6b8d8afa719271333a07e16d7ef588ba7df59944524c4d5ed4
+  data.tar.gz: 21d63e6814ad922be197ebf9e198f320eadd93f412f0a6c3127e81b7ef0040c8f20a573bacc54cd70dde1a69db69f42361ca4e19c8f9e32190d27a2772c33ffc

data/api.md

@@ -136,7 +136,7 @@ client.workspace_admin.remove_member(profile_id, workspace_id: nil) # => nil
 Retrieves the profile for the credentials accessing the API
 
 ```ruby
-client.profiles.whoami() # => Cadenya::Types::Profile
+client.profiles.whoami() # => Cadenya::Types::WhoamiResponse
 ```
 
 ## client.workspaces
@@ -147,6 +147,54 @@ List workspaces
 client.workspaces.list(limit: nil, cursor: nil, sort_order: nil, include_info: nil, labels: nil) # => Cadenya::Page of Cadenya::Types::Workspace
 ```
 
+## client.agent_pools
+
+List agent pools
+
+```ruby
+client.agent_pools.list(workspace_id: nil, limit: nil, cursor: nil, prefix: nil, query: nil, state: nil, labels: nil, sort_order: nil, include_info: nil) # => Cadenya::Page of Cadenya::Types::AgentPool
+```
+Create a new agent pool
+
+```ruby
+client.agent_pools.create(metadata:, spec:, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+Get an agent pool by ID
+
+```ruby
+client.agent_pools.retrieve(id, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+Delete an agent pool
+
+```ruby
+client.agent_pools.delete(id, workspace_id: nil) # => nil
+```
+Update an agent pool
+
+```ruby
+client.agent_pools.update(id, workspace_id: nil, metadata: nil, spec: nil, update_mask: nil) # => Cadenya::Types::AgentPool
+```
+Activate an agent pool
+
+```ruby
+client.agent_pools.activate(id, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+Archive an agent pool
+
+```ruby
+client.agent_pools.archive(id, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+Deactivate an agent pool
+
+```ruby
+client.agent_pools.deactivate(id, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+Unarchive an agent pool
+
+```ruby
+client.agent_pools.unarchive(id, workspace_id: nil) # => Cadenya::Types::AgentPool
+```
+
 ## client.agents
 
 List agents
@@ -433,7 +481,7 @@ client.objectives.list(workspace_id: nil, limit: nil, cursor: nil, agent_id: nil
 Create a new objective
 
 ```ruby
-client.objectives.create(agent_id:, system_prompt_data:, workspace_id: nil, variation_id: nil, metadata: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil) # => Cadenya::Types::Objective
+client.objectives.create(agent_id:, workspace_id: nil, variation_id: nil, metadata: nil, system_prompt_data: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil) # => Cadenya::Types::Objective
 ```
 Get an objective by ID
 
@@ -470,6 +518,16 @@ Submit feedback for an objective
 ```ruby
 client.objectives.create_feedback(objective_id, metadata:, data:, workspace_id: nil) # => Cadenya::Types::ObjectiveFeedback
 ```
+List objective queued actions
+
+```ruby
+client.objectives.list_queued_actions(objective_id, workspace_id: nil, limit: nil, cursor: nil, state: nil) # => Cadenya::Page of Cadenya::Types::ObjectiveQueuedAction
+```
+Remove a queued action
+
+```ruby
+client.objectives.remove_queued_action(objective_id, queued_action_id:, workspace_id: nil) # => Cadenya::Types::ObjectiveQueuedAction
+```
 List objective tool calls
 
 ```ruby
@@ -508,12 +566,22 @@ client.objectives.cancel(objective_id, workspace_id: nil, reason: nil) # => Cade
 Compact an objective
 
 ```ruby
-client.objectives.compact(objective_id, workspace_id: nil, compaction_config: nil) # => Cadenya::Types::CompactObjectiveResponse
+client.objectives.compact(objective_id, workspace_id: nil, compaction_config: nil) # => Cadenya::Types::ObjectiveQueuedAction
 ```
 Continue an objective
 
 ```ruby
-client.objectives.continue(objective_id, message:, workspace_id: nil, enqueue: nil) # => Cadenya::Types::ObjectiveEvent
+client.objectives.continue(objective_id, message:, workspace_id: nil, enqueue: nil) # => Cadenya::Types::ContinueObjectiveResponse
+```
+Interrupt an objective
+
+```ruby
+client.objectives.interrupt(objective_id, workspace_id: nil) # => Cadenya::Types::ObjectiveEvent
+```
+Create an objective and stream its events
+
+```ruby
+client.objectives.create_and_stream(agent_id:, metadata:, workspace_id: nil, variation_id: nil, system_prompt_data: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil, last_event_id: nil) # => Cadenya::Stream
 ```
 
 ## client.tool_search

data/lib/cadenya/client.rb

@@ -8,7 +8,7 @@ module Cadenya
   # The Cadenya API client. Resource groups are readers; unset options are
   # read from the environment.
   class Client
-    attr_reader :accounts, :api_keys, :workspace_admin, :profiles, :workspaces, :agents, :ai_provider_keys, :models, :memory_layers, :objectives, :tool_search, :tenants, :tool_sets, :uploads, :widget_sessions, :widgets, :workspace_secrets
+    attr_reader :accounts, :api_keys, :workspace_admin, :profiles, :workspaces, :agent_pools, :agents, :ai_provider_keys, :models, :memory_layers, :objectives, :tool_search, :tenants, :tool_sets, :uploads, :widget_sessions, :widgets, :workspace_secrets
 
     def initialize(api_key: nil, base_url: nil, webhook_secret: nil, max_retries: 0, connection: nil, stream_transport: nil, workspace_id: nil)
       if !api_key.nil? && api_key.to_s.strip.empty?
@@ -37,7 +37,7 @@ module Cadenya
         auth_header: ["Authorization", "Bearer #{api_key}"],
         max_retries: max_retries,
         defaults: defaults,
-        user_agent: "cadenya-ruby/1.3.0 (api 1.0)",
+        user_agent: "cadenya-ruby/1.5.0 (api 1.0)",
         connection: connection,
         stream_transport: stream_transport
       )
@@ -46,6 +46,7 @@ module Cadenya
       @workspace_admin = Resources::WorkspaceAdmin.new(core)
       @profiles = Resources::Profiles.new(core)
       @workspaces = Resources::Workspaces.new(core)
+      @agent_pools = Resources::AgentPools.new(core)
       @agents = Resources::Agents.new(core)
       @ai_provider_keys = Resources::AiProviderKeys.new(core)
       @models = Resources::Models.new(core)

data/lib/cadenya/resources/agent_pools.rb

@@ -0,0 +1,108 @@
+# frozen_string_literal: true
+
+# Code generated by redwood. DO NOT EDIT.
+
+module Cadenya
+  module Resources
+    class AgentPools
+      def initialize(core)
+        @core = core
+      end
+
+      # List agent pools
+      def list(workspace_id: nil, limit: nil, cursor: nil, prefix: nil, query: nil, state: nil, labels: nil, sort_order: nil, include_info: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools"
+        _query = {
+          "limit" => limit,
+          "cursor" => cursor,
+          "prefix" => prefix,
+          "query" => query,
+          "state" => state,
+          "labels" => labels,
+          "sortOrder" => sort_order,
+          "includeInfo" => include_info,
+        }
+        _data = @core.request(:get, _path, query: _query, request_options: request_options) || {}
+        _items = (_data["items"] || []).map { |item| Types::AgentPool.from_json(item) }
+        _next_cursor = (((_data)["pagination"] || {}))["nextCursor"].to_s
+        Page.new(_items, _next_cursor) do |_c|
+          list(workspace_id: workspace_id, limit: limit, cursor: _c, prefix: prefix, query: query, state: state, labels: labels, sort_order: sort_order, include_info: include_info, request_options: request_options)
+        end
+      end
+
+      # Create a new agent pool
+      def create(metadata:, spec:, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools"
+        _body = {
+          "metadata" => metadata.nil? ? nil : (->(_v) { Types.encode_CreateResourceMetadata(_v) }).call(metadata),
+          "spec" => spec.nil? ? nil : (->(_v) { Types.encode_AgentPoolSpec(_v) }).call(spec),
+        }.reject { |_k, v| v.nil? }
+        _data = @core.request(:post, _path, body: _body, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Get an agent pool by ID
+      def retrieve(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}"
+        _data = @core.request(:get, _path, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Delete an agent pool
+      def delete(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}"
+        @core.request(:delete, _path, expects_body: false, request_options: request_options)
+        nil
+      end
+
+      # Update an agent pool
+      def update(id, workspace_id: nil, metadata: nil, spec: nil, update_mask: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}"
+        _body = {
+          "metadata" => metadata.nil? ? nil : (->(_v) { Types.encode_UpdateResourceMetadata(_v) }).call(metadata),
+          "spec" => spec.nil? ? nil : (->(_v) { Types.encode_AgentPoolUpdateSpec(_v) }).call(spec),
+          "updateMask" => update_mask,
+        }.reject { |_k, v| v.nil? }
+        _data = @core.request(:patch, _path, body: _body, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Activate an agent pool
+      def activate(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}:activate"
+        _data = @core.request(:post, _path, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Archive an agent pool
+      def archive(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}:archive"
+        _data = @core.request(:post, _path, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Deactivate an agent pool
+      def deactivate(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}:deactivate"
+        _data = @core.request(:post, _path, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+
+      # Unarchive an agent pool
+      def unarchive(id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/agent_pools/#{Util.path_param('id', id)}:unarchive"
+        _data = @core.request(:post, _path, request_options: request_options)
+        Types::AgentPool.from_json(_data)
+      end
+    end
+  end
+end

data/lib/cadenya/resources/objectives.rb

@@ -38,7 +38,7 @@ module Cadenya
       end
 
       # Create a new objective
-      def create(agent_id:, system_prompt_data:, workspace_id: nil, variation_id: nil, metadata: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil, request_options: nil)
+      def create(agent_id:, workspace_id: nil, variation_id: nil, metadata: nil, system_prompt_data: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil, request_options: nil)
         workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
         _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/objectives"
         _body = {
@@ -155,6 +155,31 @@ module Cadenya
         Types::ObjectiveFeedback.from_json(_data)
       end
 
+      # List objective queued actions
+      def list_queued_actions(objective_id, workspace_id: nil, limit: nil, cursor: nil, state: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/objectives/#{Util.path_param('objectiveId', objective_id)}/queued_actions"
+        _query = {
+          "limit" => limit,
+          "cursor" => cursor,
+          "state" => state,
+        }
+        _data = @core.request(:get, _path, query: _query, request_options: request_options) || {}
+        _items = (_data["items"] || []).map { |item| Types::ObjectiveQueuedAction.from_json(item) }
+        _next_cursor = (((_data)["pagination"] || {}))["nextCursor"].to_s
+        Page.new(_items, _next_cursor) do |_c|
+          list_queued_actions(objective_id, workspace_id: workspace_id, limit: limit, cursor: _c, state: state, request_options: request_options)
+        end
+      end
+
+      # Remove a queued action
+      def remove_queued_action(objective_id, queued_action_id:, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/objectives/#{Util.path_param('objectiveId', objective_id)}/queued_actions/#{Util.path_param('queuedActionId', queued_action_id)}:remove"
+        _data = @core.request(:post, _path, request_options: request_options)
+        Types::ObjectiveQueuedAction.from_json(_data)
+      end
+
       # List objective tool calls
       def list_tool_calls(objective_id, workspace_id: nil, limit: nil, cursor: nil, status: nil, include_info: nil, execution_status: nil, labels: nil, request_options: nil)
         workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
@@ -248,7 +273,7 @@ module Cadenya
           "compactionConfig" => compaction_config.nil? ? nil : (->(_v) { Types.encode_AgentVariationSpec_CompactionConfig(_v) }).call(compaction_config),
         }.reject { |_k, v| v.nil? }
         _data = @core.request(:post, _path, body: _body, request_options: request_options)
-        Types::CompactObjectiveResponse.from_json(_data)
+        Types::ObjectiveQueuedAction.from_json(_data)
       end
 
       # Continue an objective
@@ -260,8 +285,42 @@ module Cadenya
           "enqueue" => enqueue,
         }.reject { |_k, v| v.nil? }
         _data = @core.request(:post, _path, body: _body, request_options: request_options)
+        Types.decode_ContinueObjectiveResponse(_data)
+      end
+
+      # Interrupt an objective
+      def interrupt(objective_id, workspace_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/objectives/#{Util.path_param('objectiveId', objective_id)}:interrupt"
+        _data = @core.request(:post, _path, request_options: request_options)
         Types::ObjectiveEvent.from_json(_data)
       end
+
+      # Create an objective and stream its events
+      def create_and_stream(agent_id:, metadata:, workspace_id: nil, variation_id: nil, system_prompt_data: nil, first_user_message: nil, secrets: nil, memory_cascade: nil, first_user_message_data: nil, episodic_memory: nil, tenant: nil, subject: nil, pinned_parameters: nil, last_event_id: nil, request_options: nil)
+        workspace_id = @core.resolve_default("workspaceId", "CADENYA_WORKSPACE_ID", workspace_id)
+        _path = "/v1/workspaces/#{Util.path_param('workspaceId', workspace_id)}/objectives:stream"
+        _body = {
+          "agentId" => agent_id,
+          "variationId" => variation_id,
+          "metadata" => metadata.nil? ? nil : (->(_v) { Types.encode_CreateAndStreamObjectiveRequest_Metadata(_v) }).call(metadata),
+          "systemPromptData" => system_prompt_data,
+          "firstUserMessage" => first_user_message,
+          "secrets" => secrets.nil? ? nil : (->(_v) { _v.is_a?(Array) ? _v.map { |_i| (->(_v) { Types.encode_CreateObjectiveRequest_Secret(_v) }).call(_i) } : _v }).call(secrets),
+          "memoryCascade" => memory_cascade.nil? ? nil : (->(_v) { _v.is_a?(Array) ? _v.map { |_i| (->(_v) { Types.encode_MemoryReference(_v) }).call(_i) } : _v }).call(memory_cascade),
+          "firstUserMessageData" => first_user_message_data,
+          "episodicMemory" => episodic_memory.nil? ? nil : (->(_v) { Types.encode_ObjectiveEpisodicConfig(_v) }).call(episodic_memory),
+          "tenant" => tenant.nil? ? nil : (->(_v) { Types.encode_TenantAssertion(_v) }).call(tenant),
+          "subject" => subject.nil? ? nil : (->(_v) { Types.encode_SubjectAssertion(_v) }).call(subject),
+          "pinnedParameters" => pinned_parameters,
+        }.reject { |_k, v| v.nil? }
+        _headers = last_event_id ? { "Last-Event-ID" => last_event_id } : nil
+        _decoder = ->(event) { Types.decode_CreateAndStreamObjectiveResponse(event) }
+        _auto_reconnect = !(request_options && request_options[:reconnect] == false)
+        Stream.new(_decoder, last_event_id: last_event_id, skip_events: ["ping", "open"], auto_reconnect: _auto_reconnect) do |_cancel, _resume, &on_chunk|
+          @core.stream_request(:post, _path, body: _body, headers: (_resume ? { "Last-Event-ID" => _resume } : nil), request_options: request_options, cancel: _cancel, &on_chunk)
+        end
+      end
     end
   end
 end

data/lib/cadenya/resources/profiles.rb

@@ -13,7 +13,7 @@ module Cadenya
       def whoami(request_options: nil)
         _path = "/v1/whoami"
         _data = @core.request(:get, _path, request_options: request_options)
-        Types::Profile.from_json(_data)
+        Types::WhoamiResponse.from_json(_data)
       end
     end
   end