feat: add flexible API loading methods to ClientPolling

Author: SepineTam

src/openai_api_polling/polling/client_polling.py

@@ -7,6 +7,10 @@
 # @Email  : sepinetam@gmail.com
 # @File   : client_polling.py
 
+from __future__ import annotations
+
+from os import PathLike
+from pathlib import Path
 from typing import List
 
 import anthropic
@@ -17,15 +21,32 @@
 
 
 class ClientPolling:
+    api_file = Path.home() / ".api"  # Default api file path
+
     def __init__(self,
-                 api_keys: APIPolling | List[str],
+                 api_keys: APIPolling | List[str] = None,
                  *args, **kwargs):
-        self.api_polling = APIPolling(api_keys)
+        if api_keys is None:
+            self.api_polling = APIPolling.load_api()
+        else:
+            if isinstance(api_keys, APIPolling):
+                self.api_polling = api_keys
+            elif isinstance(api_keys, list):
+                self.api_polling = APIPolling(api_keys)
+            else:
+                raise TypeError(f"api_keys must be APIPolling or list or None, but got {type(api_keys)}")
         self.client_kwargs = kwargs.copy()
 
     def __len__(self):
         return self.api_polling.polling_length
 
+    @classmethod
+    def load_from_api(cls,
+                      api_file: str | PathLike = None,
+                      *args, **kwargs) -> ClientPolling:
+        api_polling = APIPolling.load_api(api_file or cls.api_file)
+        return cls(api_polling, *args, **kwargs)
+
     @property
     def client(self) -> OpenAI:
         client = OpenAI(
@@ -44,6 +65,8 @@ def async_client(self) -> AsyncOpenAI:
 
 
 class GeminiClientPolling(ClientPolling):
+    api_file = Path.home() / ".gemini.api"
+
     @property
     def client(self) -> genai.Client:
         client = genai.Client(
@@ -61,6 +84,8 @@ def async_client(self) -> genai.Client:
 
 
 class ClaudeClientPolling(ClientPolling):
+    api_file = Path.home() / ".claude.api"
+
     @property
     def client(self) -> anthropic.Anthropic:
         client = anthropic.Anthropic(