Merge pull request #249 from fastlabel/feature/construct-api-with-token

API キーを Client の初期化時に渡せるように修正

Author: yoshikids

README.md

@@ -61,6 +61,13 @@ import fastlabel
 client = fastlabel.Client()
 ```
 
+Or initialize with API Key directly.
+
+```python
+import fastlabel
+client = fastlabel.Client("YOUR_ACCESS_TOKEN")
+```
+
 ### Limitation
 
 API is allowed to call 10000 times per 10 minutes. If you create/delete a large size of tasks, please wait a second for every requests.

fastlabel/__init__.py

@@ -41,8 +41,8 @@
 class Client:
     api = None
 
-    def __init__(self):
-        self.api = Api()
+    def __init__(self, access_token: Optional[str] = None):
+        self.api = Api(access_token=access_token)
 
     # Task Find
 

fastlabel/api.py

@@ -1,5 +1,5 @@
 import os
-from typing import Union
+from typing import Optional, Union
 
 import requests
 
@@ -11,12 +11,13 @@ class Api:
 
     access_token = None
 
-    def __init__(self):
-        if os.environ.get("FASTLABEL_API_URL"):
-            self.base_url = os.environ.get("FASTLABEL_API_URL")
-        if not os.environ.get("FASTLABEL_ACCESS_TOKEN"):
+    def __init__(self, access_token: Optional[str] = None):
+        if api_url := os.environ.get("FASTLABEL_API_URL"):
+            self.base_url = api_url
+        access_token = access_token or os.environ.get("FASTLABEL_ACCESS_TOKEN")
+        if not access_token:
             raise ValueError("FASTLABEL_ACCESS_TOKEN is not configured.")
-        self.access_token = "Bearer " + os.environ.get("FASTLABEL_ACCESS_TOKEN")
+        self.access_token = "Bearer " + access_token
 
     def get_request(self, endpoint: str, params=None) -> Union[dict, list]:
         """Makes a get request to an endpoint.