summaryrefslogtreecommitdiffstats
path: root/src/client/groq.rs
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-04-25 20:59:56 +0800
committerGitHub <noreply@github.com>2024-04-25 20:59:56 +0800
commit29a4ffd5146af917af3fba30c7d3b7739c7bad0c (patch)
tree20ef465d84ff017063231a2e894111c7ff62d1e3 /src/client/groq.rs
parente5432ff779641473dbd8e4466ac8307aae87de8d (diff)
downloadaichat-29a4ffd5146af917af3fba30c7d3b7739c7bad0c.tar.gz
feat: support groq (#441)
Diffstat (limited to 'src/client/groq.rs')
-rw-r--r--src/client/groq.rs13
1 files changed, 13 insertions, 0 deletions
diff --git a/src/client/groq.rs b/src/client/groq.rs
new file mode 100644
index 0000000..3ff2a41
--- /dev/null
+++ b/src/client/groq.rs
@@ -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),
+ ]
+);