Module: Collavre::UsersController::AiUserManagement

Extended by:
ActiveSupport::Concern
Included in:
Collavre::UsersController
Defined in:
app/controllers/concerns/collavre/users_controller/ai_user_management.rb

Instance Method Summary collapse

Instance Method Details

#create_aiObject



23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
# File 'app/controllers/concerns/collavre/users_controller/ai_user_management.rb', line 23

def create_ai
  ai_id = params[:ai_id].to_s.strip.downcase
  email = "#{ai_id}@ai.local"
  searchable = ActiveModel::Type::Boolean.new.cast(params.fetch(:searchable, false))

  @user = Collavre::User.new(
    name: params[:name],
    email: email,
    password: SecureRandom.hex(36),
    system_prompt: params[:system_prompt],
    llm_vendor: params[:llm_vendor].presence || "google",
    llm_model: params[:llm_model],
    llm_api_key: params[:llm_api_key],
    gateway_url: params[:gateway_url],
    tools: params[:tools] || [],
    searchable: searchable,
    email_verified_at: Time.current,
    created_by_id: Current.user.id,
    routing_expression: params[:routing_expression]
  )
  @user.agent_conf = params[:agent_conf] if @user.respond_to?(:agent_conf=) && params[:agent_conf].present?

  if @user.save
    Collavre::Contact.ensure(user: Current.user, contact_user: @user)
    share_ai_agent_to_creative(@user, params[:creative_id])
    redirect_to user_path(Current.user, tab: "contacts"), notice: I18n.t("collavre.users.create_ai.success")
  else
    flash.now[:alert] = @user.errors.full_messages.to_sentence
    @available_tools = load_available_tools
    render :new_ai, status: :unprocessable_entity
  end
end

#edit_aiObject



56
57
58
# File 'app/controllers/concerns/collavre/users_controller/ai_user_management.rb', line 56

def edit_ai
  @available_tools = load_available_tools
end

#new_aiObject



11
12
13
14
15
16
17
18
19
20
21
# File 'app/controllers/concerns/collavre/users_controller/ai_user_management.rb', line 11

def new_ai
  @available_tools = load_available_tools

  if params[:copy_from].present?
    source = Collavre::User.find_by(id: params[:copy_from])
    if source&.ai_user? && source.created_by_id == Current.user.id
      @copy_source = source
      @copy_name = "#{source.name} (copy)"
    end
  end
end

#update_aiObject



60
61
62
63
64
65
66
67
68
69
70
# File 'app/controllers/concerns/collavre/users_controller/ai_user_management.rb', line 60

def update_ai
  ai_params = params.require(:user).permit(:name, :system_prompt, :llm_vendor, :llm_model, :llm_api_key, :gateway_url, :searchable, :routing_expression, :agent_conf, tools: [])

  if @user.update(ai_params)
    redirect_to edit_ai_user_path(@user), notice: I18n.t("collavre.users.update_ai.success")
  else
    @available_tools = load_available_tools
    flash.now[:alert] = @user.errors.full_messages.to_sentence
    render :edit_ai, status: :unprocessable_entity
  end
end