119 lines
3.5 KiB
Ruby
Executable File
119 lines
3.5 KiB
Ruby
Executable File
class Api::V1::Accounts::Captain::AssistantsController < Api::V1::Accounts::BaseController
|
|
before_action :current_account
|
|
before_action -> { check_authorization(Captain::Assistant) }
|
|
|
|
before_action :set_assistant, only: [:show, :update, :destroy, :playground]
|
|
|
|
def index
|
|
@assistants = account_assistants.ordered
|
|
end
|
|
|
|
def show; end
|
|
|
|
def create
|
|
@assistant = account_assistants.create!(assistant_params)
|
|
end
|
|
|
|
def update
|
|
payload = assistant_params
|
|
overrides = system_prompt_action_overrides
|
|
if overrides.present?
|
|
payload[:config] ||= {}
|
|
payload[:config].merge!(overrides)
|
|
end
|
|
@assistant.update!(payload)
|
|
end
|
|
|
|
def destroy
|
|
@assistant.destroy
|
|
head :no_content
|
|
end
|
|
|
|
def playground
|
|
response = Captain::Llm::AssistantChatService.new(assistant: @assistant).generate_response(
|
|
additional_message: params[:message_content],
|
|
message_history: message_history
|
|
)
|
|
|
|
render json: response
|
|
end
|
|
|
|
def tools
|
|
assistant = Captain::Assistant.new(account: Current.account)
|
|
@tools = assistant.available_agent_tools
|
|
end
|
|
|
|
private
|
|
|
|
def set_assistant
|
|
@assistant = account_assistants.find(params[:id])
|
|
end
|
|
|
|
def account_assistants
|
|
@account_assistants ||= Captain::Assistant.for_account(Current.account.id)
|
|
end
|
|
|
|
def assistant_payload
|
|
params[:assistant].presence || params
|
|
end
|
|
|
|
def assistant_params
|
|
permitted = assistant_payload.permit(:name, :description, :llm_provider, :llm_model, :api_key,
|
|
config: [
|
|
:product_name, :role_name, :feature_faq, :feature_memory, :feature_citation,
|
|
:welcome_message, :handoff_message, :resolution_message,
|
|
:instructions, :temperature, :playbook, :distance_threshold, :max_rag_results,
|
|
:system_prompt,
|
|
{ system_prompt_blocks: [:key, :title, :content, :order] }
|
|
])
|
|
|
|
# Handle array parameters separately to allow partial updates
|
|
permitted[:response_guidelines] = assistant_payload[:response_guidelines] if assistant_payload.key?(:response_guidelines)
|
|
|
|
permitted[:guardrails] = assistant_payload[:guardrails] if assistant_payload.key?(:guardrails)
|
|
|
|
permitted
|
|
end
|
|
|
|
def system_prompt_action_overrides
|
|
action = assistant_payload[:system_prompt_action].to_s
|
|
return {} if action.blank?
|
|
|
|
config = @assistant.config || {}
|
|
versions = Array(config['system_prompt_versions'])
|
|
blocks = assistant_payload.dig(:config, :system_prompt_blocks)
|
|
|
|
case action
|
|
when 'save_version'
|
|
return {} if blocks.blank?
|
|
|
|
versions << {
|
|
'blocks' => blocks,
|
|
'saved_at' => Time.zone.now.to_i,
|
|
'saved_by_id' => Current.user&.id
|
|
}
|
|
{ 'system_prompt_versions' => versions.last(10) }
|
|
when 'revert_last'
|
|
last = versions.pop
|
|
return {} if last.blank?
|
|
|
|
{
|
|
'system_prompt_blocks' => last['blocks'],
|
|
'system_prompt_versions' => versions.last(10)
|
|
}
|
|
when 'restore_default'
|
|
{ 'system_prompt' => nil, 'system_prompt_blocks' => nil }
|
|
else
|
|
{}
|
|
end
|
|
end
|
|
|
|
def playground_params
|
|
params.require(:assistant).permit(:message_content, message_history: [:role, :content])
|
|
end
|
|
|
|
def message_history
|
|
(playground_params[:message_history] || []).map { |message| { role: message[:role], content: message[:content] } }
|
|
end
|
|
end
|