Minimize Repr (#1155)

langchain-ai · Oct 30, 2024 · 3f0ef5e · 3f0ef5e
1 parent 812c0b0
commit 3f0ef5e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/python/langsmith/run_trees.py b/python/langsmith/run_trees.py
@@ -479,6 +479,13 @@ def to_headers(self) -> Dict[str, str]:
         headers["baggage"] = baggage.to_header()
         return headers
 
+    def __repr__(self):
+        """Return a string representation of the RunTree object."""
+        return (
+            f"RunTree(id={self.id}, name='{self.name}', "
+            f"run_type='{self.run_type}', dotted_order='{self.dotted_order}')"
+        )
+
 
 class _Baggage:
     """Baggage header information."""

diff --git a/python/langsmith/schemas.py b/python/langsmith/schemas.py
@@ -273,6 +273,10 @@ def revision_id(self) -> Optional[UUID]:
         """Retrieve the revision ID (if any)."""
         return self.metadata.get("revision_id")
 
+    def __repr__(self):
+        """Return a string representation of the RunBase object."""
+        return f"{self.__class__}(id={self.id}, name='{self.name}', run_type='{self.run_type}')"
+
 
 class Run(RunBase):
     """Run schema when loading from the DB."""