Added max token = 10 when testing the key
This commit is contained in:
parent
6c3e4d4b8a
commit
c25a5772ce
@ -73,6 +73,7 @@ func TestFireworkKey(apiKey string) bool {
|
||||
Model: "accounts/fireworks/models/llama-v2-7b-chat",
|
||||
Messages: fireworkMessages,
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
@ -73,6 +73,7 @@ func TestGroqKey(apiKey string) bool {
|
||||
Model: "llama3-8b-8192",
|
||||
Messages: Message2RequestMessage(groqMessages, ""),
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
@ -71,6 +71,7 @@ func TestMistralKey(apiKey string) bool {
|
||||
Model: "open-mistral-7b",
|
||||
Messages: mistralMessages,
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
@ -76,6 +76,7 @@ func TestNimKey(apiKey string) bool {
|
||||
Model: "meta/llama3-8b-instruct",
|
||||
Messages: nimMessages,
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
@ -73,6 +73,7 @@ func TestOpenaiKey(apiKey string) bool {
|
||||
Model: "gpt-3.5-turbo",
|
||||
Messages: openaiMessages,
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
@ -74,6 +74,7 @@ func TestPerplexityKey(apiKey string) bool {
|
||||
Model: "llama-3-8b-instruct",
|
||||
Messages: perplexityMessages,
|
||||
Temperature: 0,
|
||||
MaxTokens: 10,
|
||||
}
|
||||
|
||||
jsonBody, err := json.Marshal(requestBody)
|
||||
|
Loading…
x
Reference in New Issue
Block a user