Update check condition

Co-authored-by: Kazuhiro Sera <[email protected]>

Author: HareeshBahuleyan

src/agents/extensions/models/litellm_model.py

@@ -500,7 +500,7 @@ def _fix_tool_message_ordering(
         return fixed_messages
 
     def _remove_not_given(self, value: Any) -> Any:
-        if isinstance(value, (NotGiven, type(omit))):
+        if value is omit or isinstance(value, NotGiven):
             return None
         return value