Luis Pater
e50cabac4b
- Updated all references from v6 to v7 for `github.com/router-for-me/CLIProxyAPI`. - Ensured consistency in imports within core libraries, tests, and integration tests. - Added missing tests for new features in Redis Protocol integration.
31 lines
1.1 KiB
Go
31 lines
1.1 KiB
Go
package cliproxy
|
|
|
|
import "github.com/router-for-me/CLIProxyAPI/v7/internal/registry"
|
|
|
|
// ModelInfo re-exports the registry model info structure.
|
|
type ModelInfo = registry.ModelInfo
|
|
|
|
// ModelRegistryHook re-exports the registry hook interface for external integrations.
|
|
type ModelRegistryHook = registry.ModelRegistryHook
|
|
|
|
// ModelRegistry describes registry operations consumed by external callers.
|
|
type ModelRegistry interface {
|
|
RegisterClient(clientID, clientProvider string, models []*ModelInfo)
|
|
UnregisterClient(clientID string)
|
|
SetModelQuotaExceeded(clientID, modelID string)
|
|
ClearModelQuotaExceeded(clientID, modelID string)
|
|
ClientSupportsModel(clientID, modelID string) bool
|
|
GetAvailableModels(handlerType string) []map[string]any
|
|
GetAvailableModelsByProvider(provider string) []*ModelInfo
|
|
}
|
|
|
|
// GlobalModelRegistry returns the shared registry instance.
|
|
func GlobalModelRegistry() ModelRegistry {
|
|
return registry.GetGlobalRegistry()
|
|
}
|
|
|
|
// SetGlobalModelRegistryHook registers an optional hook on the shared global registry instance.
|
|
func SetGlobalModelRegistryHook(hook ModelRegistryHook) {
|
|
registry.GetGlobalRegistry().SetHook(hook)
|
|
}
|