Update lambda.yaml

aws-samples · Oct 25, 2024 · 302fb83 · 302fb83
1 parent 875db13
commit 302fb83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lambda-udfs/f_titan_embedding(varchar)/lambda.yaml b/lambda-udfs/f_titan_embedding(varchar)/lambda.yaml
@@ -54,7 +54,7 @@ Resources:
 
           def generate_embeddings(text=None):
               try:
-                  input_data = {"inputText": text}
+                  input_data = {"inputText": text, "dimensions": 256}
                   response = bedrock_runtime.invoke_model(
                       body=json.dumps(input_data),
                       modelId="amazon.titan-embed-text-v1",
@@ -90,4 +90,4 @@ Resources:
                   print('Error: '+ str(e))
                   exc_type, exc_obj, exc_tb = sys.exc_info()
                   print(exc_type, exc_tb.tb_lineno)
-              return json.dumps(redshift_response)
+              return json.dumps(redshift_response)