mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2025-11-03 23:42:26 -06:00
b6ff55662e
Update dependencies:
- github.com/gin-gonic/gin v1.10.0 -> v1.10.1
- github.com/gin-contrib/sessions v1.10.3 -> v1.10.4
- github.com/jackc/pgx/v5 v5.7.4 -> v5.7.5
- github.com/minio/minio-go/v7 v7.0.91 -> v7.0.92
- github.com/pquerna/otp v1.4.0 -> v1.5.0
- github.com/tdewolff/minify/v2 v2.23.5 -> v2.23.8
- github.com/yuin/goldmark v1.7.11 -> v1.7.12
- go.opentelemetry.io/otel{,/*} v1.35.0 -> v1.36.0
- modernc.org/sqlite v1.37.0 -> v1.37.1
Reviewed-on: https://codeberg.org/superseriousbusiness/gotosocial/pulls/4188
Reviewed-by: Daenney <daenney@noreply.codeberg.org>
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
|
|
}
|