update: improve polling module with type hints and copy constructor

- Add type hints using from __future__ import annotations
- Add copy constructor support for APIPolling class
- Rename api_key_polling to api_polling for consistency
- Fix AsyncOpenAI return type annotation

Author: SepineTam

.github/workflows/publish.yml .github/workflows/publish-pypi.yml.github/workflows/publish.yml renamed to .github/workflows/publish-pypi.yml

@@ -7,13 +7,18 @@
 # @Email  : sepinetam@gmail.com
 # @File   : api_polling.py
 
+from __future__ import annotations
+
 from typing import List
 
 
 class APIPolling:
     def __init__(self,
-                 api_keys: List[str]):
-        self.api_polling = api_keys
+                 api_keys: APIPolling | List[str]):
+        if isinstance(api_keys, APIPolling):
+            self.api_polling = list(api_keys.api_polling).copy()
+        else:
+            self.api_polling = api_keys
         self._index = 0
 
     def __str__(self):

src/openai_api_polling/polling/api_polling.py

@@ -16,27 +16,26 @@
 
 class ClientPolling:
     def __init__(self,
-                 api_keys: List[str],
+                 api_keys: APIPolling | List[str],
                  *args, **kwargs):
-        self.api_key_polling = APIPolling(api_keys)
+        self.api_polling = APIPolling(api_keys)
         self.openai_kwargs = kwargs.copy()
 
     def __len__(self):
-        return self.api_key_polling.polling_length
+        return self.api_polling.polling_length
 
     @property
     def client(self) -> OpenAI:
         client = OpenAI(
-            api_key=self.api_key_polling.api_key,
+            api_key=self.api_polling.api_key,
             **self.openai_kwargs
         )
         return client
 
     @property
-    def async_client(self) -> OpenAI:
+    def async_client(self) -> AsyncOpenAI:
         client = AsyncOpenAI(
-            api_key=self.api_key_polling.api_key,
+            api_key=self.api_polling.api_key,
             **self.openai_kwargs
         )
         return client
-