mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-10-31 14:38:37 +00:00
b0310f90d3
Note, we perform permission checks on tag list against anon to ensure we do not disclose information about private tags to the llm which could get extracted.
78 lines
2.0 KiB
Ruby
78 lines
2.0 KiB
Ruby
#frozen_string_literal: true
|
|
|
|
module DiscourseAi::AiBot::Commands
|
|
class ReadCommand < Command
|
|
class << self
|
|
def name
|
|
"read"
|
|
end
|
|
|
|
def desc
|
|
"Will read a topic or a post on this Discourse instance"
|
|
end
|
|
|
|
def parameters
|
|
[
|
|
Parameter.new(
|
|
name: "topic_id",
|
|
description: "the id of the topic to read",
|
|
type: "integer",
|
|
required: true,
|
|
),
|
|
Parameter.new(
|
|
name: "post_number",
|
|
description: "the post number to read",
|
|
type: "integer",
|
|
required: false,
|
|
),
|
|
]
|
|
end
|
|
end
|
|
|
|
def description_args
|
|
{ title: @title, url: @url }
|
|
end
|
|
|
|
def process(topic_id:, post_number: nil)
|
|
not_found = { topic_id: topic_id, description: "Topic not found" }
|
|
|
|
@title = ""
|
|
|
|
topic_id = topic_id.to_i
|
|
|
|
topic = Topic.find_by(id: topic_id)
|
|
return not_found if !topic || !Guardian.new.can_see?(topic)
|
|
|
|
@title = topic.title
|
|
|
|
posts = Post.secured(Guardian.new).where(topic_id: topic_id).order(:post_number).limit(40)
|
|
@url = topic.relative_url(post_number)
|
|
|
|
posts = posts.where("post_number = ?", post_number) if post_number
|
|
|
|
content = +<<~TEXT.strip
|
|
title: #{topic.title}
|
|
TEXT
|
|
|
|
category_names = [topic.category&.parent_category&.name, topic.category&.name].compact.join(
|
|
" ",
|
|
)
|
|
content << "\ncategories: #{category_names}" if category_names.present?
|
|
|
|
if topic.tags.length > 0
|
|
tags = DiscourseTagging.filter_visible(topic.tags, Guardian.new)
|
|
content << "\ntags: #{tags.map(&:name).join(", ")}\n\n" if tags.length > 0
|
|
end
|
|
|
|
posts.each { |post| content << "\n\n#{post.username} said:\n\n#{post.raw}" }
|
|
|
|
# TODO: 16k or 100k models can handle a lot more tokens
|
|
content = tokenizer.truncate(content, 1500).squish
|
|
|
|
result = { topic_id: topic_id, content: content, complete: true }
|
|
result[:post_number] = post_number if post_number
|
|
result
|
|
end
|
|
end
|
|
end
|