feat: Use embeddings in help center search (#9227)

This commit is contained in:
Sojan Jose
2024-04-15 16:35:23 -07:00
committed by GitHub
parent ade658ad86
commit 42a457ff5d
12 changed files with 190 additions and 12 deletions

View File

@@ -1,11 +1,11 @@
class Openai::EmbeddingsService
def get_embedding(content)
fetch_embeddings(content)
def get_embedding(content, model = 'text-embedding-ada-002')
fetch_embeddings(content, model)
end
private
def fetch_embeddings(input)
def fetch_embeddings(input, model)
url = 'https://api.openai.com/v1/embeddings'
headers = {
'Authorization' => "Bearer #{ENV.fetch('OPENAI_API_KEY', '')}",
@@ -13,7 +13,7 @@ class Openai::EmbeddingsService
}
data = {
input: input,
model: 'text-embedding-ada-002'
model: model
}
response = Net::HTTP.post(URI(url), data.to_json, headers)