Merge pull request redpanda-data#2732 from rockwotj/openai-fixes

Fix openai embeddings processor
anovateam · Jul 24, 2024 · d51b395 · d51b395
2 parents 4e3b047 + c9ecaef
commit d51b395
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/internal/impl/openai/embeddings_processor.go b/internal/impl/openai/embeddings_processor.go
@@ -78,6 +78,7 @@ type embeddingsProcessor struct {
 
 func (p *embeddingsProcessor) Process(ctx context.Context, msg *service.Message) (service.MessageBatch, error) {
 	var body oai.EmbeddingsOptions
+	body.DeploymentName = &p.model
 	if p.text != nil {
 		s, err := msg.BloblangQueryValue(p.text)
 		if err != nil {

diff --git a/internal/impl/openai/translation_processor.go b/internal/impl/openai/translation_processor.go
@@ -49,10 +49,10 @@ To learn more about translation, see the https://platform.openai.com/docs/guides
 			)...,
 		).
 		Fields(
-			service.NewBloblangField(otspFieldFile).
+			service.NewBloblangField(otlpFieldFile).
 				Description("The audio file object (not file name) to translate, in one of the following formats: `flac`, `mp3`, `mp4`, `mpeg`, `mpga`, `m4a`, `ogg`, `wav`, or `webm`.").
 				Optional(),
-			service.NewInterpolatedStringField(otspFieldPrompt).
+			service.NewInterpolatedStringField(otlpFieldPrompt).
 				Description("Optional text to guide the model's style or continue a previous audio segment. The prompt should match the audio language.").
 				Optional().
 				Advanced(),