Method PromptObject

api-prompt-object.go:33–78

PromptObject performs language model inference with the prompt and referenced object as context. Inference is performed using a Lambda handler that can process the prompt and object. Currently, this functionality is limited to certain MinIO servers.

(ctx context.Context, bucketName, objectName, prompt string, opts PromptObjectOptions)

// Inference is performed using a Lambda handler that can process the prompt and object.
// Currently, this functionality is limited to certain MinIO servers.
func (c *Client) PromptObject(ctx context.Context, bucketName, objectName, prompt string, opts PromptObjectOptions) (io.ReadCloser, error) {
	// Input validation.
	if err := s3utils.CheckValidBucketName(bucketName); err != nil {
		return nil, ErrorResponse{
			StatusCode: http.StatusBadRequest,
			Code:       InvalidBucketName,
			Message:    err.Error(),
		}
	}
	if err := s3utils.CheckValidObjectName(objectName); err != nil {
		return nil, ErrorResponse{
			StatusCode: http.StatusBadRequest,
			Code:       XMinioInvalidObjectName,
			Message:    err.Error(),
		}
	}

	opts.AddLambdaArnToReqParams(opts.LambdaArn)
	opts.SetHeader("Content-Type", "application/json")
	opts.AddPromptArg("prompt", prompt)
	promptReqBytes, err := json.Marshal(opts.PromptArgs)
	if err != nil {
		return nil, err
	}

	// Execute POST on bucket/object.
	resp, err := c.executeMethod(ctx, http.MethodPost, requestMetadata{
		bucketName:       bucketName,
		objectName:       objectName,
		queryValues:      opts.toQueryValues(),
		customHeader:     opts.Header(),
		contentSHA256Hex: sum256Hex(promptReqBytes),
		contentBody:      bytes.NewReader(promptReqBytes),
		contentLength:    int64(len(promptReqBytes)),
	})
	if err != nil {
		return nil, err
	}

	if resp.StatusCode != http.StatusOK {
		defer closeResponse(resp)
		return nil, httpRespToErrorResponse(resp, bucketName, objectName)
	}

	return resp.Body, nil
}