feat: restructure client APIs and remove provider registration

This commit restructures the client APIs to be more streamlined and remove the need for explicit provider registration.

The following changes were made:

- Removed provider registration functions (RegisterLLMProvider, RegisterTTSProvider, RegisterEmbeddingProvider) and the associated provider maps.
- Moved client creation logic into the respective driver functions (NewLLMClient, NewTTSClient, NewEmbeddingClient).
- Added an error (ErrNoSuchProvider) for when a requested provider is not found.
- Modified the client creation functions to take a provider string as an argument and return an error if the provider is not found.

These changes simplify the API and make it easier for users to use Coord.

Author: lemon-mint

coord.go

@@ -1,97 +1,47 @@
 package coord
 
 import (
-	"sort"
-	"sync"
+	"context"
+	"errors"
 
+	"github.com/lemon-mint/coord/pconf"
 	"github.com/lemon-mint/coord/provider"
 )
 
-var (
-	llmProvidersMu sync.RWMutex
-	llmProviders   = make(map[string]provider.LLMProvider)
+var ErrNoSuchProvider = errors.New("coord: no such provider")
 
-	ttsProvidersMu sync.RWMutex
-	ttsProviders   = make(map[string]provider.TTSProvider)
-
-	embeddingProvidersMu sync.RWMutex
-	embeddingProviders   = make(map[string]provider.EmbeddingProvider)
-)
-
-// ListLLMProviders returns the names of the registered llm providers.
-func ListLLMProviders() []string {
+func NewLLMClient(ctx context.Context, provider string, configs ...pconf.Config) (provider.LLMClient, error) {
 	llmProvidersMu.RLock()
 	defer llmProvidersMu.RUnlock()
-	list := make([]string, 0, len(llmProviders))
-	for name := range llmProviders {
-		list = append(list, name)
+
+	driver, ok := llmProviders[provider]
+	if !ok {
+		return nil, ErrNoSuchProvider
 	}
-	sort.Strings(list)
-	return list
-}
 
-// RegisterLLMProvider registers a llm provider.
-func RegisterLLMProvider(name string, p provider.LLMProvider) {
-	llmProvidersMu.Lock()
-	defer llmProvidersMu.Unlock()
-	llmProviders[name] = p
+	return driver.NewLLMClient(ctx, configs...)
 }
 
-// RemoveLLMProvider removes a llm provider.
-func RemoveLLMProvider(name string) {
-	llmProvidersMu.Lock()
-	defer llmProvidersMu.Unlock()
-	delete(llmProviders, name)
-}
+func NewEmbeddingClient(ctx context.Context, provider string, configs ...pconf.Config) (provider.EmbeddingClient, error) {
+	embeddingProvidersMu.RLock()
+	defer embeddingProvidersMu.RUnlock()
 
-// ListTTSProviders returns the names of the registered tts providers.
-func ListTTSProviders() []string {
-	ttsProvidersMu.RLock()
-	defer ttsProvidersMu.RUnlock()
-	list := make([]string, 0, len(ttsProviders))
-	for name := range ttsProviders {
-		list = append(list, name)
+	driver, ok := embeddingProviders[provider]
+	if !ok {
+		return nil, ErrNoSuchProvider
 	}
-	sort.Strings(list)
-	return list
-}
 
-// RegisterTTSProvider registers a tts provider.
-func RegisterTTSProvider(name string, p provider.TTSProvider) {
-	ttsProvidersMu.Lock()
-	defer ttsProvidersMu.Unlock()
-	ttsProviders[name] = p
+	return driver.NewEmbeddingClient(ctx, configs...)
 }
 
-// RemoveTTSProvider removes a tts provider.
-func RemoveTTSProvider(name string) {
-	ttsProvidersMu.Lock()
-	defer ttsProvidersMu.Unlock()
-	delete(ttsProviders, name)
-}
+func NewTTSClient(ctx context.Context, provider string, configs ...pconf.Config) (provider.TTSClient, error) {
+	ttsProvidersMu.RLock()
+	defer ttsProvidersMu.RUnlock()
 
-// ListEmbeddingProviders returns the names of the registered embedding providers.
-func ListEmbeddingProviders() []string {
-	embeddingProvidersMu.RLock()
-	defer embeddingProvidersMu.RUnlock()
-	list := make([]string, 0, len(embeddingProviders))
-	for name := range embeddingProviders {
-		list = append(list, name)
+	driver, ok := ttsProviders[provider]
+	if !ok {
+		return nil, ErrNoSuchProvider
 	}
-	sort.Strings(list)
-	return list
-}
-
-// RegisterEmbeddingProvider registers an embedding provider.
-func RegisterEmbeddingProvider(name string, p provider.EmbeddingProvider) {
-	embeddingProvidersMu.Lock()
-	defer embeddingProvidersMu.Unlock()
-	embeddingProviders[name] = p
-}
 
-// RemoveEmbeddingProvider removes an embedding provider.
-func RemoveEmbeddingProvider(name string) {
-	embeddingProvidersMu.Lock()
-	defer embeddingProvidersMu.Unlock()
-	delete(embeddingProviders, name)
+	return driver.NewTTSClient(ctx, configs...)
 }

providers.go

@@ -0,0 +1,97 @@
+package coord
+
+import (
+	"sort"
+	"sync"
+
+	"github.com/lemon-mint/coord/provider"
+)
+
+var (
+	llmProvidersMu sync.RWMutex
+	llmProviders   = make(map[string]provider.LLMProvider)
+
+	ttsProvidersMu sync.RWMutex
+	ttsProviders   = make(map[string]provider.TTSProvider)
+
+	embeddingProvidersMu sync.RWMutex
+	embeddingProviders   = make(map[string]provider.EmbeddingProvider)
+)
+
+// ListLLMProviders returns the names of the registered llm providers.
+func ListLLMProviders() []string {
+	llmProvidersMu.RLock()
+	defer llmProvidersMu.RUnlock()
+	list := make([]string, 0, len(llmProviders))
+	for name := range llmProviders {
+		list = append(list, name)
+	}
+	sort.Strings(list)
+	return list
+}
+
+// RegisterLLMProvider registers a llm provider.
+func RegisterLLMProvider(name string, p provider.LLMProvider) {
+	llmProvidersMu.Lock()
+	defer llmProvidersMu.Unlock()
+	llmProviders[name] = p
+}
+
+// RemoveLLMProvider removes a llm provider.
+func RemoveLLMProvider(name string) {
+	llmProvidersMu.Lock()
+	defer llmProvidersMu.Unlock()
+	delete(llmProviders, name)
+}
+
+// ListTTSProviders returns the names of the registered tts providers.
+func ListTTSProviders() []string {
+	ttsProvidersMu.RLock()
+	defer ttsProvidersMu.RUnlock()
+	list := make([]string, 0, len(ttsProviders))
+	for name := range ttsProviders {
+		list = append(list, name)
+	}
+	sort.Strings(list)
+	return list
+}
+
+// RegisterTTSProvider registers a tts provider.
+func RegisterTTSProvider(name string, p provider.TTSProvider) {
+	ttsProvidersMu.Lock()
+	defer ttsProvidersMu.Unlock()
+	ttsProviders[name] = p
+}
+
+// RemoveTTSProvider removes a tts provider.
+func RemoveTTSProvider(name string) {
+	ttsProvidersMu.Lock()
+	defer ttsProvidersMu.Unlock()
+	delete(ttsProviders, name)
+}
+
+// ListEmbeddingProviders returns the names of the registered embedding providers.
+func ListEmbeddingProviders() []string {
+	embeddingProvidersMu.RLock()
+	defer embeddingProvidersMu.RUnlock()
+	list := make([]string, 0, len(embeddingProviders))
+	for name := range embeddingProviders {
+		list = append(list, name)
+	}
+	sort.Strings(list)
+	return list
+}
+
+// RegisterEmbeddingProvider registers an embedding provider.
+func RegisterEmbeddingProvider(name string, p provider.EmbeddingProvider) {
+	embeddingProvidersMu.Lock()
+	defer embeddingProvidersMu.Unlock()
+	embeddingProviders[name] = p
+}
+
+// RemoveEmbeddingProvider removes an embedding provider.
+func RemoveEmbeddingProvider(name string) {
+	embeddingProvidersMu.Lock()
+	defer embeddingProvidersMu.Unlock()
+	delete(embeddingProviders, name)
+}