#43: Using model name from the config & fixing the provider name

EinStack · Jan 9, 2024 · d4d400d · d4d400d
1 parent e2df092
commit d4d400d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pkg/providers/cohere/chat.go b/pkg/providers/cohere/chat.go
@@ -180,7 +180,7 @@ func (c *Client) doChatRequest(ctx context.Context, payload *ChatRequest) (*sche
 		ID:       cohereCompletion.ResponseID,
 		Created:  int(time.Now().UTC().Unix()), // Cohere doesn't provide this
 		Provider: providerName,
-		Model:    "command-light", // TODO: this needs to come from config or router as Cohere doesn't provide this
+		Model:    c.config.Model,
 		Cached:   false,
 		ModelResponse: schemas.ProviderResponse{
 			ResponseID: map[string]string{

diff --git a/pkg/providers/cohere/client.go b/pkg/providers/cohere/client.go
@@ -14,7 +14,7 @@ import (
 // TODO: Explore Hertz TLS & resource pooling
 
 const (
-	providerName = "openai"
+	providerName = "cohere"
 )
 
 // ErrEmptyResponse is returned when the Cohere API returns an empty response.