handle exceptions

Author: alrocar

README.md

@@ -6,6 +6,10 @@ If you want to manage Workspaces, Data Sources and Pipes you might be looking fo
 
 The SDK is meant to programatically ingest `NDJSON` data or send any request to an `API` instance.
 
+It contains handlers for:
+- logging events to a Tinybird Data Source from your Python module.
+- logging events from [litellm](https://www.litellm.ai/) to a Tinybird Data Source.
+
 ## Ingest to a Tinybird DataSource
 
 ```python
@@ -108,6 +112,8 @@ pip install tinybird-python-sdk[ai]
 Then use the following handler:
 
 ```python
+import litellm
+from litellm import acompletion
 from tb.litellm.handler import TinybirdLitellmAsyncHandler
 
 customHandler = TinybirdLitellmAsyncHandler(

pyproject.toml

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

tb/litellm/handler.py

@@ -11,6 +11,7 @@
 from datetime import datetime
 from tb.a.api import AsyncAPI as AsyncTinybird
 from tb.api import API as Tinybird
+import logging
 
 
 class CustomJSONEncoder(json.JSONEncoder):
@@ -90,33 +91,44 @@ def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     def log_success_event(self, kwargs, response_obj, start_time, end_time):
-        data = self._extract_data(kwargs, response_obj, start_time, end_time)
-        self.api.send(
-            f"events?name={self.datasource_name}",
-            data=data
-        )
-
+        try:
+            data = self._extract_data(kwargs, response_obj, start_time, end_time)
+            self.api.send(
+                f"events?name={self.datasource_name}",
+                data=data
+            )
+        except Exception as e:
+            logging.error(f"Error logging success event: {e}")
     def log_failure_event(self, kwargs, response_obj, start_time, end_time):
-        data = self._extract_data(kwargs, response_obj, start_time, end_time)
-        self.api.send(
-            f"events?name={self.datasource_name}",
-            data=data
-        )
+        try:
+            data = self._extract_data(kwargs, response_obj, start_time, end_time)
+            self.api.send(
+                f"events?name={self.datasource_name}",
+                data=data
+            )
+        except Exception as e:
+            logging.error(f"Error logging failure event: {e}")
 
 class TinybirdLitellmAsyncHandler(TinybirdLitellmHandler):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
 
     async def async_log_success_event(self, kwargs, response_obj, start_time, end_time):
-        data = self._extract_data(kwargs, response_obj, start_time, end_time)
-        await self.async_api.send(
-            f"events?name={self.datasource_name}",
-            data=data
-    )
+        try:
+            data = self._extract_data(kwargs, response_obj, start_time, end_time)
+            await self.async_api.send(
+                f"events?name={self.datasource_name}",
+                data=data
+            )
+        except Exception as e:
+            logging.error(f"Error logging success event: {e}")
 
     async def async_log_failure_event(self, kwargs, response_obj, start_time, end_time):
-        data = self._extract_data(kwargs, response_obj, start_time, end_time)
-        await self.async_api.send(
-            f"events?name={self.datasource_name}",
-            data=data
-        )
+        try:
+            data = self._extract_data(kwargs, response_obj, start_time, end_time)
+            await self.async_api.send(
+                f"events?name={self.datasource_name}",
+                data=data
+            )
+        except Exception as e:
+            logging.error(f"Error logging failure event: {e}")