add tracking decorator

Author: Daniel OBrien

ldai/tracker.py

@@ -1,3 +1,4 @@
+import time
 from typing import Dict, Union
 from ldclient import Context, LDClient
 from ldai.types import BedrockTokenUsage, FeedbackKind, TokenUsage, UnderscoreTokenUsage
@@ -18,6 +19,16 @@ def get_track_data(self):
     def track_duration(self, duration: int) -> None:
         self.ld_client.track('$ld:ai:duration:total', self.context, self.get_track_data(), duration)
 
+    def track_duration_of(self, func):
+            def wrapper(*args, **kwargs):
+                start_time = time.time()
+                result = func(*args, **kwargs)
+                end_time = time.time()
+                duration = int((end_time - start_time) * 1000)  # duration in milliseconds
+                self.track_duration(duration)
+                return result
+            return wrapper
+
     def track_tokens(self, tokens: Union[TokenUsage, UnderscoreTokenUsage, BedrockTokenUsage]) -> None:
         token_metrics = tokens.to_metrics()
         if token_metrics['total'] > 0: