99
1010from isaacus import Isaacus , AsyncIsaacus
1111from tests .utils import assert_matches_type
12- from isaacus .types import EmbeddingCreateResponse
12+ from isaacus .types import Embedding
1313
1414base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
1515
@@ -24,7 +24,7 @@ def test_method_create(self, client: Isaacus) -> None:
2424 model = "kanon-2-embedder" ,
2525 texts = ["Are restraints of trade enforceable under English law?" , "What is a non-compete clause?" ],
2626 )
27- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
27+ assert_matches_type (Embedding , embedding , path = ["response" ])
2828
2929 @pytest .mark .skip (reason = "Prism tests are disabled" )
3030 @parametrize
@@ -36,7 +36,7 @@ def test_method_create_with_all_params(self, client: Isaacus) -> None:
3636 overflow_strategy = "drop_end" ,
3737 task = "retrieval/query" ,
3838 )
39- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
39+ assert_matches_type (Embedding , embedding , path = ["response" ])
4040
4141 @pytest .mark .skip (reason = "Prism tests are disabled" )
4242 @parametrize
@@ -49,7 +49,7 @@ def test_raw_response_create(self, client: Isaacus) -> None:
4949 assert response .is_closed is True
5050 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
5151 embedding = response .parse ()
52- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
52+ assert_matches_type (Embedding , embedding , path = ["response" ])
5353
5454 @pytest .mark .skip (reason = "Prism tests are disabled" )
5555 @parametrize
@@ -62,7 +62,7 @@ def test_streaming_response_create(self, client: Isaacus) -> None:
6262 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
6363
6464 embedding = response .parse ()
65- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
65+ assert_matches_type (Embedding , embedding , path = ["response" ])
6666
6767 assert cast (Any , response .is_closed ) is True
6868
@@ -79,7 +79,7 @@ async def test_method_create(self, async_client: AsyncIsaacus) -> None:
7979 model = "kanon-2-embedder" ,
8080 texts = ["Are restraints of trade enforceable under English law?" , "What is a non-compete clause?" ],
8181 )
82- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
82+ assert_matches_type (Embedding , embedding , path = ["response" ])
8383
8484 @pytest .mark .skip (reason = "Prism tests are disabled" )
8585 @parametrize
@@ -91,7 +91,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncIsaacus) -
9191 overflow_strategy = "drop_end" ,
9292 task = "retrieval/query" ,
9393 )
94- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
94+ assert_matches_type (Embedding , embedding , path = ["response" ])
9595
9696 @pytest .mark .skip (reason = "Prism tests are disabled" )
9797 @parametrize
@@ -104,7 +104,7 @@ async def test_raw_response_create(self, async_client: AsyncIsaacus) -> None:
104104 assert response .is_closed is True
105105 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
106106 embedding = await response .parse ()
107- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
107+ assert_matches_type (Embedding , embedding , path = ["response" ])
108108
109109 @pytest .mark .skip (reason = "Prism tests are disabled" )
110110 @parametrize
@@ -117,6 +117,6 @@ async def test_streaming_response_create(self, async_client: AsyncIsaacus) -> No
117117 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
118118
119119 embedding = await response .parse ()
120- assert_matches_type (EmbeddingCreateResponse , embedding , path = ["response" ])
120+ assert_matches_type (Embedding , embedding , path = ["response" ])
121121
122122 assert cast (Any , response .is_closed ) is True
0 commit comments