feat: Use embeddings in help center search (#9227)
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user