fix ugly rendering of 'omit' in UI due to incorrect strip

python/langsmith/wrappers/_openai.py

@@ -1,3 +1,3 @@
 from __future__ import annotations
 
 import functools
@@ -35,24 +35,33 @@
 
 
 @functools.lru_cache
-def _get_not_given() -> Optional[type]:
+def _get_omit_types() -> tuple[type, ...]:
+    """Get NotGiven/Omit sentinel types used by OpenAI SDK."""
+    types = []
     try:
         from openai._types import NotGiven
 
-        return NotGiven
+        types.append(NotGiven)
     except ImportError:
-        return None
+        pass
+    try:
+        from openai._types import Omit
+
+        types.append(Omit)
+    except ImportError:
+        pass
+    return tuple(types)
 
 
 def _strip_not_given(d: dict) -> dict:
     try:
-        not_given = _get_not_given()
-        if not_given is None:
+        omit_types = _get_omit_types()
+        if not omit_types:
             return d
         return {
             k: v
             for k, v in d.items()
-            if not (isinstance(v, not_given) or (k.startswith("extra_") and v is None))
+            if not (isinstance(v, omit_types) or (k.startswith("extra_") and v is None))
         }
     except Exception as e:
         logger.error(f"Error stripping NotGiven: {e}")