feat: add instructor tracking

Author: joschkabraun

parea/client.py

@@ -76,19 +76,25 @@ def __attrs_post_init__(self):
             parea_logger.set_client(self._client)
             parea_logger.set_project_uuid(self.project_uuid)
 
-    def wrap_openai_client(self, client: "OpenAI") -> None:
+    def wrap_openai_client(self, client: "OpenAI", integration: Optional[str] = None) -> None:
         """Only necessary for instance client with OpenAI version >= 1.0.0"""
         from parea.wrapper import OpenAIWrapper
         from parea.wrapper.openai_beta_wrapper import BetaWrappers
 
         OpenAIWrapper().init(log=logger_all_possible, cache=self.cache, module_client=client)
         BetaWrappers(client).init()
 
-    def wrap_anthropic_client(self, client: "Anthropic") -> None:
+        if integration:
+            self._client.add_integration(integration)
+
+    def wrap_anthropic_client(self, client: "Anthropic", integration: Optional[str] = None) -> None:
         from parea.wrapper.anthropic.anthropic import AnthropicWrapper
 
         AnthropicWrapper().init(log=logger_all_possible, cache=self.cache, client=client)
 
+        if integration:
+            self._client.add_integration(integration)
+
     def auto_trace_openai_clients(self) -> None:
         import openai
 

parea/parea_logger.py

@@ -66,7 +66,7 @@ def record_vendor_log(self, data: Dict[str, Any], vendor: TraceIntegrations) ->
         data["project_uuid"] = self._project_uuid
         if experiment_uuid := os.getenv(PAREA_OS_ENV_EXPERIMENT_UUID, None):
             data["experiment_uuid"] = experiment_uuid
-        self._client.add_integration('langchain')
+        self._client.add_integration("langchain")
         self._client.request(
             "POST",
             VENDOR_LOG_ENDPOINT.format(vendor=vendor.value),
@@ -77,7 +77,7 @@ async def arecord_vendor_log(self, data: Dict[str, Any], vendor: TraceIntegratio
         data["project_uuid"] = self._project_uuid
         if experiment_uuid := os.getenv(PAREA_OS_ENV_EXPERIMENT_UUID, None):
             data["experiment_uuid"] = experiment_uuid
-        self._client.add_integration('langchain')
+        self._client.add_integration("langchain")
         await self._client.request_async(
             "POST",
             VENDOR_LOG_ENDPOINT.format(vendor=vendor.value),