add groq provider

xavimondev · Aug 2, 2024 · 74ba342 · 74ba342
1 parent 2ccfe26
commit 74ba342
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,11 @@
 # seo-ai
 
+## 0.5.0
+
+### Minor Changes
+
+- add groq provider
+
 ## 0.4.0
 
 ### Minor Changes

diff --git a/README.md b/README.md
@@ -34,6 +34,15 @@ npx seo-ai config set OPENAI_API_KEY=<your-key>
 npx seo-ai config set MISTRAL_API_KEY=<your-key>
 ```
 
+### GROQ_API_KEY
+
+- Go to the [Groq Console](https://console.groq.com/keys) and generate your API key.
+- Set the key:
+
+```bash
+npx seo-ai config set GROQ_API_KEY=<your-key>
+```
+
 ### Run
 
 ```sh

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "seo-ai",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "Command-line tool to generate SEO metadata and HTML meta tags using AI models",
   "type": "module",
   "license": "MIT",

diff --git a/src/commands/generate.ts b/src/commands/generate.ts
@@ -95,6 +95,12 @@ export const generate = new Command()
             compatibility: 'strict'
           })
           model = openai('gpt-4o')
+        } else if (lastProvider === 'groq') {
+          const groq = createOpenAI({
+            baseURL: 'https://api.groq.com/openai/v1',
+            apiKey: apiKey
+          })
+          model = groq('llama-3.1-70b-versatile')
         } else {
           logger.error('Invalid provider')
           process.exit(0)

diff --git a/src/utils/conf.ts b/src/utils/conf.ts
@@ -1,10 +1,11 @@
 import Conf from 'conf'
 
-export type Providers = 'openai' | 'mistral'
+export type Providers = 'openai' | 'mistral' | 'groq'
 
 const providers: Record<string, Providers> = {
   OPENAI_API_KEY: 'openai',
-  MISTRAL_API_KEY: 'mistral'
+  MISTRAL_API_KEY: 'mistral',
+  GROQ_API_KEY: 'groq'
 }
 
 export const getProvider = ({ keyName }: { keyName: string }) => providers[keyName]