From 345162c764460459b4602c217f8dd4207a2edb3e Mon Sep 17 00:00:00 2001
From: Tanay K <128210022+chainSAW-crypto@users.noreply.github.com>
Date: Sun, 14 Sep 2025 10:48:43 +0530
Subject: [PATCH] Add GroqClient class for Groq API integration

Since groq is also a widely used api service, enabling groq client will definitely support the wider horizon and support the community
---
 rank_gpt.py | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/rank_gpt.py b/rank_gpt.py
index d4eaf70..5ce7621 100644
--- a/rank_gpt.py
+++ b/rank_gpt.py
@@ -52,6 +52,34 @@ def text(self, *args, return_text=False, reduce_length=False, **kwargs):
         return completion
 
 
+class GroqClient:
+    def __init__(self, keys=None):
+        from groq import Groq
+        if isinstance(keys, str):
+            keys = [keys]
+        if keys is None:
+            raise ValueError("Please provide Groq API Key.")
+        
+        self.key = keys
+        self.api_key = keys[0]  # Use first key
+        self.client = Groq(api_key=self.api_key)
+
+    def chat(self, *args, return_text=False, reduce_length=False, **kwargs):
+        while True:
+            try:
+                completion = self.client.chat.completions.create(*args, **kwargs, timeout=30)
+                break
+            except Exception as e:
+                print(str(e))
+                if "maximum context length" in str(e).lower():
+                    print('reduce_length')
+                    return 'ERROR::reduce_length'
+                time.sleep(0.1)
+        if return_text:
+            completion = completion.choices[0].message.content
+        return completion
+
+
 class ClaudeClient:
     def __init__(self, keys):
         from anthropic import Anthropic, HUMAN_PROMPT, AI_PROMPT