Revert "chore: upgraded cohere backend (#580)" (#581)

This reverts commit 43b0d707e7.
This commit is contained in:
Alex Jones
2023-07-26 16:59:49 +01:00
committed by GitHub
parent 43b0d707e7
commit 5bb91ff2c9
3 changed files with 6 additions and 30 deletions

View File

@@ -63,9 +63,9 @@ func (c *CohereClient) GetCompletion(ctx context.Context, prompt, promptTmpl str
resp, err := c.client.Generate(cohere.GenerateOptions{
Model: c.model,
Prompt: fmt.Sprintf(strings.TrimSpace(promptTmpl), c.language, prompt),
MaxTokens: uint(2048),
Temperature: float64(0.75),
K: int(0),
MaxTokens: cohere.Uint(2048),
Temperature: cohere.Float64(0.75),
K: cohere.Int(0),
StopSequences: []string{},
ReturnLikelihoods: "NONE",
})