mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-10-29 18:02:25 -05:00
8b0ea56027
- github.com/KimMachineGun/automemlimit v0.7.2 => v0.7.3
- github.com/gin-contrib/cors v1.7.5 => v1.7.6
- github.com/minio/minio-go/v7 v7.0.92 => v7.0.94
- github.com/spf13/cast v1.8.0 => v1.9.2
- github.com/uptrace/bun{,/*} v1.2.11 => v1.2.14
- golang.org/x/image v0.27.0 => v0.28.0
- golang.org/x/net v0.40.0 => v0.41.0
- code.superseriousbusiness.org/go-swagger v0.31.0-gts-go1.23-fix => v0.32.3-gts-go1.23-fix
Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4304
Co-authored-by: kim <grufwub@gmail.com>
Co-committed-by: kim <grufwub@gmail.com>
78 lines
2.4 KiB
Go
78 lines
2.4 KiB
Go
/*
|
|
* MinIO Go Library for Amazon S3 Compatible Cloud Storage
|
|
* Copyright 2015-2024 MinIO, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package minio
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
|
|
"github.com/minio/minio-go/v7/internal/json"
|
|
"github.com/minio/minio-go/v7/pkg/s3utils"
|
|
)
|
|
|
|
// 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.
|
|
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
|
|
}
|