feat: support groq (#441)

pull/442/head
sigoden 1 month ago committed by GitHub
parent e5432ff779
commit 29a4ffd514
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -65,6 +65,10 @@ clients:
- type: cohere
api_key: xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
# See https://console.groq.com/docs/quickstart
- type: groq
api_key: gsk_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
# Any openai-compatible API providers
- type: openai-compatible
name: localai

@ -0,0 +1,13 @@
openai_compatible_module!(
GroqConfig,
GroqClient,
"https://api.groq.com/openai/v1",
[
// https://console.groq.com/docs/models
("llama3-8b-8192", "text", 8192),
("llama3-70b-8192", "text", 8192),
("llama2-70b-4096", "text", 4096),
("mixtral-8x7b-32768", "text", 32768),
("gemma-7b-it", "text", 8192),
]
);

@ -28,6 +28,7 @@ register_client!(
(claude, "claude", ClaudeConfig, ClaudeClient),
(mistral, "mistral", MistralConfig, MistralClient),
(cohere, "cohere", CohereConfig, CohereClient),
(groq, "groq", GroqConfig, GroqClient),
(ollama, "ollama", OllamaConfig, OllamaClient),
(ernie, "ernie", ErnieConfig, ErnieClient),
(qianwen, "qianwen", QianwenConfig, QianwenClient),

Loading…
Cancel
Save