Skip to content

Commit 8eb53ac

Browse files
feat(api): api update
1 parent 9194990 commit 8eb53ac

File tree

6 files changed

+27
-18
lines changed

6 files changed

+27
-18
lines changed

.stats.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
configured_endpoints: 18
2-
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/supermemory--inc%2Fsupermemory-new-bc38a818a02d2611e24859c4fd59ce98a24499af22ac9d16fc93a3d1e0b3c94f.yml
3-
openapi_spec_hash: e25323c2c0f792f2fafc664748774e46
2+
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/supermemory--inc%2Fsupermemory-new-b95fdde5f2544bce843df7bb502b9205217caaf71a28ed7f784a3d481d13d874.yml
3+
openapi_spec_hash: 39d70b150e54ccd103e6063506521ace
44
config_hash: 9b9291a6c872b063900a46386729ba3c

api.md

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,7 @@ from supermemory.types import (
6060
ConnectionDeleteByProviderResponse,
6161
ConnectionGetByIDResponse,
6262
ConnectionGetByTagsResponse,
63+
ConnectionImportResponse,
6364
ConnectionListDocumentsResponse,
6465
)
6566
```
@@ -72,5 +73,5 @@ Methods:
7273
- <code title="delete /v3/connections/{provider}">client.connections.<a href="./src/supermemory/resources/connections.py">delete_by_provider</a>(provider, \*\*<a href="src/supermemory/types/connection_delete_by_provider_params.py">params</a>) -> <a href="./src/supermemory/types/connection_delete_by_provider_response.py">ConnectionDeleteByProviderResponse</a></code>
7374
- <code title="get /v3/connections/{connectionId}">client.connections.<a href="./src/supermemory/resources/connections.py">get_by_id</a>(connection_id) -> <a href="./src/supermemory/types/connection_get_by_id_response.py">ConnectionGetByIDResponse</a></code>
7475
- <code title="post /v3/connections/{provider}/connection">client.connections.<a href="./src/supermemory/resources/connections.py">get_by_tags</a>(provider, \*\*<a href="src/supermemory/types/connection_get_by_tags_params.py">params</a>) -> <a href="./src/supermemory/types/connection_get_by_tags_response.py">ConnectionGetByTagsResponse</a></code>
75-
- <code title="post /v3/connections/{provider}/import">client.connections.<a href="./src/supermemory/resources/connections.py">import\_</a>(provider, \*\*<a href="src/supermemory/types/connection_import_params.py">params</a>) -> None</code>
76+
- <code title="post /v3/connections/{provider}/import">client.connections.<a href="./src/supermemory/resources/connections.py">import\_</a>(provider, \*\*<a href="src/supermemory/types/connection_import_params.py">params</a>) -> str</code>
7677
- <code title="post /v3/connections/{provider}/documents">client.connections.<a href="./src/supermemory/resources/connections.py">list_documents</a>(provider, \*\*<a href="src/supermemory/types/connection_list_documents_params.py">params</a>) -> <a href="./src/supermemory/types/connection_list_documents_response.py">ConnectionListDocumentsResponse</a></code>

src/supermemory/resources/connections.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
connection_list_documents_params,
1616
connection_delete_by_provider_params,
1717
)
18-
from .._types import NOT_GIVEN, Body, Query, Headers, NoneType, NotGiven
18+
from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
1919
from .._utils import maybe_transform, async_maybe_transform
2020
from .._compat import cached_property
2121
from .._resource import SyncAPIResource, AsyncAPIResource
@@ -293,7 +293,7 @@ def import_(
293293
extra_query: Query | None = None,
294294
extra_body: Body | None = None,
295295
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
296-
) -> None:
296+
) -> str:
297297
"""
298298
Initiate a manual sync of connections
299299
@@ -310,14 +310,14 @@ def import_(
310310
"""
311311
if not provider:
312312
raise ValueError(f"Expected a non-empty value for `provider` but received {provider!r}")
313-
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
313+
extra_headers = {"Accept": "text/plain", **(extra_headers or {})}
314314
return self._post(
315315
f"/v3/connections/{provider}/import",
316316
body=maybe_transform({"container_tags": container_tags}, connection_import_params.ConnectionImportParams),
317317
options=make_request_options(
318318
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
319319
),
320-
cast_to=NoneType,
320+
cast_to=str,
321321
)
322322

323323
def list_documents(
@@ -618,7 +618,7 @@ async def import_(
618618
extra_query: Query | None = None,
619619
extra_body: Body | None = None,
620620
timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
621-
) -> None:
621+
) -> str:
622622
"""
623623
Initiate a manual sync of connections
624624
@@ -635,7 +635,7 @@ async def import_(
635635
"""
636636
if not provider:
637637
raise ValueError(f"Expected a non-empty value for `provider` but received {provider!r}")
638-
extra_headers = {"Accept": "*/*", **(extra_headers or {})}
638+
extra_headers = {"Accept": "text/plain", **(extra_headers or {})}
639639
return await self._post(
640640
f"/v3/connections/{provider}/import",
641641
body=await async_maybe_transform(
@@ -644,7 +644,7 @@ async def import_(
644644
options=make_request_options(
645645
extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
646646
),
647-
cast_to=NoneType,
647+
cast_to=str,
648648
)
649649

650650
async def list_documents(

src/supermemory/types/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@
2424
from .memory_upload_file_params import MemoryUploadFileParams as MemoryUploadFileParams
2525
from .search_documents_response import SearchDocumentsResponse as SearchDocumentsResponse
2626
from .connection_create_response import ConnectionCreateResponse as ConnectionCreateResponse
27+
from .connection_import_response import ConnectionImportResponse as ConnectionImportResponse
2728
from .memory_upload_file_response import MemoryUploadFileResponse as MemoryUploadFileResponse
2829
from .connection_get_by_id_response import ConnectionGetByIDResponse as ConnectionGetByIDResponse
2930
from .connection_get_by_tags_params import ConnectionGetByTagsParams as ConnectionGetByTagsParams
Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
2+
3+
from typing_extensions import TypeAlias
4+
5+
__all__ = ["ConnectionImportResponse"]
6+
7+
ConnectionImportResponse: TypeAlias = str

tests/api_resources/test_connections.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -271,7 +271,7 @@ def test_method_import(self, client: Supermemory) -> None:
271271
connection = client.connections.import_(
272272
provider="notion",
273273
)
274-
assert connection is None
274+
assert_matches_type(str, connection, path=["response"])
275275

276276
@pytest.mark.skip(reason="Prism tests are disabled")
277277
@parametrize
@@ -280,7 +280,7 @@ def test_method_import_with_all_params(self, client: Supermemory) -> None:
280280
provider="notion",
281281
container_tags=["user_123", "project_123"],
282282
)
283-
assert connection is None
283+
assert_matches_type(str, connection, path=["response"])
284284

285285
@pytest.mark.skip(reason="Prism tests are disabled")
286286
@parametrize
@@ -292,7 +292,7 @@ def test_raw_response_import(self, client: Supermemory) -> None:
292292
assert response.is_closed is True
293293
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
294294
connection = response.parse()
295-
assert connection is None
295+
assert_matches_type(str, connection, path=["response"])
296296

297297
@pytest.mark.skip(reason="Prism tests are disabled")
298298
@parametrize
@@ -304,7 +304,7 @@ def test_streaming_response_import(self, client: Supermemory) -> None:
304304
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
305305

306306
connection = response.parse()
307-
assert connection is None
307+
assert_matches_type(str, connection, path=["response"])
308308

309309
assert cast(Any, response.is_closed) is True
310310

@@ -603,7 +603,7 @@ async def test_method_import(self, async_client: AsyncSupermemory) -> None:
603603
connection = await async_client.connections.import_(
604604
provider="notion",
605605
)
606-
assert connection is None
606+
assert_matches_type(str, connection, path=["response"])
607607

608608
@pytest.mark.skip(reason="Prism tests are disabled")
609609
@parametrize
@@ -612,7 +612,7 @@ async def test_method_import_with_all_params(self, async_client: AsyncSupermemor
612612
provider="notion",
613613
container_tags=["user_123", "project_123"],
614614
)
615-
assert connection is None
615+
assert_matches_type(str, connection, path=["response"])
616616

617617
@pytest.mark.skip(reason="Prism tests are disabled")
618618
@parametrize
@@ -624,7 +624,7 @@ async def test_raw_response_import(self, async_client: AsyncSupermemory) -> None
624624
assert response.is_closed is True
625625
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
626626
connection = await response.parse()
627-
assert connection is None
627+
assert_matches_type(str, connection, path=["response"])
628628

629629
@pytest.mark.skip(reason="Prism tests are disabled")
630630
@parametrize
@@ -636,7 +636,7 @@ async def test_streaming_response_import(self, async_client: AsyncSupermemory) -
636636
assert response.http_request.headers.get("X-Stainless-Lang") == "python"
637637

638638
connection = await response.parse()
639-
assert connection is None
639+
assert_matches_type(str, connection, path=["response"])
640640

641641
assert cast(Any, response.is_closed) is True
642642

0 commit comments

Comments
 (0)