12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667 |
- package chat
- import (
- "context"
- "errors"
- "wechat-api/ent"
- "wechat-api/internal/svc"
- "wechat-api/internal/types"
- "wechat-api/internal/utils/compapi"
- "wechat-api/internal/utils/contextkey"
- "github.com/zeromicro/go-zero/core/logx"
- )
- type ChatCompletionsLogic struct {
- logx.Logger
- ctx context.Context
- svcCtx *svc.ServiceContext
- }
- func NewChatCompletionsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ChatCompletionsLogic {
- return &ChatCompletionsLogic{
- Logger: logx.WithContext(ctx),
- ctx: ctx,
- svcCtx: svcCtx}
- }
- func (l *ChatCompletionsLogic) ChatCompletions(req *types.CompApiReq) (resp *types.CompOpenApiResp, err error) {
-
-
- var (
- apiKeyObj *ent.ApiKey
- ok bool
- )
- workToken := compapi.GetWorkTokenByID(req.EventType, req.WorkId)
- apiKeyObj, ok = contextkey.AuthTokenInfoKey.GetValue(l.ctx)
- if !ok {
- return nil, errors.New("content get token err")
- }
-
- if len(apiKeyObj.OpenaiBase) == 0 || len(workToken) == 0 {
- return nil, errors.New("not auth info")
- }
- return l.workForFastgpt(req, workToken, apiKeyObj.OpenaiBase)
- }
- func (l *ChatCompletionsLogic) workForFastgpt(req *types.CompApiReq, apiKey string, apiBase string) (resp *types.CompOpenApiResp, err error) {
-
- return compapi.NewFastgptChatCompletions(l.ctx, apiKey, apiBase, req)
- }
|