admin/MonkeyCode
1
0
Fork 0
mirror of https://github.com/chaitin/MonkeyCode.git synced 2026-02-12 03:33:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(model): 增加缓存机制

Browse Source
This commit is contained in:
yokowu
2025-07-13 00:05:39 +08:00
parent d6574ad44c
commit 963ad4eabc
7 changed files with 40 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
backend/internal/proxy/usecase/proxy.go
View File

@@ -10,11 +10,12 @@ import (
)
type ProxyUsecase struct {
repo domain.ProxyRepo
repo domain.ProxyRepo
modelRepo domain.ModelRepo
}
func NewProxyUsecase(repo domain.ProxyRepo) domain.ProxyUsecase {
return &ProxyUsecase{repo: repo}
func NewProxyUsecase(repo domain.ProxyRepo, modelRepo domain.ModelRepo) domain.ProxyUsecase {
return &ProxyUsecase{repo: repo, modelRepo: modelRepo}
}
func (p *ProxyUsecase) Record(ctx context.Context, record *domain.RecordParam) error {
@@ -23,7 +24,7 @@ func (p *ProxyUsecase) Record(ctx context.Context, record *domain.RecordParam) e
// SelectModelWithLoadBalancing implements domain.ProxyUsecase.
func (p *ProxyUsecase) SelectModelWithLoadBalancing(modelName string, modelType consts.ModelType) (*domain.Model, error) {
model, err := p.repo.SelectModelWithLoadBalancing(modelName, modelType)
model, err := p.modelRepo.GetWithCache(context.Background(), modelType)
if err != nil {
return nil, err
}