mirror of
https://github.com/discourse/discourse.git
synced 2026-08-08 17:53:55 +08:00
165e13af62Previously to triage with an ai tool you would need to first call a persona this could be inefficient since the persona would issue a call to the llm just to trigger the tool This new automation bypasses that also - Adds the ability to set custom fields in the API - Live refresh UI when setting tags
69 lines
2.2 KiB
Ruby
Vendored
69 lines
2.2 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Automation
|
|
def self.spam_based_flag_types
|
|
%w[spam spam_silence]
|
|
end
|
|
|
|
def self.flag_types
|
|
[
|
|
{ id: "review", translated_name: I18n.t("discourse_automation.ai.flag_types.review") },
|
|
{
|
|
id: "review_hide",
|
|
translated_name: I18n.t("discourse_automation.ai.flag_types.review_hide"),
|
|
},
|
|
{
|
|
id: "review_delete",
|
|
translated_name: I18n.t("discourse_automation.ai.flag_types.review_delete"),
|
|
},
|
|
{
|
|
id: "review_delete_silence",
|
|
translated_name: I18n.t("discourse_automation.ai.flag_types.review_delete_silence"),
|
|
},
|
|
{ id: "spam", translated_name: I18n.t("discourse_automation.ai.flag_types.spam") },
|
|
{
|
|
id: "spam_silence",
|
|
translated_name: I18n.t("discourse_automation.ai.flag_types.spam_silence"),
|
|
},
|
|
]
|
|
end
|
|
|
|
def self.available_custom_tools
|
|
available_tools(
|
|
scope: AiTool.where(enabled: true).where("parameters = '[]'::jsonb"),
|
|
description_field: :description,
|
|
)
|
|
end
|
|
|
|
def self.available_tools_all
|
|
available_tools(description_field: :summary)
|
|
end
|
|
|
|
def self.available_tools(scope: AiTool.where(enabled: true), description_field: :description)
|
|
scope
|
|
.pluck(:id, :name, description_field)
|
|
.map { |id, name, desc| { id: id, translated_name: name, description: desc } }
|
|
end
|
|
|
|
def self.available_models
|
|
DB.query_hash(<<~SQL).each { |value_h| value_h["id"] = "#{value_h["id"]}" }
|
|
SELECT display_name AS translated_name, id AS id
|
|
FROM llm_models
|
|
SQL
|
|
end
|
|
|
|
def self.available_persona_choices(require_user: true, require_default_llm: true)
|
|
relation = AiPersona.includes(:user)
|
|
relation = relation.where.not(user_id: nil) if require_user
|
|
relation = relation.where.not(default_llm: nil) if require_default_llm
|
|
relation.map do |persona|
|
|
phash = { id: persona.id, translated_name: persona.name, description: persona.name }
|
|
|
|
phash[:description] += " (#{persona&.user&.username})" if require_user
|
|
|
|
phash
|
|
end
|
|
end
|
|
end
|
|
end
|
