Skip to content

Commit 6160e3f

Browse files
committed
Add initial RubyLLM and Provider Rerank implementation
1 parent 2f73ae6 commit 6160e3f

File tree

2 files changed

+10
-0
lines changed

2 files changed

+10
-0
lines changed

lib/ruby_llm.rb

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,10 @@ def embed(...)
4646
Embedding.embed(...)
4747
end
4848

49+
def rerank(...)
50+
Rerank.rank(...)
51+
end
52+
4953
def paint(...)
5054
Image.paint(...)
5155
end

lib/ruby_llm/provider.rb

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,12 @@ def paint(prompt, model:, size:, connection:)
4343
parse_image_response(response, model:)
4444
end
4545

46+
def rank(query, documents, model:, connection:, top_n:, max_tokens_per_doc:) # rubocop:disable Metrics/ParameterLists
47+
payload = render_rerank_payload(query, documents, model:, top_n:, max_tokens_per_doc:)
48+
response = connection.post(rerank_url(model:), payload)
49+
parse_rerank_response(response, model:)
50+
end
51+
4652
def configured?(config = nil)
4753
config ||= RubyLLM.config
4854
missing_configs(config).empty?

0 commit comments

Comments
 (0)