mirror of
https://gh.wpcy.net/https://github.com/discourse/discourse.git
synced 2026-06-19 07:43:46 +08:00
bcb0949a2f
Redesigns the Data Explorer `/queries/new` page to be AI-first, with a non-AI fallback if AI is disabled. Instead of the old dual-form layout (manual create + separate AI generate that redirects to the edit page), the new page lets you describe what you want, generates the SQL inline, and saves when you're ready. Reviewer notes: - there's no database record created until user clicks "Save query" - new POST /queries/generate endpoint for query generation, no DB record - agent rework - related PR https://github.com/discourse/discourse-ai-evals/pull/16 - switched to agentic execution mode - removed ValidateSql tool (redundant with RunSql) - added full Data Explorer parameter type reference - evaluator for the DE agent
105 lines
3.2 KiB
Ruby
Vendored
105 lines
3.2 KiB
Ruby
Vendored
# frozen_string_literal: true
|
|
|
|
module DiscourseAi
|
|
module Evals
|
|
module Runners
|
|
class Base
|
|
class << self
|
|
def can_handle?(_feature)
|
|
raise NotImplemented
|
|
end
|
|
|
|
def find_runner(feature, agent_prompt)
|
|
registry = [
|
|
DiscourseAi::Evals::Runners::AiHelper,
|
|
DiscourseAi::Evals::Runners::Translation,
|
|
DiscourseAi::Evals::Runners::Hyde,
|
|
DiscourseAi::Evals::Runners::Discoveries,
|
|
DiscourseAi::Evals::Runners::Spam,
|
|
DiscourseAi::Evals::Runners::Summarization,
|
|
DiscourseAi::Evals::Runners::Inference,
|
|
DiscourseAi::Evals::Runners::DataExplorer,
|
|
]
|
|
klass = registry.find { |runner| runner.can_handle?(feature) }
|
|
klass&.new(feature.split(":").last, agent_prompt) if klass
|
|
end
|
|
end
|
|
|
|
attr_reader :feature_name, :agent_prompt_override
|
|
|
|
def initialize(feature, agent_prompt_override = nil)
|
|
@feature_name = feature
|
|
@agent_prompt_override = agent_prompt_override
|
|
end
|
|
|
|
private
|
|
|
|
def resolve_agent(agent_class: nil)
|
|
if agent_class.nil?
|
|
raise ArgumentError, "Unable to resolve agent for runner (#{self.class.name})"
|
|
end
|
|
|
|
agent = agent_class.new
|
|
|
|
if agent_prompt_override.present?
|
|
override = agent_prompt_override
|
|
agent.define_singleton_method(:system_prompt) { override }
|
|
end
|
|
|
|
agent
|
|
end
|
|
|
|
def capture_plain_response(bot, context, execution_context:)
|
|
buffer = +""
|
|
bot.reply(context, execution_context:) do |partial, _, type|
|
|
buffer << partial if type.blank?
|
|
end
|
|
buffer
|
|
end
|
|
|
|
def capture_structured_response(
|
|
bot,
|
|
context,
|
|
schema_key:,
|
|
schema_type: "string",
|
|
execution_context:
|
|
)
|
|
key = schema_key&.to_sym
|
|
raise ArgumentError, "schema_key is required for structured capture" if key.nil?
|
|
|
|
accumulator = schema_type == "array" ? [] : +""
|
|
|
|
bot.reply(context, execution_context:) do |partial, _, type|
|
|
if type == :structured_output
|
|
chunk = partial.read_buffered_property(key)
|
|
accumulator = append_structured_chunk(accumulator, schema_type, chunk)
|
|
elsif type.blank?
|
|
accumulator = append_structured_chunk(accumulator, schema_type, partial)
|
|
end
|
|
end
|
|
|
|
accumulator
|
|
end
|
|
|
|
def append_structured_chunk(accumulator, schema_type, chunk)
|
|
return accumulator if chunk.nil? || (chunk.respond_to?(:empty?) && chunk.empty?)
|
|
|
|
case schema_type
|
|
when "array"
|
|
Array(chunk).each { |item| accumulator << item if accumulator.exclude?(item) }
|
|
accumulator
|
|
when "string"
|
|
accumulator << chunk.to_s
|
|
else
|
|
chunk
|
|
end
|
|
end
|
|
|
|
def wrap_result(raw, metadata = nil)
|
|
metadata = metadata&.compact
|
|
metadata.present? ? { raw:, metadata: metadata } : { raw: raw }
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
