Merge pull request #885 from parea-ai/PAI-1151-auto-trace-marvin

feat: auto trace marvin

parea/client.py

@@ -120,6 +120,14 @@ def auto_trace_openai_clients(self, integration: Optional[str]) -> None:
         if integration:
             self._client.add_integration(integration)
 
+        if integration == "marvin":
+            import marvin
+
+            marvin.utilities.openai.Client = patch_openai_client_classes(marvin.utilities.openai.Client, self)
+            marvin.utilities.openai.AsyncClient = patch_openai_client_classes(marvin.utilities.openai.AsyncClient, self)
+            marvin.utilities.openai.AzureOpenAI = patch_openai_client_classes(marvin.utilities.openai.AzureOpenAI, self)
+            marvin.utilities.openai.AsyncAzureOpenAI = patch_openai_client_classes(marvin.utilities.openai.AsyncAzureOpenAI, self)
+
     def trace_dspy(self):
         from parea.utils.trace_integrations.dspy import DSPyInstrumentor
 

parea/cookbook/trace_marvin.py

@@ -0,0 +1,21 @@
+import os
+
+import marvin
+from dotenv import load_dotenv
+from pydantic import BaseModel, Field
+
+from parea import Parea
+
+load_dotenv()
+
+p = Parea(api_key=os.getenv("PAREA_API_KEY"))
+p.auto_trace_openai_clients("marvin")
+
+
+class Location(BaseModel):
+    city: str
+    state: str = Field(description="2-letter abbreviation")
+
+
+result = marvin.cast("the big apple", Location)
+print(result)

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 [tool.poetry]
 name = "parea-ai"
 packages = [{ include = "parea" }]
-version = "0.2.155"
+version = "0.2.156"
 description = "Parea python sdk"
 readme = "README.md"
 authors = ["joel-parea-ai <[email protected]>"]