Skip to content

Commit 21a6e5b

Browse files
add tests for legacy methods
1 parent 7b1536e commit 21a6e5b

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

tests/test_client_v2.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,12 @@ def test_chat_stream(self) -> None:
3636
self.assertTrue("content-delta" in events)
3737
self.assertTrue("content-end" in events)
3838
self.assertTrue("message-end" in events)
39+
40+
def test_legacy_methods_available(self) -> None:
41+
self.assertTrue(hasattr(co, "generate"))
42+
self.assertTrue(callable(getattr(co, "generate")))
43+
self.assertTrue(hasattr(co, "generate_stream"))
44+
self.assertTrue(callable(getattr(co, "generate_stream")))
3945

4046
@unittest.skip("Skip v2 test for now")
4147
def test_chat_documents(self) -> None:

0 commit comments

Comments
 (0)