Skip to content

Commit 57f72db

Browse files
make sure embedding fits (#240)
* make sure embedding fits * fix test fail
1 parent 24ee72f commit 57f72db

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

app/controllers/concerns/gpt_concern.rb

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,12 @@ def call_salesforce_connect_gpt_embedding(input)
8383
access_token = oauth_token['access_token']
8484
instance_url = oauth_token['instance_url']
8585

86+
# 15000 characters is about equivalent to 8192 tokens for the text-embedding-ada-002 model.
87+
truncated_input = input.truncate(15000.to_i)
88+
8689
new_endpoint_url = "#{instance_url}/services/data/v58.0/einstein/llm/embeddings"
8790
request_body = {
88-
prompts: { wrappedListString: [input] },
91+
prompts: { wrappedListString: [truncated_input] },
8992
additionalConfig: {
9093
applicationName: 'fack',
9194
model: ENV.fetch('EMBEDDING_MODEL','llmgateway__AzureOpenAITextEmbeddingAda_002')

0 commit comments

Comments
 (0)