Skip to content

Commit 5f3c943

Browse files
authored
Merge pull request chaitin#172 from yokowu/chore-model-provider
chore: 百智云模型列表参数修正
2 parents e126f68 + 2a8966c commit 5f3c943

1 file changed

Lines changed: 5 additions & 1 deletion

File tree

backend/internal/model/usecase/model.go

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -332,14 +332,18 @@ func (m *ModelUsecase) GetProviderModelList(ctx context.Context, req *domain.Get
332332
if err != nil {
333333
return nil, err
334334
}
335+
st := string(req.Type)
336+
if req.Type == consts.ModelTypeLLM {
337+
st = "chat"
338+
}
335339
client := request.NewClient(u.Scheme, u.Host, m.client.Timeout, request.WithClient(m.client))
336340
resp, err := request.Get[domain.OpenAIResp](client, "/v1/models", request.WithHeader(
337341
request.Header{
338342
"Authorization": fmt.Sprintf("Bearer %s", req.APIKey),
339343
},
340344
), request.WithQuery(request.Query{
341345
"type": "text",
342-
"sub_type": "chat",
346+
"sub_type": st,
343347
}))
344348
if err != nil {
345349
return nil, err

0 commit comments

Comments
 (0)