feat: support Gemini Client and Anthropic Client

Author: SepineTam

pyproject.toml

@@ -8,6 +8,8 @@ authors = [
 ]
 requires-python = ">=3.11"
 dependencies = [
+    "anthropic>=0.75.0",
+    "google-genai>=1.55.0",
     "openai>=2.0.0",
     "tiktoken>=0.10.0",
 ]

src/openai_api_polling/polling/client_polling.py

@@ -9,6 +9,8 @@
 
 from typing import List
 
+import anthropic
+from google import genai
 from openai import AsyncOpenAI, OpenAI
 
 from .api_polling import APIPolling
@@ -19,7 +21,7 @@ def __init__(self,
                  api_keys: APIPolling | List[str],
                  *args, **kwargs):
         self.api_polling = APIPolling(api_keys)
-        self.openai_kwargs = kwargs.copy()
+        self.client_kwargs = kwargs.copy()
 
     def __len__(self):
         return self.api_polling.polling_length
@@ -28,14 +30,49 @@ def __len__(self):
     def client(self) -> OpenAI:
         client = OpenAI(
             api_key=self.api_polling.api_key,
-            **self.openai_kwargs
+            **self.client_kwargs
         )
         return client
 
     @property
     def async_client(self) -> AsyncOpenAI:
         client = AsyncOpenAI(
             api_key=self.api_polling.api_key,
-            **self.openai_kwargs
+            **self.client_kwargs
+        )
+        return client
+
+
+class GeminiClientPolling(ClientPolling):
+    @property
+    def client(self) -> genai.Client:
+        client = genai.Client(
+            api_key=self.api_polling.api_key,
+            **self.client_kwargs
+        )
+        return client
+
+    @property
+    def async_client(self) -> genai.Client:
+        """
+        Google GenAI SDK not support async client. (At least I have not found that)
+        """
+        return self.client
+
+
+class ClaudeClientPolling(ClientPolling):
+    @property
+    def client(self) -> anthropic.Anthropic:
+        client = anthropic.Anthropic(
+            api_key=self.api_polling.api_key,
+            **self.client_kwargs
+        )
+        return client
+
+    @property
+    def async_client(self) -> anthropic.AsyncAnthropic:
+        client = anthropic.AsyncAnthropic(
+            api_key=self.api_polling.api_key,
+            **self.client_kwargs
         )
         return client