eed2eaceb0
Co-authored-by: aakashb95 <aakash@chatwoot.com> Co-authored-by: Shivam Mishra <scm.mymail@gmail.com>
40 lines
1.4 KiB
Ruby
40 lines
1.4 KiB
Ruby
class Captain::Tools::Copilot::SearchArticlesService < Captain::Tools::BaseTool
|
|
def self.name
|
|
'search_articles'
|
|
end
|
|
description 'Search articles based on parameters'
|
|
params do
|
|
string :query, description: 'Search articles by title or content (partial match)'
|
|
number :category_id, description: 'Filter articles by category ID'
|
|
any_of :status, description: 'Filter articles by status' do
|
|
string enum: %w[draft published archived]
|
|
end
|
|
end
|
|
|
|
def execute(query: nil, category_id: nil, status: nil)
|
|
articles = fetch_articles(query: query, category_id: category_id, status: status)
|
|
return 'No articles found' unless articles.exists?
|
|
|
|
total_count = articles.count
|
|
articles = articles.limit(100)
|
|
<<~RESPONSE
|
|
#{total_count > 100 ? "Found #{total_count} articles (showing first 100)" : "Total number of articles: #{total_count}"}
|
|
#{articles.map(&:to_llm_text).join("\n---\n")}
|
|
RESPONSE
|
|
end
|
|
|
|
def active?
|
|
user_has_permission('knowledge_base_manage')
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_articles(query:, category_id:, status:)
|
|
articles = Article.where(account_id: @assistant.account_id)
|
|
articles = articles.where('title ILIKE :query OR content ILIKE :query', query: "%#{query}%") if query.present?
|
|
articles = articles.where(category_id: category_id) if category_id.present?
|
|
articles = articles.where(status: status) if status.present?
|
|
articles
|
|
end
|
|
end
|