6f19546c3c
ref: https://linear.app/chatwoot/issue/CW-2464/bot-should-also-return-links-to-the-information
23 lines
531 B
Ruby
23 lines
531 B
Ruby
class Openai::EmbeddingsService
|
|
def get_embedding(content)
|
|
fetch_embeddings(content)
|
|
end
|
|
|
|
private
|
|
|
|
def fetch_embeddings(input)
|
|
url = 'https://api.openai.com/v1/embeddings'
|
|
headers = {
|
|
'Authorization' => "Bearer #{ENV.fetch('OPENAI_API_KEY', '')}",
|
|
'Content-Type' => 'application/json'
|
|
}
|
|
data = {
|
|
input: input,
|
|
model: 'text-embedding-ada-002'
|
|
}
|
|
|
|
response = Net::HTTP.post(URI(url), data.to_json, headers)
|
|
JSON.parse(response.body)['data']&.pick('embedding')
|
|
end
|
|
end
|