discourse/discourse-ai
1
0
Fork 0
mirror of https://github.com/discourse/discourse-ai.git synced 2025-03-06 09:20:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

DEV: Mark hypothetical_post_from method public (#607)

Browse Source
This commit is contained in:
Rafael dos Santos Silva 2024-05-07 15:17:26 -03:00 committed by GitHub
parent 27827c0898
commit ab4544d897
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
lib/embeddings/semantic_search.rb
View File

@ -151,18 +151,6 @@ module DiscourseAi
guardian.filter_allowed_categories(reranked_semantic_results)
end
private
attr_reader :guardian
def build_hyde_key(digest, hyde_model)
"semantic-search-#{digest}-#{hyde_model}"
end
def build_embedding_key(digest, hyde_model, embedding_model)
"#{build_hyde_key(digest, hyde_model)}-#{embedding_model}"
end
def hypothetical_post_from(search_term)
prompt = DiscourseAi::Completions::Prompt.new(<<~TEXT.strip)
You are a content creator for a forum. The forum description is as follows:
@ -185,6 +173,18 @@ module DiscourseAi
Nokogiri::HTML5.fragment(llm_response).at("ai")&.text&.presence || llm_response
end
private
attr_reader :guardian
def build_hyde_key(digest, hyde_model)
"semantic-search-#{digest}-#{hyde_model}"
end
def build_embedding_key(digest, hyde_model, embedding_model)
"#{build_hyde_key(digest, hyde_model)}-#{embedding_model}"
end
end
end
end