Skip to content

Commit 61ae37d

Browse files
committed
fix: ai balance bug
1 parent b55d68e commit 61ae37d

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

module/ai-balance/iml.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -106,10 +106,11 @@ func (i *imlBalanceModule) Create(ctx context.Context, input *ai_balance_dto.Cre
106106
func newRelease(item *ai_balance.Balance, base string) *gateway.DynamicRelease {
107107

108108
cfg := make(map[string]interface{})
109-
cfg["provider"] = item.Id
109+
cfg["provider"] = item.Provider
110110
cfg["model"] = item.Model
111111
cfg["model_config"] = ai_provider_local.OllamaConfig
112112
cfg["base"] = base
113+
cfg["priority"] = item.Priority
113114
return &gateway.DynamicRelease{
114115
BasicItem: &gateway.BasicItem{
115116
ID: item.Id,

0 commit comments

Comments
 (0)