From 2e75740d300c7a8968292043ba3c42f3b507a3e3 Mon Sep 17 00:00:00 2001
From: Darren Shepherd <darren@acorn.io>
Date: Thu, 15 Aug 2024 08:43:54 -0700
Subject: [PATCH] bug: fix default model provider

---
 pkg/remote/remote.go | 13 ++-----------
 1 file changed, 2 insertions(+), 11 deletions(-)

diff --git a/pkg/remote/remote.go b/pkg/remote/remote.go
index baa54677..fa1d40c2 100644
--- a/pkg/remote/remote.go
+++ b/pkg/remote/remote.go
@@ -25,7 +25,6 @@ type Client struct {
 	clientsLock     sync.Mutex
 	cache           *cache.Client
 	clients         map[string]clientInfo
-	modelToProvider map[string]string
 	runner          *runner.Runner
 	envs            []string
 	credStore       credentials.CredentialStore
@@ -39,17 +38,13 @@ func New(r *runner.Runner, envs []string, cache *cache.Client, credStore credent
 		envs:            envs,
 		credStore:       credStore,
 		defaultProvider: defaultProvider,
-		modelToProvider: make(map[string]string),
 		clients:         make(map[string]clientInfo),
 	}
 }
 
 func (c *Client) Call(ctx context.Context, messageRequest types.CompletionRequest, status chan<- types.CompletionStatus) (*types.CompletionMessage, error) {
-	c.clientsLock.Lock()
-	provider, ok := c.modelToProvider[messageRequest.Model]
-	c.clientsLock.Unlock()
-
-	if !ok {
+	_, provider := c.parseModel(messageRequest.Model)
+	if provider == "" {
 		return nil, fmt.Errorf("failed to find remote model %s", messageRequest.Model)
 	}
 
@@ -108,10 +103,6 @@ func (c *Client) Supports(ctx context.Context, modelString string) (bool, error)
 		return false, err
 	}
 
-	c.clientsLock.Lock()
-	defer c.clientsLock.Unlock()
-
-	c.modelToProvider[modelString] = providerName
 	return true, nil
 }