remove validation

Author: alrocar

README.md

@@ -127,10 +127,18 @@ litellm.callbacks = [customHandler]
 response = await acompletion(
     model="gpt-3.5-turbo", 
     messages=[{"role": "user", "content": "Hi 👋 - i'm openai"}],
-    stream=True
+    stream=True,
+    metadata={
+        "organization": "tinybird",
+        "environment": "dev",
+        "project": "litellm_test",
+        "chat_id": "1234567890",
+    },
 )
 ```
 
+Track custom metadata using the `metadata` dictionary.
+
 This is the schema for the `litellm` data source:
 
 ```sql

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tinybird-python-sdk"
-version = "0.3.2"
+version = "0.3.3"
 description = "Python SDK for Tinybird"
 readme = "README.md"
 authors = [

tb/a/api.py

@@ -25,7 +25,6 @@ def __init__(
         TOKEN_ERROR = f"Token must be a valid Tinybird token for {self.api_url}. Check the `api_url` param is correct and the token has the right permissions. {self.ui_url()}/tokens"
         if not token:
             logging.critical(TOKEN_ERROR)
-            sys.exit(-1)
         self.token = token
         self._session: Optional[aiohttp.ClientSession] = None
 

tb/api.py

@@ -21,7 +21,6 @@ def __init__(
         TOKEN_ERROR = f"Token must be a valid Tinybird token for {self.api_url}. Check the `api_url` param is correct and the token has the right permissions. {self.ui_url()}/tokens"
         if not token:
             logging.critical(TOKEN_ERROR)
-            sys.exit(-1)
         self.token = token
         retry = Retry(total=retry_total, backoff_factor=backoff_factor)
         adapter = HTTPAdapter(max_retries=retry)
@@ -35,14 +34,6 @@ def __init__(
         self.rate_limit_reset = 0
         self.retry_after = 1
 
-        # check the token is valid
-        try:
-            self.get("/datasources")
-        except requests.HTTPError as e:
-            if e.response.status_code == 403:
-                logging.error(TOKEN_ERROR)
-                sys.exit(-1)
-
     def ui_url(self):
         return self.api_url.replace("api", "ui")
 

tb/litellm/handler.py

@@ -40,12 +40,6 @@ def __init__(
         self.api_url = api_url
         self.datasource_name = datasource_name
         self.api_version = api_version
-        self.api = Tinybird(
-            token=self.token, api_url=self.api_url, version=self.api_version
-        )
-        self.async_api = AsyncTinybird(
-            token=self.token, api_url=self.api_url, version=self.api_version
-        )
 
     def _extract_data(self, kwargs, response_obj, start_time, end_time):
         api_key = kwargs.get("api_key")
@@ -108,6 +102,9 @@ def _extract_data(self, kwargs, response_obj, start_time, end_time):
 class TinybirdLitellmSyncHandler(TinybirdLitellmHandler):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+        self.api = Tinybird(
+            token=self.token, api_url=self.api_url, version=self.api_version
+        )
 
     def log_success_event(self, kwargs, response_obj, start_time, end_time):
         try:
@@ -127,6 +124,9 @@ def log_failure_event(self, kwargs, response_obj, start_time, end_time):
 class TinybirdLitellmAsyncHandler(TinybirdLitellmHandler):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
+        self.async_api = AsyncTinybird(
+            token=self.token, api_url=self.api_url, version=self.api_version
+        )
 
     async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
         try: