diff --git a/.release-please-manifest.json b/.release-please-manifest.json index 6e1a4864..32634a8c 100644 --- a/.release-please-manifest.json +++ b/.release-please-manifest.json @@ -1,3 +1,3 @@ { - ".": "3.0.0-alpha.21" + ".": "3.0.0-alpha.22" } \ No newline at end of file diff --git a/.stats.yml b/.stats.yml index 3667d419..b801420b 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,4 +1,4 @@ configured_endpoints: 16 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/supermemory--inc%2Fsupermemory-new-1a2a84a9cc99c25a9d726bc9300a72871f9469e3ceacebd5e71fd37e87891318.yml openapi_spec_hash: e71e86a645bc47a86664080c8697b8db -config_hash: b560219f71fa815fec30fe25ca5a71f5 +config_hash: be10c837d5319a33f30809a3ec223caf diff --git a/CHANGELOG.md b/CHANGELOG.md index ea4431f9..c995c2c9 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -1,5 +1,13 @@ # Changelog +## 3.0.0-alpha.22 (2025-07-03) + +Full Changelog: [v3.0.0-alpha.21...v3.0.0-alpha.22](https://github.com/supermemoryai/python-sdk/compare/v3.0.0-alpha.21...v3.0.0-alpha.22) + +### Features + +* **api:** manual updates ([2a863a1](https://github.com/supermemoryai/python-sdk/commit/2a863a166b5c39208ef910d84530a27898ed0c71)) + ## 3.0.0-alpha.21 (2025-07-03) Full Changelog: [v3.0.0-alpha.20...v3.0.0-alpha.21](https://github.com/supermemoryai/python-sdk/compare/v3.0.0-alpha.20...v3.0.0-alpha.21) diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 583ed499..0f988c37 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -36,7 +36,7 @@ $ pip install -r requirements-dev.lock Most of the SDK is generated code. Modifications to code will be persisted between generations, but may result in merge conflicts between manual patches and changes from the generator. The generator will never -modify the contents of the `src/supermemory_new/lib/` and `examples/` directories. +modify the contents of the `src/supermemory/lib/` and `examples/` directories. ## Adding and running examples diff --git a/README.md b/README.md index 609596e8..c78d9389 100644 --- a/README.md +++ b/README.md @@ -25,7 +25,7 @@ The full API of this library can be found in [api.md](api.md). ```python import os -from supermemory_new import Supermemory +from supermemory import Supermemory client = Supermemory( api_key=os.environ.get("SUPERMEMORY_API_KEY"), # This is the default and can be omitted @@ -49,7 +49,7 @@ Simply import `AsyncSupermemory` instead of `Supermemory` and use `await` with e ```python import os import asyncio -from supermemory_new import AsyncSupermemory +from supermemory import AsyncSupermemory client = AsyncSupermemory( api_key=os.environ.get("SUPERMEMORY_API_KEY"), # This is the default and can be omitted @@ -84,8 +84,8 @@ Then you can enable it by instantiating the client with `http_client=DefaultAioH ```python import os import asyncio -from supermemory_new import DefaultAioHttpClient -from supermemory_new import AsyncSupermemory +from supermemory import DefaultAioHttpClient +from supermemory import AsyncSupermemory async def main() -> None: @@ -113,16 +113,16 @@ Typed requests and responses provide autocomplete and documentation within your ## Handling errors -When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `supermemory_new.APIConnectionError` is raised. +When the library is unable to connect to the API (for example, due to network connection problems or a timeout), a subclass of `supermemory.APIConnectionError` is raised. When the API returns a non-success status code (that is, 4xx or 5xx -response), a subclass of `supermemory_new.APIStatusError` is raised, containing `status_code` and `response` properties. +response), a subclass of `supermemory.APIStatusError` is raised, containing `status_code` and `response` properties. -All errors inherit from `supermemory_new.APIError`. +All errors inherit from `supermemory.APIError`. ```python -import supermemory_new -from supermemory_new import Supermemory +import supermemory +from supermemory import Supermemory client = Supermemory() @@ -130,12 +130,12 @@ try: client.memories.add( content="This is a detailed article about machine learning concepts...", ) -except supermemory_new.APIConnectionError as e: +except supermemory.APIConnectionError as e: print("The server could not be reached") print(e.__cause__) # an underlying Exception, likely raised within httpx. -except supermemory_new.RateLimitError as e: +except supermemory.RateLimitError as e: print("A 429 status code was received; we should back off a bit.") -except supermemory_new.APIStatusError as e: +except supermemory.APIStatusError as e: print("Another non-200-range status code was received") print(e.status_code) print(e.response) @@ -163,7 +163,7 @@ Connection errors (for example, due to a network connectivity problem), 408 Requ You can use the `max_retries` option to configure or disable retry settings: ```python -from supermemory_new import Supermemory +from supermemory import Supermemory # Configure the default for all requests: client = Supermemory( @@ -183,7 +183,7 @@ By default requests time out after 1 minute. You can configure this with a `time which accepts a float or an [`httpx.Timeout`](https://www.python-httpx.org/advanced/timeouts/#fine-tuning-the-configuration) object: ```python -from supermemory_new import Supermemory +from supermemory import Supermemory # Configure the default for all requests: client = Supermemory( @@ -237,7 +237,7 @@ if response.my_field is None: The "raw" Response object can be accessed by prefixing `.with_raw_response.` to any HTTP method call, e.g., ```py -from supermemory_new import Supermemory +from supermemory import Supermemory client = Supermemory() response = client.memories.with_raw_response.add( @@ -249,9 +249,9 @@ memory = response.parse() # get the object that `memories.add()` would have ret print(memory.id) ``` -These methods return an [`APIResponse`](https://github.com/supermemoryai/python-sdk/tree/main/src/supermemory_new/_response.py) object. +These methods return an [`APIResponse`](https://github.com/supermemoryai/python-sdk/tree/main/src/supermemory/_response.py) object. -The async client returns an [`AsyncAPIResponse`](https://github.com/supermemoryai/python-sdk/tree/main/src/supermemory_new/_response.py) with the same structure, the only difference being `await`able methods for reading the response content. +The async client returns an [`AsyncAPIResponse`](https://github.com/supermemoryai/python-sdk/tree/main/src/supermemory/_response.py) with the same structure, the only difference being `await`able methods for reading the response content. #### `.with_streaming_response` @@ -315,7 +315,7 @@ You can directly override the [httpx client](https://www.python-httpx.org/api/#c ```python import httpx -from supermemory_new import Supermemory, DefaultHttpxClient +from supermemory import Supermemory, DefaultHttpxClient client = Supermemory( # Or use the `SUPERMEMORY_BASE_URL` env var @@ -338,7 +338,7 @@ client.with_options(http_client=DefaultHttpxClient(...)) By default the library closes underlying HTTP connections whenever the client is [garbage collected](https://docs.python.org/3/reference/datamodel.html#object.__del__). You can manually close the client using the `.close()` method if desired, or with a context manager that closes when exiting. ```py -from supermemory_new import Supermemory +from supermemory import Supermemory with Supermemory() as client: # make requests here @@ -366,8 +366,8 @@ If you've upgraded to the latest version but aren't seeing any new features you You can determine the version that is being used at runtime with: ```py -import supermemory_new -print(supermemory_new.__version__) +import supermemory +print(supermemory.__version__) ``` ## Requirements diff --git a/api.md b/api.md index d7034d2e..b20d9d9a 100644 --- a/api.md +++ b/api.md @@ -3,7 +3,7 @@ Types: ```python -from supermemory_new.types import ( +from supermemory.types import ( MemoryUpdateResponse, MemoryListResponse, MemoryAddResponse, @@ -13,43 +13,43 @@ from supermemory_new.types import ( Methods: -- client.memories.update(id, \*\*params) -> MemoryUpdateResponse -- client.memories.list(\*\*params) -> MemoryListResponse -- client.memories.delete(id) -> None -- client.memories.add(\*\*params) -> MemoryAddResponse -- client.memories.get(id) -> MemoryGetResponse +- client.memories.update(id, \*\*params) -> MemoryUpdateResponse +- client.memories.list(\*\*params) -> MemoryListResponse +- client.memories.delete(id) -> None +- client.memories.add(\*\*params) -> MemoryAddResponse +- client.memories.get(id) -> MemoryGetResponse # Search Types: ```python -from supermemory_new.types import SearchExecuteResponse +from supermemory.types import SearchExecuteResponse ``` Methods: -- client.search.execute(\*\*params) -> SearchExecuteResponse +- client.search.execute(\*\*params) -> SearchExecuteResponse # Settings Types: ```python -from supermemory_new.types import SettingUpdateResponse, SettingGetResponse +from supermemory.types import SettingUpdateResponse, SettingGetResponse ``` Methods: -- client.settings.update(\*\*params) -> SettingUpdateResponse -- client.settings.get() -> SettingGetResponse +- client.settings.update(\*\*params) -> SettingUpdateResponse +- client.settings.get() -> SettingGetResponse # Connections Types: ```python -from supermemory_new.types import ( +from supermemory.types import ( ConnectionCreateResponse, ConnectionListResponse, ConnectionDeleteByIDResponse, @@ -62,11 +62,11 @@ from supermemory_new.types import ( Methods: -- client.connections.create(provider, \*\*params) -> ConnectionCreateResponse -- client.connections.list(\*\*params) -> ConnectionListResponse -- client.connections.delete_by_id(connection_id) -> ConnectionDeleteByIDResponse -- client.connections.delete_by_provider(provider, \*\*params) -> ConnectionDeleteByProviderResponse -- client.connections.get_by_id(connection_id) -> ConnectionGetByIDResponse -- client.connections.get_by_tags(provider, \*\*params) -> ConnectionGetByTagsResponse -- client.connections.import\_(provider, \*\*params) -> None -- client.connections.list_documents(provider, \*\*params) -> ConnectionListDocumentsResponse +- client.connections.create(provider, \*\*params) -> ConnectionCreateResponse +- client.connections.list(\*\*params) -> ConnectionListResponse +- client.connections.delete_by_id(connection_id) -> ConnectionDeleteByIDResponse +- client.connections.delete_by_provider(provider, \*\*params) -> ConnectionDeleteByProviderResponse +- client.connections.get_by_id(connection_id) -> ConnectionGetByIDResponse +- client.connections.get_by_tags(provider, \*\*params) -> ConnectionGetByTagsResponse +- client.connections.import\_(provider, \*\*params) -> None +- client.connections.list_documents(provider, \*\*params) -> ConnectionListDocumentsResponse diff --git a/mypy.ini b/mypy.ini index b9a84617..08c8f486 100644 --- a/mypy.ini +++ b/mypy.ini @@ -8,7 +8,7 @@ show_error_codes = True # # We also exclude our `tests` as mypy doesn't always infer # types correctly and Pyright will still catch any type errors. -exclude = ^(src/supermemory_new/_files\.py|_dev/.*\.py|tests/.*)$ +exclude = ^(src/supermemory/_files\.py|_dev/.*\.py|tests/.*)$ strict_equality = True implicit_reexport = True diff --git a/pyproject.toml b/pyproject.toml index 4318d690..a33b5b82 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = "supermemory" -version = "3.0.0-alpha.21" +version = "3.0.0-alpha.22" description = "The official Python library for the supermemory API" dynamic = ["readme"] license = "Apache-2.0" @@ -78,14 +78,14 @@ format = { chain = [ "check:ruff" = "ruff check ." "fix:ruff" = "ruff check --fix ." -"check:importable" = "python -c 'import supermemory_new'" +"check:importable" = "python -c 'import supermemory'" typecheck = { chain = [ "typecheck:pyright", "typecheck:mypy" ]} "typecheck:pyright" = "pyright" -"typecheck:verify-types" = "pyright --verifytypes supermemory_new --ignoreexternal" +"typecheck:verify-types" = "pyright --verifytypes supermemory --ignoreexternal" "typecheck:mypy" = "mypy ." [build-system] @@ -98,7 +98,7 @@ include = [ ] [tool.hatch.build.targets.wheel] -packages = ["src/supermemory_new"] +packages = ["src/supermemory"] [tool.hatch.build.targets.sdist] # Basically everything except hidden files/directories (such as .github, .devcontainers, .python-version, etc) @@ -201,7 +201,7 @@ length-sort = true length-sort-straight = true combine-as-imports = true extra-standard-library = ["typing_extensions"] -known-first-party = ["supermemory_new", "tests"] +known-first-party = ["supermemory", "tests"] [tool.ruff.lint.per-file-ignores] "bin/**.py" = ["T201", "T203"] diff --git a/release-please-config.json b/release-please-config.json index e2aa4047..247d1fda 100644 --- a/release-please-config.json +++ b/release-please-config.json @@ -61,6 +61,6 @@ ], "release-type": "python", "extra-files": [ - "src/supermemory_new/_version.py" + "src/supermemory/_version.py" ] } \ No newline at end of file diff --git a/scripts/lint b/scripts/lint index 64dfb350..d4303570 100755 --- a/scripts/lint +++ b/scripts/lint @@ -8,4 +8,4 @@ echo "==> Running lints" rye run lint echo "==> Making sure it imports" -rye run python -c 'import supermemory_new' +rye run python -c 'import supermemory' diff --git a/src/supermemory_new/__init__.py b/src/supermemory/__init__.py similarity index 94% rename from src/supermemory_new/__init__.py rename to src/supermemory/__init__.py index 64e2798a..bb87a0cf 100644 --- a/src/supermemory_new/__init__.py +++ b/src/supermemory/__init__.py @@ -89,12 +89,12 @@ # Update the __module__ attribute for exported symbols so that # error messages point to this module instead of the module # it was originally defined in, e.g. -# supermemory_new._exceptions.NotFoundError -> supermemory_new.NotFoundError +# supermemory._exceptions.NotFoundError -> supermemory.NotFoundError __locals = locals() for __name in __all__: if not __name.startswith("__"): try: - __locals[__name].__module__ = "supermemory_new" + __locals[__name].__module__ = "supermemory" except (TypeError, AttributeError): # Some of our exported symbols are builtins which we can't set attributes for. pass diff --git a/src/supermemory_new/_base_client.py b/src/supermemory/_base_client.py similarity index 99% rename from src/supermemory_new/_base_client.py rename to src/supermemory/_base_client.py index d15f975b..9a46a403 100644 --- a/src/supermemory_new/_base_client.py +++ b/src/supermemory/_base_client.py @@ -389,7 +389,7 @@ def __init__( if max_retries is None: # pyright: ignore[reportUnnecessaryComparison] raise TypeError( - "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `supermemory_new.DEFAULT_MAX_RETRIES`" + "max_retries cannot be None. If you want to disable retries, pass `0`; if you want unlimited retries, pass `math.inf` or a very high number; if you want the default behavior, pass `supermemory.DEFAULT_MAX_RETRIES`" ) def _enforce_trailing_slash(self, url: URL) -> URL: diff --git a/src/supermemory_new/_client.py b/src/supermemory/_client.py similarity index 100% rename from src/supermemory_new/_client.py rename to src/supermemory/_client.py diff --git a/src/supermemory_new/_compat.py b/src/supermemory/_compat.py similarity index 100% rename from src/supermemory_new/_compat.py rename to src/supermemory/_compat.py diff --git a/src/supermemory_new/_constants.py b/src/supermemory/_constants.py similarity index 100% rename from src/supermemory_new/_constants.py rename to src/supermemory/_constants.py diff --git a/src/supermemory_new/_exceptions.py b/src/supermemory/_exceptions.py similarity index 100% rename from src/supermemory_new/_exceptions.py rename to src/supermemory/_exceptions.py diff --git a/src/supermemory_new/_files.py b/src/supermemory/_files.py similarity index 100% rename from src/supermemory_new/_files.py rename to src/supermemory/_files.py diff --git a/src/supermemory_new/_models.py b/src/supermemory/_models.py similarity index 100% rename from src/supermemory_new/_models.py rename to src/supermemory/_models.py diff --git a/src/supermemory_new/_qs.py b/src/supermemory/_qs.py similarity index 100% rename from src/supermemory_new/_qs.py rename to src/supermemory/_qs.py diff --git a/src/supermemory_new/_resource.py b/src/supermemory/_resource.py similarity index 100% rename from src/supermemory_new/_resource.py rename to src/supermemory/_resource.py diff --git a/src/supermemory_new/_response.py b/src/supermemory/_response.py similarity index 99% rename from src/supermemory_new/_response.py rename to src/supermemory/_response.py index c3381986..9a6b956f 100644 --- a/src/supermemory_new/_response.py +++ b/src/supermemory/_response.py @@ -218,7 +218,7 @@ def _parse(self, *, to: type[_T] | None = None) -> R | _T: and issubclass(origin, pydantic.BaseModel) ): raise TypeError( - "Pydantic models must subclass our base model type, e.g. `from supermemory_new import BaseModel`" + "Pydantic models must subclass our base model type, e.g. `from supermemory import BaseModel`" ) if ( @@ -285,7 +285,7 @@ def parse(self, *, to: type[_T] | None = None) -> R | _T: the `to` argument, e.g. ```py - from supermemory_new import BaseModel + from supermemory import BaseModel class MyModel(BaseModel): @@ -387,7 +387,7 @@ async def parse(self, *, to: type[_T] | None = None) -> R | _T: the `to` argument, e.g. ```py - from supermemory_new import BaseModel + from supermemory import BaseModel class MyModel(BaseModel): @@ -558,7 +558,7 @@ async def stream_to_file( class MissingStreamClassError(TypeError): def __init__(self) -> None: super().__init__( - "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `supermemory_new._streaming` for reference", + "The `stream` argument was set to `True` but the `stream_cls` argument was not given. See `supermemory._streaming` for reference", ) diff --git a/src/supermemory_new/_streaming.py b/src/supermemory/_streaming.py similarity index 100% rename from src/supermemory_new/_streaming.py rename to src/supermemory/_streaming.py diff --git a/src/supermemory_new/_types.py b/src/supermemory/_types.py similarity index 99% rename from src/supermemory_new/_types.py rename to src/supermemory/_types.py index f63365fc..8491d513 100644 --- a/src/supermemory_new/_types.py +++ b/src/supermemory/_types.py @@ -81,7 +81,7 @@ # This unfortunately means that you will either have # to import this type and pass it explicitly: # -# from supermemory_new import NoneType +# from supermemory import NoneType # client.get('/foo', cast_to=NoneType) # # or build it yourself: diff --git a/src/supermemory_new/_utils/__init__.py b/src/supermemory/_utils/__init__.py similarity index 100% rename from src/supermemory_new/_utils/__init__.py rename to src/supermemory/_utils/__init__.py diff --git a/src/supermemory_new/_utils/_logs.py b/src/supermemory/_utils/_logs.py similarity index 75% rename from src/supermemory_new/_utils/_logs.py rename to src/supermemory/_utils/_logs.py index 219dd778..b4d8cbf3 100644 --- a/src/supermemory_new/_utils/_logs.py +++ b/src/supermemory/_utils/_logs.py @@ -1,12 +1,12 @@ import os import logging -logger: logging.Logger = logging.getLogger("supermemory_new") +logger: logging.Logger = logging.getLogger("supermemory") httpx_logger: logging.Logger = logging.getLogger("httpx") def _basic_config() -> None: - # e.g. [2023-10-05 14:12:26 - supermemory_new._base_client:818 - DEBUG] HTTP Request: POST http://127.0.0.1:4010/foo/bar "200 OK" + # e.g. [2023-10-05 14:12:26 - supermemory._base_client:818 - DEBUG] HTTP Request: POST http://127.0.0.1:4010/foo/bar "200 OK" logging.basicConfig( format="[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)s", datefmt="%Y-%m-%d %H:%M:%S", diff --git a/src/supermemory_new/_utils/_proxy.py b/src/supermemory/_utils/_proxy.py similarity index 100% rename from src/supermemory_new/_utils/_proxy.py rename to src/supermemory/_utils/_proxy.py diff --git a/src/supermemory_new/_utils/_reflection.py b/src/supermemory/_utils/_reflection.py similarity index 100% rename from src/supermemory_new/_utils/_reflection.py rename to src/supermemory/_utils/_reflection.py diff --git a/src/supermemory_new/_utils/_resources_proxy.py b/src/supermemory/_utils/_resources_proxy.py similarity index 50% rename from src/supermemory_new/_utils/_resources_proxy.py rename to src/supermemory/_utils/_resources_proxy.py index 5186de41..53b457d1 100644 --- a/src/supermemory_new/_utils/_resources_proxy.py +++ b/src/supermemory/_utils/_resources_proxy.py @@ -7,17 +7,17 @@ class ResourcesProxy(LazyProxy[Any]): - """A proxy for the `supermemory_new.resources` module. + """A proxy for the `supermemory.resources` module. - This is used so that we can lazily import `supermemory_new.resources` only when - needed *and* so that users can just import `supermemory_new` and reference `supermemory_new.resources` + This is used so that we can lazily import `supermemory.resources` only when + needed *and* so that users can just import `supermemory` and reference `supermemory.resources` """ @override def __load__(self) -> Any: import importlib - mod = importlib.import_module("supermemory_new.resources") + mod = importlib.import_module("supermemory.resources") return mod diff --git a/src/supermemory_new/_utils/_streams.py b/src/supermemory/_utils/_streams.py similarity index 100% rename from src/supermemory_new/_utils/_streams.py rename to src/supermemory/_utils/_streams.py diff --git a/src/supermemory_new/_utils/_sync.py b/src/supermemory/_utils/_sync.py similarity index 100% rename from src/supermemory_new/_utils/_sync.py rename to src/supermemory/_utils/_sync.py diff --git a/src/supermemory_new/_utils/_transform.py b/src/supermemory/_utils/_transform.py similarity index 100% rename from src/supermemory_new/_utils/_transform.py rename to src/supermemory/_utils/_transform.py diff --git a/src/supermemory_new/_utils/_typing.py b/src/supermemory/_utils/_typing.py similarity index 100% rename from src/supermemory_new/_utils/_typing.py rename to src/supermemory/_utils/_typing.py diff --git a/src/supermemory_new/_utils/_utils.py b/src/supermemory/_utils/_utils.py similarity index 100% rename from src/supermemory_new/_utils/_utils.py rename to src/supermemory/_utils/_utils.py diff --git a/src/supermemory/_version.py b/src/supermemory/_version.py new file mode 100644 index 00000000..fe9b6240 --- /dev/null +++ b/src/supermemory/_version.py @@ -0,0 +1,4 @@ +# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. + +__title__ = "supermemory" +__version__ = "3.0.0-alpha.22" # x-release-please-version diff --git a/src/supermemory/lib/.keep b/src/supermemory/lib/.keep new file mode 100644 index 00000000..5e2c99fd --- /dev/null +++ b/src/supermemory/lib/.keep @@ -0,0 +1,4 @@ +File generated from our OpenAPI spec by Stainless. + +This directory can be used to store custom files to expand the SDK. +It is ignored by Stainless code generation and its content (other than this keep file) won't be touched. \ No newline at end of file diff --git a/src/supermemory_new/py.typed b/src/supermemory/py.typed similarity index 100% rename from src/supermemory_new/py.typed rename to src/supermemory/py.typed diff --git a/src/supermemory_new/resources/__init__.py b/src/supermemory/resources/__init__.py similarity index 100% rename from src/supermemory_new/resources/__init__.py rename to src/supermemory/resources/__init__.py diff --git a/src/supermemory_new/resources/connections.py b/src/supermemory/resources/connections.py similarity index 100% rename from src/supermemory_new/resources/connections.py rename to src/supermemory/resources/connections.py diff --git a/src/supermemory_new/resources/memories.py b/src/supermemory/resources/memories.py similarity index 100% rename from src/supermemory_new/resources/memories.py rename to src/supermemory/resources/memories.py diff --git a/src/supermemory_new/resources/search.py b/src/supermemory/resources/search.py similarity index 100% rename from src/supermemory_new/resources/search.py rename to src/supermemory/resources/search.py diff --git a/src/supermemory_new/resources/settings.py b/src/supermemory/resources/settings.py similarity index 100% rename from src/supermemory_new/resources/settings.py rename to src/supermemory/resources/settings.py diff --git a/src/supermemory_new/types/__init__.py b/src/supermemory/types/__init__.py similarity index 100% rename from src/supermemory_new/types/__init__.py rename to src/supermemory/types/__init__.py diff --git a/src/supermemory_new/types/connection_create_params.py b/src/supermemory/types/connection_create_params.py similarity index 100% rename from src/supermemory_new/types/connection_create_params.py rename to src/supermemory/types/connection_create_params.py diff --git a/src/supermemory_new/types/connection_create_response.py b/src/supermemory/types/connection_create_response.py similarity index 100% rename from src/supermemory_new/types/connection_create_response.py rename to src/supermemory/types/connection_create_response.py diff --git a/src/supermemory_new/types/connection_delete_by_id_response.py b/src/supermemory/types/connection_delete_by_id_response.py similarity index 100% rename from src/supermemory_new/types/connection_delete_by_id_response.py rename to src/supermemory/types/connection_delete_by_id_response.py diff --git a/src/supermemory_new/types/connection_delete_by_provider_params.py b/src/supermemory/types/connection_delete_by_provider_params.py similarity index 100% rename from src/supermemory_new/types/connection_delete_by_provider_params.py rename to src/supermemory/types/connection_delete_by_provider_params.py diff --git a/src/supermemory_new/types/connection_delete_by_provider_response.py b/src/supermemory/types/connection_delete_by_provider_response.py similarity index 100% rename from src/supermemory_new/types/connection_delete_by_provider_response.py rename to src/supermemory/types/connection_delete_by_provider_response.py diff --git a/src/supermemory_new/types/connection_get_by_id_response.py b/src/supermemory/types/connection_get_by_id_response.py similarity index 100% rename from src/supermemory_new/types/connection_get_by_id_response.py rename to src/supermemory/types/connection_get_by_id_response.py diff --git a/src/supermemory_new/types/connection_get_by_tags_params.py b/src/supermemory/types/connection_get_by_tags_params.py similarity index 100% rename from src/supermemory_new/types/connection_get_by_tags_params.py rename to src/supermemory/types/connection_get_by_tags_params.py diff --git a/src/supermemory_new/types/connection_get_by_tags_response.py b/src/supermemory/types/connection_get_by_tags_response.py similarity index 100% rename from src/supermemory_new/types/connection_get_by_tags_response.py rename to src/supermemory/types/connection_get_by_tags_response.py diff --git a/src/supermemory_new/types/connection_import_params.py b/src/supermemory/types/connection_import_params.py similarity index 100% rename from src/supermemory_new/types/connection_import_params.py rename to src/supermemory/types/connection_import_params.py diff --git a/src/supermemory_new/types/connection_list_documents_params.py b/src/supermemory/types/connection_list_documents_params.py similarity index 100% rename from src/supermemory_new/types/connection_list_documents_params.py rename to src/supermemory/types/connection_list_documents_params.py diff --git a/src/supermemory_new/types/connection_list_documents_response.py b/src/supermemory/types/connection_list_documents_response.py similarity index 100% rename from src/supermemory_new/types/connection_list_documents_response.py rename to src/supermemory/types/connection_list_documents_response.py diff --git a/src/supermemory_new/types/connection_list_params.py b/src/supermemory/types/connection_list_params.py similarity index 100% rename from src/supermemory_new/types/connection_list_params.py rename to src/supermemory/types/connection_list_params.py diff --git a/src/supermemory_new/types/connection_list_response.py b/src/supermemory/types/connection_list_response.py similarity index 100% rename from src/supermemory_new/types/connection_list_response.py rename to src/supermemory/types/connection_list_response.py diff --git a/src/supermemory_new/types/memory_add_params.py b/src/supermemory/types/memory_add_params.py similarity index 100% rename from src/supermemory_new/types/memory_add_params.py rename to src/supermemory/types/memory_add_params.py diff --git a/src/supermemory_new/types/memory_add_response.py b/src/supermemory/types/memory_add_response.py similarity index 100% rename from src/supermemory_new/types/memory_add_response.py rename to src/supermemory/types/memory_add_response.py diff --git a/src/supermemory_new/types/memory_get_response.py b/src/supermemory/types/memory_get_response.py similarity index 100% rename from src/supermemory_new/types/memory_get_response.py rename to src/supermemory/types/memory_get_response.py diff --git a/src/supermemory_new/types/memory_list_params.py b/src/supermemory/types/memory_list_params.py similarity index 100% rename from src/supermemory_new/types/memory_list_params.py rename to src/supermemory/types/memory_list_params.py diff --git a/src/supermemory_new/types/memory_list_response.py b/src/supermemory/types/memory_list_response.py similarity index 100% rename from src/supermemory_new/types/memory_list_response.py rename to src/supermemory/types/memory_list_response.py diff --git a/src/supermemory_new/types/memory_update_params.py b/src/supermemory/types/memory_update_params.py similarity index 100% rename from src/supermemory_new/types/memory_update_params.py rename to src/supermemory/types/memory_update_params.py diff --git a/src/supermemory_new/types/memory_update_response.py b/src/supermemory/types/memory_update_response.py similarity index 100% rename from src/supermemory_new/types/memory_update_response.py rename to src/supermemory/types/memory_update_response.py diff --git a/src/supermemory_new/types/search_execute_params.py b/src/supermemory/types/search_execute_params.py similarity index 100% rename from src/supermemory_new/types/search_execute_params.py rename to src/supermemory/types/search_execute_params.py diff --git a/src/supermemory_new/types/search_execute_response.py b/src/supermemory/types/search_execute_response.py similarity index 100% rename from src/supermemory_new/types/search_execute_response.py rename to src/supermemory/types/search_execute_response.py diff --git a/src/supermemory_new/types/setting_get_response.py b/src/supermemory/types/setting_get_response.py similarity index 100% rename from src/supermemory_new/types/setting_get_response.py rename to src/supermemory/types/setting_get_response.py diff --git a/src/supermemory_new/types/setting_update_params.py b/src/supermemory/types/setting_update_params.py similarity index 100% rename from src/supermemory_new/types/setting_update_params.py rename to src/supermemory/types/setting_update_params.py diff --git a/src/supermemory_new/types/setting_update_response.py b/src/supermemory/types/setting_update_response.py similarity index 100% rename from src/supermemory_new/types/setting_update_response.py rename to src/supermemory/types/setting_update_response.py diff --git a/src/supermemory_new/_version.py b/src/supermemory_new/_version.py deleted file mode 100644 index 48f3cfd5..00000000 --- a/src/supermemory_new/_version.py +++ /dev/null @@ -1,4 +0,0 @@ -# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. - -__title__ = "supermemory_new" -__version__ = "3.0.0-alpha.21" # x-release-please-version diff --git a/tests/api_resources/test_connections.py b/tests/api_resources/test_connections.py index a220bcb4..f9f8a871 100644 --- a/tests/api_resources/test_connections.py +++ b/tests/api_resources/test_connections.py @@ -7,9 +7,9 @@ import pytest +from supermemory import Supermemory, AsyncSupermemory from tests.utils import assert_matches_type -from supermemory_new import Supermemory, AsyncSupermemory -from supermemory_new.types import ( +from supermemory.types import ( ConnectionListResponse, ConnectionCreateResponse, ConnectionGetByIDResponse, diff --git a/tests/api_resources/test_memories.py b/tests/api_resources/test_memories.py index bbe21739..a522fa8a 100644 --- a/tests/api_resources/test_memories.py +++ b/tests/api_resources/test_memories.py @@ -7,9 +7,9 @@ import pytest +from supermemory import Supermemory, AsyncSupermemory from tests.utils import assert_matches_type -from supermemory_new import Supermemory, AsyncSupermemory -from supermemory_new.types import ( +from supermemory.types import ( MemoryAddResponse, MemoryGetResponse, MemoryListResponse, diff --git a/tests/api_resources/test_search.py b/tests/api_resources/test_search.py index 5aec7f5b..78674d7a 100644 --- a/tests/api_resources/test_search.py +++ b/tests/api_resources/test_search.py @@ -7,9 +7,9 @@ import pytest +from supermemory import Supermemory, AsyncSupermemory from tests.utils import assert_matches_type -from supermemory_new import Supermemory, AsyncSupermemory -from supermemory_new.types import SearchExecuteResponse +from supermemory.types import SearchExecuteResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/api_resources/test_settings.py b/tests/api_resources/test_settings.py index de333dc0..b2fb17fe 100644 --- a/tests/api_resources/test_settings.py +++ b/tests/api_resources/test_settings.py @@ -7,9 +7,9 @@ import pytest +from supermemory import Supermemory, AsyncSupermemory from tests.utils import assert_matches_type -from supermemory_new import Supermemory, AsyncSupermemory -from supermemory_new.types import SettingGetResponse, SettingUpdateResponse +from supermemory.types import SettingGetResponse, SettingUpdateResponse base_url = os.environ.get("TEST_API_BASE_URL", "http://127.0.0.1:4010") diff --git a/tests/conftest.py b/tests/conftest.py index 475e33e2..2baaab81 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -10,15 +10,15 @@ import pytest from pytest_asyncio import is_async_test -from supermemory_new import Supermemory, AsyncSupermemory, DefaultAioHttpClient -from supermemory_new._utils import is_dict +from supermemory import Supermemory, AsyncSupermemory, DefaultAioHttpClient +from supermemory._utils import is_dict if TYPE_CHECKING: from _pytest.fixtures import FixtureRequest # pyright: ignore[reportPrivateImportUsage] pytest.register_assert_rewrite("tests.utils") -logging.getLogger("supermemory_new").setLevel(logging.DEBUG) +logging.getLogger("supermemory").setLevel(logging.DEBUG) # automatically add `pytest.mark.asyncio()` to all of our async tests diff --git a/tests/test_client.py b/tests/test_client.py index 22628172..ecc086ec 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -21,11 +21,11 @@ from respx import MockRouter from pydantic import ValidationError -from supermemory_new import Supermemory, AsyncSupermemory, APIResponseValidationError -from supermemory_new._types import Omit -from supermemory_new._models import BaseModel, FinalRequestOptions -from supermemory_new._exceptions import APIStatusError, APITimeoutError, SupermemoryError, APIResponseValidationError -from supermemory_new._base_client import ( +from supermemory import Supermemory, AsyncSupermemory, APIResponseValidationError +from supermemory._types import Omit +from supermemory._models import BaseModel, FinalRequestOptions +from supermemory._exceptions import APIStatusError, APITimeoutError, SupermemoryError, APIResponseValidationError +from supermemory._base_client import ( DEFAULT_TIMEOUT, HTTPX_DEFAULT_TIMEOUT, BaseClient, @@ -232,10 +232,10 @@ def add_leak(leaks: list[tracemalloc.StatisticDiff], diff: tracemalloc.Statistic # to_raw_response_wrapper leaks through the @functools.wraps() decorator. # # removing the decorator fixes the leak for reasons we don't understand. - "supermemory_new/_legacy_response.py", - "supermemory_new/_response.py", + "supermemory/_legacy_response.py", + "supermemory/_response.py", # pydantic.BaseModel.model_dump || pydantic.BaseModel.dict leak memory for some reason. - "supermemory_new/_compat.py", + "supermemory/_compat.py", # Standard library leaks we don't care about. "/logging/__init__.py", ] @@ -721,7 +721,7 @@ def test_parse_retry_after_header(self, remaining_retries: int, retry_after: str calculated = client._calculate_retry_timeout(remaining_retries, options, headers) assert calculated == pytest.approx(timeout, 0.5 * 0.875) # pyright: ignore[reportUnknownMemberType] - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, client: Supermemory) -> None: respx_mock.post("/v3/memories").mock(side_effect=httpx.TimeoutException("Test timeout error")) @@ -733,7 +733,7 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter, clien assert _get_open_connections(self.client) == 0 - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client: Supermemory) -> None: respx_mock.post("/v3/memories").mock(return_value=httpx.Response(500)) @@ -745,7 +745,7 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter, client assert _get_open_connections(self.client) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.parametrize("failure_mode", ["status", "exception"]) def test_retries_taken( @@ -778,7 +778,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert int(response.http_request.headers.get("x-stainless-retry-count")) == failures_before_success @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_omit_retry_count_header( self, client: Supermemory, failures_before_success: int, respx_mock: MockRouter @@ -804,7 +804,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert len(response.http_request.headers.get_list("x-stainless-retry-count")) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) def test_overwrite_retry_count_header( self, client: Supermemory, failures_before_success: int, respx_mock: MockRouter @@ -1055,10 +1055,10 @@ def add_leak(leaks: list[tracemalloc.StatisticDiff], diff: tracemalloc.Statistic # to_raw_response_wrapper leaks through the @functools.wraps() decorator. # # removing the decorator fixes the leak for reasons we don't understand. - "supermemory_new/_legacy_response.py", - "supermemory_new/_response.py", + "supermemory/_legacy_response.py", + "supermemory/_response.py", # pydantic.BaseModel.model_dump || pydantic.BaseModel.dict leak memory for some reason. - "supermemory_new/_compat.py", + "supermemory/_compat.py", # Standard library leaks we don't care about. "/logging/__init__.py", ] @@ -1548,7 +1548,7 @@ async def test_parse_retry_after_header(self, remaining_retries: int, retry_afte calculated = client._calculate_retry_timeout(remaining_retries, options, headers) assert calculated == pytest.approx(timeout, 0.5 * 0.875) # pyright: ignore[reportUnknownMemberType] - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) async def test_retrying_timeout_errors_doesnt_leak( self, respx_mock: MockRouter, async_client: AsyncSupermemory @@ -1562,7 +1562,7 @@ async def test_retrying_timeout_errors_doesnt_leak( assert _get_open_connections(self.client) == 0 - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) async def test_retrying_status_errors_doesnt_leak( self, respx_mock: MockRouter, async_client: AsyncSupermemory @@ -1576,7 +1576,7 @@ async def test_retrying_status_errors_doesnt_leak( assert _get_open_connections(self.client) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio @pytest.mark.parametrize("failure_mode", ["status", "exception"]) @@ -1610,7 +1610,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert int(response.http_request.headers.get("x-stainless-retry-count")) == failures_before_success @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio async def test_omit_retry_count_header( @@ -1637,7 +1637,7 @@ def retry_handler(_request: httpx.Request) -> httpx.Response: assert len(response.http_request.headers.get_list("x-stainless-retry-count")) == 0 @pytest.mark.parametrize("failures_before_success", [0, 2, 4]) - @mock.patch("supermemory_new._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) + @mock.patch("supermemory._base_client.BaseClient._calculate_retry_timeout", _low_retry_timeout) @pytest.mark.respx(base_url=base_url) @pytest.mark.asyncio async def test_overwrite_retry_count_header( @@ -1674,8 +1674,8 @@ def test_get_platform(self) -> None: import nest_asyncio import threading - from supermemory_new._utils import asyncify - from supermemory_new._base_client import get_platform + from supermemory._utils import asyncify + from supermemory._base_client import get_platform async def test_main() -> None: result = await asyncify(get_platform)() diff --git a/tests/test_deepcopy.py b/tests/test_deepcopy.py index 5db64d01..c464e633 100644 --- a/tests/test_deepcopy.py +++ b/tests/test_deepcopy.py @@ -1,4 +1,4 @@ -from supermemory_new._utils import deepcopy_minimal +from supermemory._utils import deepcopy_minimal def assert_different_identities(obj1: object, obj2: object) -> None: diff --git a/tests/test_extract_files.py b/tests/test_extract_files.py index 68166331..b5940c71 100644 --- a/tests/test_extract_files.py +++ b/tests/test_extract_files.py @@ -4,8 +4,8 @@ import pytest -from supermemory_new._types import FileTypes -from supermemory_new._utils import extract_files +from supermemory._types import FileTypes +from supermemory._utils import extract_files def test_removes_files_from_input() -> None: diff --git a/tests/test_files.py b/tests/test_files.py index bd750637..fe8388cc 100644 --- a/tests/test_files.py +++ b/tests/test_files.py @@ -4,7 +4,7 @@ import pytest from dirty_equals import IsDict, IsList, IsBytes, IsTuple -from supermemory_new._files import to_httpx_files, async_to_httpx_files +from supermemory._files import to_httpx_files, async_to_httpx_files readme_path = Path(__file__).parent.parent.joinpath("README.md") diff --git a/tests/test_models.py b/tests/test_models.py index da648259..4e795b1e 100644 --- a/tests/test_models.py +++ b/tests/test_models.py @@ -7,9 +7,9 @@ import pydantic from pydantic import Field -from supermemory_new._utils import PropertyInfo -from supermemory_new._compat import PYDANTIC_V2, parse_obj, model_dump, model_json -from supermemory_new._models import BaseModel, construct_type +from supermemory._utils import PropertyInfo +from supermemory._compat import PYDANTIC_V2, parse_obj, model_dump, model_json +from supermemory._models import BaseModel, construct_type class BasicModel(BaseModel): diff --git a/tests/test_qs.py b/tests/test_qs.py index cfee5c95..8b0e27bb 100644 --- a/tests/test_qs.py +++ b/tests/test_qs.py @@ -4,7 +4,7 @@ import pytest -from supermemory_new._qs import Querystring, stringify +from supermemory._qs import Querystring, stringify def test_empty() -> None: diff --git a/tests/test_required_args.py b/tests/test_required_args.py index 12634af4..d886edee 100644 --- a/tests/test_required_args.py +++ b/tests/test_required_args.py @@ -2,7 +2,7 @@ import pytest -from supermemory_new._utils import required_args +from supermemory._utils import required_args def test_too_many_positional_params() -> None: diff --git a/tests/test_response.py b/tests/test_response.py index c6d78148..829b3c8b 100644 --- a/tests/test_response.py +++ b/tests/test_response.py @@ -6,8 +6,8 @@ import pytest import pydantic -from supermemory_new import BaseModel, Supermemory, AsyncSupermemory -from supermemory_new._response import ( +from supermemory import BaseModel, Supermemory, AsyncSupermemory +from supermemory._response import ( APIResponse, BaseAPIResponse, AsyncAPIResponse, @@ -15,8 +15,8 @@ AsyncBinaryAPIResponse, extract_response_type, ) -from supermemory_new._streaming import Stream -from supermemory_new._base_client import FinalRequestOptions +from supermemory._streaming import Stream +from supermemory._base_client import FinalRequestOptions class ConcreteBaseAPIResponse(APIResponse[bytes]): ... @@ -37,7 +37,7 @@ def test_extract_response_type_direct_classes() -> None: def test_extract_response_type_direct_class_missing_type_arg() -> None: with pytest.raises( RuntimeError, - match="Expected type to have a type argument at index 0 but it did not", + match="Expected type to have a type argument at index 0 but it did not", ): extract_response_type(AsyncAPIResponse) @@ -68,7 +68,7 @@ def test_response_parse_mismatched_basemodel(client: Supermemory) -> None: with pytest.raises( TypeError, - match="Pydantic models must subclass our base model type, e.g. `from supermemory_new import BaseModel`", + match="Pydantic models must subclass our base model type, e.g. `from supermemory import BaseModel`", ): response.parse(to=PydanticModel) @@ -86,7 +86,7 @@ async def test_async_response_parse_mismatched_basemodel(async_client: AsyncSupe with pytest.raises( TypeError, - match="Pydantic models must subclass our base model type, e.g. `from supermemory_new import BaseModel`", + match="Pydantic models must subclass our base model type, e.g. `from supermemory import BaseModel`", ): await response.parse(to=PydanticModel) diff --git a/tests/test_streaming.py b/tests/test_streaming.py index ac61ea11..ee2d2c5f 100644 --- a/tests/test_streaming.py +++ b/tests/test_streaming.py @@ -5,8 +5,8 @@ import httpx import pytest -from supermemory_new import Supermemory, AsyncSupermemory -from supermemory_new._streaming import Stream, AsyncStream, ServerSentEvent +from supermemory import Supermemory, AsyncSupermemory +from supermemory._streaming import Stream, AsyncStream, ServerSentEvent @pytest.mark.asyncio diff --git a/tests/test_transform.py b/tests/test_transform.py index 3bac94c1..8a3a412e 100644 --- a/tests/test_transform.py +++ b/tests/test_transform.py @@ -8,15 +8,15 @@ import pytest -from supermemory_new._types import NOT_GIVEN, Base64FileInput -from supermemory_new._utils import ( +from supermemory._types import NOT_GIVEN, Base64FileInput +from supermemory._utils import ( PropertyInfo, transform as _transform, parse_datetime, async_transform as _async_transform, ) -from supermemory_new._compat import PYDANTIC_V2 -from supermemory_new._models import BaseModel +from supermemory._compat import PYDANTIC_V2 +from supermemory._models import BaseModel _T = TypeVar("_T") diff --git a/tests/test_utils/test_proxy.py b/tests/test_utils/test_proxy.py index 0a17be19..74440350 100644 --- a/tests/test_utils/test_proxy.py +++ b/tests/test_utils/test_proxy.py @@ -2,7 +2,7 @@ from typing import Any from typing_extensions import override -from supermemory_new._utils import LazyProxy +from supermemory._utils import LazyProxy class RecursiveLazyProxy(LazyProxy[Any]): diff --git a/tests/test_utils/test_typing.py b/tests/test_utils/test_typing.py index f5da46f3..5e9a15ec 100644 --- a/tests/test_utils/test_typing.py +++ b/tests/test_utils/test_typing.py @@ -2,7 +2,7 @@ from typing import Generic, TypeVar, cast -from supermemory_new._utils import extract_type_var_from_base +from supermemory._utils import extract_type_var_from_base _T = TypeVar("_T") _T2 = TypeVar("_T2") diff --git a/tests/utils.py b/tests/utils.py index 9795d49f..0d6779c4 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -8,8 +8,8 @@ from datetime import date, datetime from typing_extensions import Literal, get_args, get_origin, assert_type -from supermemory_new._types import Omit, NoneType -from supermemory_new._utils import ( +from supermemory._types import Omit, NoneType +from supermemory._utils import ( is_dict, is_list, is_list_type, @@ -18,8 +18,8 @@ is_annotated_type, is_type_alias_type, ) -from supermemory_new._compat import PYDANTIC_V2, field_outer_type, get_model_fields -from supermemory_new._models import BaseModel +from supermemory._compat import PYDANTIC_V2, field_outer_type, get_model_fields +from supermemory._models import BaseModel BaseModelT = TypeVar("BaseModelT", bound=BaseModel)