2121)
2222from ._utils import is_given , get_async_library
2323from ._version import __version__
24- from .resources import memory , search , settings , connection
24+ from .resources import memories , settings , connections
2525from ._streaming import Stream as Stream , AsyncStream as AsyncStream
2626from ._exceptions import APIStatusError , SupermemoryError
2727from ._base_client import (
4343
4444
4545class Supermemory (SyncAPIClient ):
46+ memories : memories .MemoriesResource
4647 settings : settings .SettingsResource
47- memory : memory .MemoryResource
48- search : search .SearchResource
49- connection : connection .ConnectionResource
48+ connections : connections .ConnectionsResource
5049 with_raw_response : SupermemoryWithRawResponse
5150 with_streaming_response : SupermemoryWithStreamedResponse
5251
@@ -91,7 +90,7 @@ def __init__(
9190 if base_url is None :
9291 base_url = os .environ .get ("SUPERMEMORY_BASE_URL" )
9392 if base_url is None :
94- base_url = f"https://v2. api.supermemory.ai"
93+ base_url = f"https://api.supermemory.ai/ "
9594
9695 super ().__init__ (
9796 version = __version__ ,
@@ -104,10 +103,9 @@ def __init__(
104103 _strict_response_validation = _strict_response_validation ,
105104 )
106105
106+ self .memories = memories .MemoriesResource (self )
107107 self .settings = settings .SettingsResource (self )
108- self .memory = memory .MemoryResource (self )
109- self .search = search .SearchResource (self )
110- self .connection = connection .ConnectionResource (self )
108+ self .connections = connections .ConnectionsResource (self )
111109 self .with_raw_response = SupermemoryWithRawResponse (self )
112110 self .with_streaming_response = SupermemoryWithStreamedResponse (self )
113111
@@ -116,6 +114,12 @@ def __init__(
116114 def qs (self ) -> Querystring :
117115 return Querystring (array_format = "comma" )
118116
117+ @property
118+ @override
119+ def auth_headers (self ) -> dict [str , str ]:
120+ api_key = self .api_key
121+ return {"Authorization" : f"Bearer { api_key } " }
122+
119123 @property
120124 @override
121125 def default_headers (self ) -> dict [str , str | Omit ]:
@@ -211,10 +215,9 @@ def _make_status_error(
211215
212216
213217class AsyncSupermemory (AsyncAPIClient ):
218+ memories : memories .AsyncMemoriesResource
214219 settings : settings .AsyncSettingsResource
215- memory : memory .AsyncMemoryResource
216- search : search .AsyncSearchResource
217- connection : connection .AsyncConnectionResource
220+ connections : connections .AsyncConnectionsResource
218221 with_raw_response : AsyncSupermemoryWithRawResponse
219222 with_streaming_response : AsyncSupermemoryWithStreamedResponse
220223
@@ -259,7 +262,7 @@ def __init__(
259262 if base_url is None :
260263 base_url = os .environ .get ("SUPERMEMORY_BASE_URL" )
261264 if base_url is None :
262- base_url = f"https://v2. api.supermemory.ai"
265+ base_url = f"https://api.supermemory.ai/ "
263266
264267 super ().__init__ (
265268 version = __version__ ,
@@ -272,10 +275,9 @@ def __init__(
272275 _strict_response_validation = _strict_response_validation ,
273276 )
274277
278+ self .memories = memories .AsyncMemoriesResource (self )
275279 self .settings = settings .AsyncSettingsResource (self )
276- self .memory = memory .AsyncMemoryResource (self )
277- self .search = search .AsyncSearchResource (self )
278- self .connection = connection .AsyncConnectionResource (self )
280+ self .connections = connections .AsyncConnectionsResource (self )
279281 self .with_raw_response = AsyncSupermemoryWithRawResponse (self )
280282 self .with_streaming_response = AsyncSupermemoryWithStreamedResponse (self )
281283
@@ -284,6 +286,12 @@ def __init__(
284286 def qs (self ) -> Querystring :
285287 return Querystring (array_format = "comma" )
286288
289+ @property
290+ @override
291+ def auth_headers (self ) -> dict [str , str ]:
292+ api_key = self .api_key
293+ return {"Authorization" : f"Bearer { api_key } " }
294+
287295 @property
288296 @override
289297 def default_headers (self ) -> dict [str , str | Omit ]:
@@ -380,34 +388,30 @@ def _make_status_error(
380388
381389class SupermemoryWithRawResponse :
382390 def __init__ (self , client : Supermemory ) -> None :
391+ self .memories = memories .MemoriesResourceWithRawResponse (client .memories )
383392 self .settings = settings .SettingsResourceWithRawResponse (client .settings )
384- self .memory = memory .MemoryResourceWithRawResponse (client .memory )
385- self .search = search .SearchResourceWithRawResponse (client .search )
386- self .connection = connection .ConnectionResourceWithRawResponse (client .connection )
393+ self .connections = connections .ConnectionsResourceWithRawResponse (client .connections )
387394
388395
389396class AsyncSupermemoryWithRawResponse :
390397 def __init__ (self , client : AsyncSupermemory ) -> None :
398+ self .memories = memories .AsyncMemoriesResourceWithRawResponse (client .memories )
391399 self .settings = settings .AsyncSettingsResourceWithRawResponse (client .settings )
392- self .memory = memory .AsyncMemoryResourceWithRawResponse (client .memory )
393- self .search = search .AsyncSearchResourceWithRawResponse (client .search )
394- self .connection = connection .AsyncConnectionResourceWithRawResponse (client .connection )
400+ self .connections = connections .AsyncConnectionsResourceWithRawResponse (client .connections )
395401
396402
397403class SupermemoryWithStreamedResponse :
398404 def __init__ (self , client : Supermemory ) -> None :
405+ self .memories = memories .MemoriesResourceWithStreamingResponse (client .memories )
399406 self .settings = settings .SettingsResourceWithStreamingResponse (client .settings )
400- self .memory = memory .MemoryResourceWithStreamingResponse (client .memory )
401- self .search = search .SearchResourceWithStreamingResponse (client .search )
402- self .connection = connection .ConnectionResourceWithStreamingResponse (client .connection )
407+ self .connections = connections .ConnectionsResourceWithStreamingResponse (client .connections )
403408
404409
405410class AsyncSupermemoryWithStreamedResponse :
406411 def __init__ (self , client : AsyncSupermemory ) -> None :
412+ self .memories = memories .AsyncMemoriesResourceWithStreamingResponse (client .memories )
407413 self .settings = settings .AsyncSettingsResourceWithStreamingResponse (client .settings )
408- self .memory = memory .AsyncMemoryResourceWithStreamingResponse (client .memory )
409- self .search = search .AsyncSearchResourceWithStreamingResponse (client .search )
410- self .connection = connection .AsyncConnectionResourceWithStreamingResponse (client .connection )
414+ self .connections = connections .AsyncConnectionsResourceWithStreamingResponse (client .connections )
411415
412416
413417Client = Supermemory
0 commit comments