1010from hyperspell import Hyperspell , AsyncHyperspell
1111from tests .utils import assert_matches_type
1212from hyperspell .types import (
13+ Memory ,
1314 MemoryStatus ,
14- MemoryGetResponse ,
1515 MemoryListResponse ,
1616 MemoryDeleteResponse ,
1717 MemoryStatusResponse ,
@@ -240,7 +240,7 @@ def test_method_get(self, client: Hyperspell) -> None:
240240 resource_id = "resource_id" ,
241241 source = "reddit" ,
242242 )
243- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
243+ assert_matches_type (Memory , memory , path = ["response" ])
244244
245245 @parametrize
246246 def test_raw_response_get (self , client : Hyperspell ) -> None :
@@ -252,7 +252,7 @@ def test_raw_response_get(self, client: Hyperspell) -> None:
252252 assert response .is_closed is True
253253 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
254254 memory = response .parse ()
255- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
255+ assert_matches_type (Memory , memory , path = ["response" ])
256256
257257 @parametrize
258258 def test_streaming_response_get (self , client : Hyperspell ) -> None :
@@ -264,7 +264,7 @@ def test_streaming_response_get(self, client: Hyperspell) -> None:
264264 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
265265
266266 memory = response .parse ()
267- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
267+ assert_matches_type (Memory , memory , path = ["response" ])
268268
269269 assert cast (Any , response .is_closed ) is True
270270
@@ -657,7 +657,7 @@ async def test_method_get(self, async_client: AsyncHyperspell) -> None:
657657 resource_id = "resource_id" ,
658658 source = "reddit" ,
659659 )
660- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
660+ assert_matches_type (Memory , memory , path = ["response" ])
661661
662662 @parametrize
663663 async def test_raw_response_get (self , async_client : AsyncHyperspell ) -> None :
@@ -669,7 +669,7 @@ async def test_raw_response_get(self, async_client: AsyncHyperspell) -> None:
669669 assert response .is_closed is True
670670 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
671671 memory = await response .parse ()
672- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
672+ assert_matches_type (Memory , memory , path = ["response" ])
673673
674674 @parametrize
675675 async def test_streaming_response_get (self , async_client : AsyncHyperspell ) -> None :
@@ -681,7 +681,7 @@ async def test_streaming_response_get(self, async_client: AsyncHyperspell) -> No
681681 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
682682
683683 memory = await response .parse ()
684- assert_matches_type (MemoryGetResponse , memory , path = ["response" ])
684+ assert_matches_type (Memory , memory , path = ["response" ])
685685
686686 assert cast (Any , response .is_closed ) is True
687687
0 commit comments