test: fix failing timestamp granularities tests

Likely due to openai/openai-python#1757
speaches-ai · Jan 26, 2025 · b57deee · b57deee
1 parent 18c4dab
commit b57deee
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 12 deletions.
diff --git a/tests/api_timestamp_granularities_test.py b/tests/api_timestamp_granularities_test.py
@@ -36,11 +36,10 @@ async def test_api_verbose_json_response_format_and_timestamp_granularities_comb
         timestamp_granularities=timestamp_granularities,
     )
 
-    assert transcription.__pydantic_extra__
     if "word" in timestamp_granularities:
-        assert transcription.__pydantic_extra__.get("segments") is not None
-        assert transcription.__pydantic_extra__.get("words") is not None
+        assert transcription.segments is not None
+        assert transcription.words is not None
     else:
         # Unless explicitly requested, words are not present
-        assert transcription.__pydantic_extra__.get("segments") is not None
-        assert transcription.__pydantic_extra__.get("words") is None
+        assert transcription.segments is not None
+        assert transcription.words is None
diff --git a/tests/openai_timestamp_granularities_test.py b/tests/openai_timestamp_granularities_test.py
@@ -46,15 +46,14 @@ async def test_openai_verbose_json_response_format_and_timestamp_granularities_c
         timestamp_granularities=timestamp_granularities,
     )
 
-    assert transcription.__pydantic_extra__
     if timestamp_granularities == ["word"]:
         # This is an exception where segments are not present
-        assert transcription.__pydantic_extra__.get("segments") is None
-        assert transcription.__pydantic_extra__.get("words") is not None
+        assert transcription.segments is None
+        assert transcription.words is not None
     elif "word" in timestamp_granularities:
-        assert transcription.__pydantic_extra__.get("segments") is not None
-        assert transcription.__pydantic_extra__.get("words") is not None
+        assert transcription.segments is not None
+        assert transcription.words is not None
     else:
         # Unless explicitly requested, words are not present
-        assert transcription.__pydantic_extra__.get("segments") is not None
-        assert transcription.__pydantic_extra__.get("words") is None
+        assert transcription.segments is not None
+        assert transcription.words is None