feat: add api-polling and openai-client-polling

Author: SepineTam

src/openai_api_polling/polling/__init__.py

@@ -0,0 +1,7 @@
+from .api_polling import APIPolling
+from .client_polling import ClientPolling
+
+__all__ = [
+    "APIPolling",
+    "ClientPolling"
+]

src/openai_api_polling/polling/api_polling.py

@@ -0,0 +1,74 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2025 - Present Sepine Tam, Inc. All Rights Reserved
+#
+# @Author : Sepine Tam (谭淞)
+# @Email  : sepinetam@gmail.com
+# @File   : api_polling.py
+
+from typing import List
+
+
+class APIPolling:
+    def __init__(self,
+                 api_keys: List[str]):
+        self.api_polling = api_keys
+        self._index = 0
+
+    def __str__(self):
+        r = f"API Polling, this polling has {self.polling_length} APIs:\n"
+        for index in range(self.polling_length):
+            r += f"index: {index:>3} | API: {self.mask_api_key(self.api_polling[index])}\n"
+        return r
+
+    def __len__(self):
+        return self.polling_length
+
+    @property
+    def api_key(self):
+        return self._get_next_api()
+
+    @property
+    def polling_length(self):
+        return len(self.api_polling)
+
+    def _index_plus(self):
+        self._index += 1
+        if self._index >= self.polling_length:
+            self.reset_index(0)
+        return self._index
+
+    def _get_next_api(self):
+        next_api = self.api_polling[self._index]
+        self._index_plus()
+        return next_api
+
+    def reset_index(self, index: int = 0):
+        if index > self.polling_length:
+            index = 0
+        self._index = index
+
+    @staticmethod
+    def mask_api_key(api_key: str,
+                     visible: int = 3,
+                     mask_char: str = "*",
+                     mask_len: int = 3):
+        if not api_key:
+            return ""
+        prefix = api_key[:visible]
+        suffix = api_key[-visible:]
+        return prefix + (mask_char * mask_len) + suffix
+
+
+if __name__ == "__main__":
+    api_list = [
+        "sk-thisiskey1", "sk-thisiskey2", "sk-thisiskey3",
+    ]
+    api_polling = APIPolling(api_list)
+
+    print(api_polling)
+
+    # To show 10 times api
+    for _ in range(10):
+        print(api_polling.api_key)

src/openai_api_polling/polling/client_polling.py

@@ -0,0 +1,42 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2025 - Present Sepine Tam, Inc. All Rights Reserved
+#
+# @Author : Sepine Tam (谭淞)
+# @Email  : sepinetam@gmail.com
+# @File   : client_polling.py
+
+from typing import List
+
+from openai import AsyncOpenAI, OpenAI
+
+from .api_polling import APIPolling
+
+
+class ClientPolling:
+    def __init__(self,
+                 api_keys: List[str],
+                 *args, **kwargs):
+        self.api_key_polling = APIPolling(api_keys)
+        self.openai_kwargs = kwargs.copy()
+
+    def __len__(self):
+        return self.api_key_polling.polling_length
+
+    @property
+    def client(self) -> OpenAI:
+        client = OpenAI(
+            api_key=self.api_key_polling.api_key,
+            **self.openai_kwargs
+        )
+        return client
+
+    @property
+    def async_client(self) -> OpenAI:
+        client = AsyncOpenAI(
+            api_key=self.api_key_polling.api_key,
+            **self.openai_kwargs
+        )
+        return client
+