mirror of
https://gitcode.com/JianFeeeee/ModelRouter.git
synced 2026-10-03 23:54:06 +00:00
feat(auto): AUTO-only priority chain with tiered slots + live source probing; CSV export w/ key names; audit persistence; fix prompt token accounting & deepseek thinking
This commit is contained in:
@ -49,14 +49,16 @@ func (s *Store) Upsert(src Source) error {
|
||||
for i := range s.data.Sources {
|
||||
if s.data.Sources[i].Name == src.Name {
|
||||
s.data.Sources[i] = src
|
||||
s.data.DeletedSources = removeString(s.data.DeletedSources, src.Name)
|
||||
return s.persistLocked()
|
||||
}
|
||||
}
|
||||
s.data.Sources = append(s.data.Sources, src)
|
||||
s.data.DeletedSources = removeString(s.data.DeletedSources, src.Name)
|
||||
return s.persistLocked()
|
||||
}
|
||||
|
||||
// Remove deletes a runtime source and persists.
|
||||
// Remove deletes a runtime source or hides a base YAML source and persists.
|
||||
func (s *Store) Remove(name string) (bool, error) {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
@ -69,11 +71,47 @@ func (s *Store) Remove(name string) (bool, error) {
|
||||
}
|
||||
kept = append(kept, src)
|
||||
}
|
||||
if !removed {
|
||||
return false, nil
|
||||
}
|
||||
s.data.Sources = kept
|
||||
return true, s.persistLocked()
|
||||
if !containsString(s.data.DeletedSources, name) {
|
||||
s.data.DeletedSources = append(s.data.DeletedSources, name)
|
||||
}
|
||||
return removed || containsString(s.data.DeletedSources, name), s.persistLocked()
|
||||
}
|
||||
|
||||
func (s *Store) DeletedSources() map[string]bool {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
out := map[string]bool{}
|
||||
for _, name := range s.data.DeletedSources {
|
||||
out[name] = true
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func (s *Store) DeleteAdapter(name string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
if !containsString(s.data.DeletedAdapters, name) {
|
||||
s.data.DeletedAdapters = append(s.data.DeletedAdapters, name)
|
||||
}
|
||||
return s.persistLocked()
|
||||
}
|
||||
|
||||
func (s *Store) RestoreAdapter(name string) error {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
s.data.DeletedAdapters = removeString(s.data.DeletedAdapters, name)
|
||||
return s.persistLocked()
|
||||
}
|
||||
|
||||
func (s *Store) DeletedAdapters() map[string]bool {
|
||||
s.mu.Lock()
|
||||
defer s.mu.Unlock()
|
||||
out := map[string]bool{}
|
||||
for _, name := range s.data.DeletedAdapters {
|
||||
out[name] = true
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func (s *Store) persistLocked() error {
|
||||
@ -84,6 +122,25 @@ func (s *Store) persistLocked() error {
|
||||
return os.WriteFile(s.path, b, 0644)
|
||||
}
|
||||
|
||||
func containsString(list []string, s string) bool {
|
||||
for _, x := range list {
|
||||
if x == s {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func removeString(list []string, s string) []string {
|
||||
out := list[:0]
|
||||
for _, x := range list {
|
||||
if x != s {
|
||||
out = append(out, x)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
// ListKeys returns the persisted gateway keys.
|
||||
func (s *Store) ListKeys() []GWKey {
|
||||
s.mu.Lock()
|
||||
|
||||
Reference in New Issue
Block a user