supermemory 3.0.0a1__py3-none-any.whl → 3.0.0a2__py3-none-any.whl

supermemory/resources/search.py

@@ -1,296 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing import List
-from typing_extensions import Literal
-
-import httpx
-
-from ..types import search_execute_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import maybe_transform, async_maybe_transform
-from .._compat import cached_property
-from .._resource import SyncAPIResource, AsyncAPIResource
-from .._response import (
-    to_raw_response_wrapper,
-    to_streamed_response_wrapper,
-    async_to_raw_response_wrapper,
-    async_to_streamed_response_wrapper,
-)
-from .._base_client import make_request_options
-from ..types.search_execute_response import SearchExecuteResponse
-
-__all__ = ["SearchResource", "AsyncSearchResource"]
-
-
-class SearchResource(SyncAPIResource):
-    @cached_property
-    def with_raw_response(self) -> SearchResourceWithRawResponse:
-        """
-        This property can be used as a prefix for any HTTP method call to return
-        the raw response object instead of the parsed content.
-
-        For more information, see https://www.github.com/supermemoryai/python-sdk#accessing-raw-response-data-eg-headers
-        """
-        return SearchResourceWithRawResponse(self)
-
-    @cached_property
-    def with_streaming_response(self) -> SearchResourceWithStreamingResponse:
-        """
-        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
-
-        For more information, see https://www.github.com/supermemoryai/python-sdk#with_streaming_response
-        """
-        return SearchResourceWithStreamingResponse(self)
-
-    def execute(
-        self,
-        *,
-        q: str,
-        categories_filter: List[Literal["technology", "science", "business", "health"]] | NotGiven = NOT_GIVEN,
-        chunk_threshold: float | NotGiven = NOT_GIVEN,
-        doc_id: str | NotGiven = NOT_GIVEN,
-        document_threshold: float | NotGiven = NOT_GIVEN,
-        filters: search_execute_params.Filters | NotGiven = NOT_GIVEN,
-        include_summary: bool | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        only_matching_chunks: bool | NotGiven = NOT_GIVEN,
-        rerank: bool | NotGiven = NOT_GIVEN,
-        rewrite_query: bool | NotGiven = NOT_GIVEN,
-        user_id: str | NotGiven = NOT_GIVEN,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SearchExecuteResponse:
-        """
-        Search memories with filtering
-
-        Args:
-          q: Search query string
-
-          categories_filter: Optional category filters
-
-          chunk_threshold: Threshold / sensitivity for chunk selection. 0 is least sensitive (returns most
-              chunks, more results), 1 is most sensitive (returns lesser chunks, accurate
-              results)
-
-          doc_id: Optional document ID to search within. You can use this to find chunks in a very
-              large document.
-
-          document_threshold: Threshold / sensitivity for document selection. 0 is least sensitive (returns
-              most documents, more results), 1 is most sensitive (returns lesser documents,
-              accurate results)
-
-          filters: Optional filters to apply to the search
-
-          include_summary: If true, include document summary in the response. This is helpful if you want a
-              chatbot to know the full context of the document.
-
-          limit: Maximum number of results to return
-
-          only_matching_chunks: If true, only return matching chunks without context. Normally, we send the
-              previous and next chunk to provide more context for LLMs. If you only want the
-              matching chunk, set this to true.
-
-          rerank: If true, rerank the results based on the query. This is helpful if you want to
-              ensure the most relevant results are returned.
-
-          rewrite_query: If true, rewrites the query to make it easier to find documents. This increases
-              the latency by about 400ms
-
-          user_id: End user ID this search is associated with. NOTE: This also acts as a filter for
-              the search.
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return self._get(
-            "/v3/search",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "q": q,
-                        "categories_filter": categories_filter,
-                        "chunk_threshold": chunk_threshold,
-                        "doc_id": doc_id,
-                        "document_threshold": document_threshold,
-                        "filters": filters,
-                        "include_summary": include_summary,
-                        "limit": limit,
-                        "only_matching_chunks": only_matching_chunks,
-                        "rerank": rerank,
-                        "rewrite_query": rewrite_query,
-                        "user_id": user_id,
-                    },
-                    search_execute_params.SearchExecuteParams,
-                ),
-            ),
-            cast_to=SearchExecuteResponse,
-        )
-
-
-class AsyncSearchResource(AsyncAPIResource):
-    @cached_property
-    def with_raw_response(self) -> AsyncSearchResourceWithRawResponse:
-        """
-        This property can be used as a prefix for any HTTP method call to return
-        the raw response object instead of the parsed content.
-
-        For more information, see https://www.github.com/supermemoryai/python-sdk#accessing-raw-response-data-eg-headers
-        """
-        return AsyncSearchResourceWithRawResponse(self)
-
-    @cached_property
-    def with_streaming_response(self) -> AsyncSearchResourceWithStreamingResponse:
-        """
-        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
-
-        For more information, see https://www.github.com/supermemoryai/python-sdk#with_streaming_response
-        """
-        return AsyncSearchResourceWithStreamingResponse(self)
-
-    async def execute(
-        self,
-        *,
-        q: str,
-        categories_filter: List[Literal["technology", "science", "business", "health"]] | NotGiven = NOT_GIVEN,
-        chunk_threshold: float | NotGiven = NOT_GIVEN,
-        doc_id: str | NotGiven = NOT_GIVEN,
-        document_threshold: float | NotGiven = NOT_GIVEN,
-        filters: search_execute_params.Filters | NotGiven = NOT_GIVEN,
-        include_summary: bool | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        only_matching_chunks: bool | NotGiven = NOT_GIVEN,
-        rerank: bool | NotGiven = NOT_GIVEN,
-        rewrite_query: bool | NotGiven = NOT_GIVEN,
-        user_id: str | NotGiven = NOT_GIVEN,
-        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
-        # The extra values given here take precedence over values defined on the client or passed to this method.
-        extra_headers: Headers | None = None,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SearchExecuteResponse:
-        """
-        Search memories with filtering
-
-        Args:
-          q: Search query string
-
-          categories_filter: Optional category filters
-
-          chunk_threshold: Threshold / sensitivity for chunk selection. 0 is least sensitive (returns most
-              chunks, more results), 1 is most sensitive (returns lesser chunks, accurate
-              results)
-
-          doc_id: Optional document ID to search within. You can use this to find chunks in a very
-              large document.
-
-          document_threshold: Threshold / sensitivity for document selection. 0 is least sensitive (returns
-              most documents, more results), 1 is most sensitive (returns lesser documents,
-              accurate results)
-
-          filters: Optional filters to apply to the search
-
-          include_summary: If true, include document summary in the response. This is helpful if you want a
-              chatbot to know the full context of the document.
-
-          limit: Maximum number of results to return
-
-          only_matching_chunks: If true, only return matching chunks without context. Normally, we send the
-              previous and next chunk to provide more context for LLMs. If you only want the
-              matching chunk, set this to true.
-
-          rerank: If true, rerank the results based on the query. This is helpful if you want to
-              ensure the most relevant results are returned.
-
-          rewrite_query: If true, rewrites the query to make it easier to find documents. This increases
-              the latency by about 400ms
-
-          user_id: End user ID this search is associated with. NOTE: This also acts as a filter for
-              the search.
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
-
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        return await self._get(
-            "/v3/search",
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=await async_maybe_transform(
-                    {
-                        "q": q,
-                        "categories_filter": categories_filter,
-                        "chunk_threshold": chunk_threshold,
-                        "doc_id": doc_id,
-                        "document_threshold": document_threshold,
-                        "filters": filters,
-                        "include_summary": include_summary,
-                        "limit": limit,
-                        "only_matching_chunks": only_matching_chunks,
-                        "rerank": rerank,
-                        "rewrite_query": rewrite_query,
-                        "user_id": user_id,
-                    },
-                    search_execute_params.SearchExecuteParams,
-                ),
-            ),
-            cast_to=SearchExecuteResponse,
-        )
-
-
-class SearchResourceWithRawResponse:
-    def __init__(self, search: SearchResource) -> None:
-        self._search = search
-
-        self.execute = to_raw_response_wrapper(
-            search.execute,
-        )
-
-
-class AsyncSearchResourceWithRawResponse:
-    def __init__(self, search: AsyncSearchResource) -> None:
-        self._search = search
-
-        self.execute = async_to_raw_response_wrapper(
-            search.execute,
-        )
-
-
-class SearchResourceWithStreamingResponse:
-    def __init__(self, search: SearchResource) -> None:
-        self._search = search
-
-        self.execute = to_streamed_response_wrapper(
-            search.execute,
-        )
-
-
-class AsyncSearchResourceWithStreamingResponse:
-    def __init__(self, search: AsyncSearchResource) -> None:
-        self._search = search
-
-        self.execute = async_to_streamed_response_wrapper(
-            search.execute,
-        )

supermemory/types/connection_list_params.py

@@ -1,13 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing_extensions import Annotated, TypedDict
-
-from .._utils import PropertyInfo
-
-__all__ = ["ConnectionListParams"]
-
-
-class ConnectionListParams(TypedDict, total=False):
-    end_user_id: Annotated[str, PropertyInfo(alias="endUserId")]

supermemory/types/connection_list_response.py

@@ -1,25 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing import Dict, List, Optional
-from typing_extensions import TypeAlias
-
-from pydantic import Field as FieldInfo
-
-from .._models import BaseModel
-
-__all__ = ["ConnectionListResponse", "ConnectionListResponseItem"]
-
-
-class ConnectionListResponseItem(BaseModel):
-    id: str
-
-    created_at: float = FieldInfo(alias="createdAt")
-
-    provider: str
-
-    expires_at: Optional[float] = FieldInfo(alias="expiresAt", default=None)
-
-    metadata: Optional[Dict[str, object]] = None
-
-
-ConnectionListResponse: TypeAlias = List[ConnectionListResponseItem]

supermemory/types/memory_delete_response.py

@@ -1,9 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from .._models import BaseModel
-
-__all__ = ["MemoryDeleteResponse"]
-
-
-class MemoryDeleteResponse(BaseModel):
-    success: bool

supermemory/types/memory_list_params.py

@@ -1,24 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing_extensions import Literal, TypedDict
-
-__all__ = ["MemoryListParams"]
-
-
-class MemoryListParams(TypedDict, total=False):
-    filters: str
-    """Optional filters to apply to the search"""
-
-    limit: str
-    """Number of items per page"""
-
-    order: Literal["asc", "desc"]
-    """Sort order"""
-
-    page: str
-    """Page number to fetch"""
-
-    sort: Literal["createdAt", "updatedAt"]
-    """Field to sort by"""

supermemory/types/memory_list_response.py

@@ -1,95 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing import Dict, List, Union, Optional
-from datetime import datetime
-from typing_extensions import Literal
-
-from pydantic import Field as FieldInfo
-
-from .._models import BaseModel
-
-__all__ = ["MemoryListResponse", "Memory", "Pagination"]
-
-
-class Memory(BaseModel):
-    id: str
-    """Unique identifier of the memory."""
-
-    content: Optional[str] = None
-    """The content to extract and process into a memory.
-
-    This can be a URL to a website, a PDF, an image, or a video.
-
-    Plaintext: Any plaintext format
-
-    URL: A URL to a website, PDF, image, or video
-
-    We automatically detect the content type from the url's response format.
-    """
-
-    created_at: datetime = FieldInfo(alias="createdAt")
-    """Creation timestamp"""
-
-    custom_id: Optional[str] = FieldInfo(alias="customId", default=None)
-    """Optional custom ID of the memory.
-
-    This could be an ID from your database that will uniquely identify this memory.
-    """
-
-    metadata: Union[str, float, bool, Dict[str, object], List[object], None] = None
-    """Optional metadata for the memory.
-
-    This is used to store additional information about the memory. You can use this
-    to store any additional information you need about the memory. Metadata can be
-    filtered through. Keys must be strings and are case sensitive. Values can be
-    strings, numbers, or booleans. You cannot nest objects.
-    """
-
-    og_image: Optional[str] = FieldInfo(alias="ogImage", default=None)
-
-    source: Optional[str] = None
-
-    status: Literal["unknown", "queued", "extracting", "chunking", "embedding", "indexing", "done", "failed"]
-    """Status of the memory"""
-
-    summary: Optional[str] = None
-    """Summary of the memory content"""
-
-    title: Optional[str] = None
-    """Title of the memory"""
-
-    type: Literal["text", "pdf", "tweet", "google_doc", "image", "video", "notion_doc", "webpage"]
-    """Type of the memory"""
-
-    updated_at: datetime = FieldInfo(alias="updatedAt")
-    """Last update timestamp"""
-
-    url: Optional[str] = None
-    """URL of the memory"""
-
-    container_tags: Optional[List[str]] = FieldInfo(alias="containerTags", default=None)
-    """Optional tags this memory should be containerized by.
-
-    This can be an ID for your user, a project ID, or any other identifier you wish
-    to use to group memories.
-    """
-
-    raw: None = None
-    """Raw content of the memory"""
-
-
-class Pagination(BaseModel):
-    current_page: float = FieldInfo(alias="currentPage")
-
-    limit: float
-
-    total_items: float = FieldInfo(alias="totalItems")
-
-    total_pages: float = FieldInfo(alias="totalPages")
-
-
-class MemoryListResponse(BaseModel):
-    memories: List[Memory]
-
-    pagination: Pagination
-    """Pagination metadata"""

supermemory/types/memory_upload_file_params.py

@@ -1,13 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing_extensions import Required, TypedDict
-
-from .._types import FileTypes
-
-__all__ = ["MemoryUploadFileParams"]
-
-
-class MemoryUploadFileParams(TypedDict, total=False):
-    file: Required[FileTypes]

supermemory/types/memory_upload_file_response.py

@@ -1,11 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from .._models import BaseModel
-
-__all__ = ["MemoryUploadFileResponse"]
-
-
-class MemoryUploadFileResponse(BaseModel):
-    id: str
-
-    status: str

supermemory/types/search_execute_params.py

@@ -1,86 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from __future__ import annotations
-
-from typing import Dict, List, Union, Iterable
-from typing_extensions import Literal, Required, Annotated, TypeAlias, TypedDict
-
-from .._utils import PropertyInfo
-
-__all__ = ["SearchExecuteParams", "Filters", "FiltersUnionMember0"]
-
-
-class SearchExecuteParams(TypedDict, total=False):
-    q: Required[str]
-    """Search query string"""
-
-    categories_filter: Annotated[
-        List[Literal["technology", "science", "business", "health"]], PropertyInfo(alias="categoriesFilter")
-    ]
-    """Optional category filters"""
-
-    chunk_threshold: Annotated[float, PropertyInfo(alias="chunkThreshold")]
-    """Threshold / sensitivity for chunk selection.
-
-    0 is least sensitive (returns most chunks, more results), 1 is most sensitive
-    (returns lesser chunks, accurate results)
-    """
-
-    doc_id: Annotated[str, PropertyInfo(alias="docId")]
-    """Optional document ID to search within.
-
-    You can use this to find chunks in a very large document.
-    """
-
-    document_threshold: Annotated[float, PropertyInfo(alias="documentThreshold")]
-    """Threshold / sensitivity for document selection.
-
-    0 is least sensitive (returns most documents, more results), 1 is most sensitive
-    (returns lesser documents, accurate results)
-    """
-
-    filters: Filters
-    """Optional filters to apply to the search"""
-
-    include_summary: Annotated[bool, PropertyInfo(alias="includeSummary")]
-    """If true, include document summary in the response.
-
-    This is helpful if you want a chatbot to know the full context of the document.
-    """
-
-    limit: int
-    """Maximum number of results to return"""
-
-    only_matching_chunks: Annotated[bool, PropertyInfo(alias="onlyMatchingChunks")]
-    """If true, only return matching chunks without context.
-
-    Normally, we send the previous and next chunk to provide more context for LLMs.
-    If you only want the matching chunk, set this to true.
-    """
-
-    rerank: bool
-    """If true, rerank the results based on the query.
-
-    This is helpful if you want to ensure the most relevant results are returned.
-    """
-
-    rewrite_query: Annotated[bool, PropertyInfo(alias="rewriteQuery")]
-    """If true, rewrites the query to make it easier to find documents.
-
-    This increases the latency by about 400ms
-    """
-
-    user_id: Annotated[str, PropertyInfo(alias="userId")]
-    """End user ID this search is associated with.
-
-    NOTE: This also acts as a filter for the search.
-    """
-
-
-class FiltersUnionMember0(TypedDict, total=False):
-    and_: Annotated[Iterable[object], PropertyInfo(alias="AND")]
-
-    or_: Annotated[Iterable[object], PropertyInfo(alias="OR")]
-
-
-Filters: TypeAlias = Union[FiltersUnionMember0, Dict[str, object]]

supermemory/types/search_execute_response.py

@@ -1,52 +0,0 @@
-# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
-
-from typing import Dict, List, Optional
-from datetime import datetime
-
-from pydantic import Field as FieldInfo
-
-from .._models import BaseModel
-
-__all__ = ["SearchExecuteResponse", "Result", "ResultChunk"]
-
-
-class ResultChunk(BaseModel):
-    content: str
-    """Content of the matching chunk"""
-
-    is_relevant: bool = FieldInfo(alias="isRelevant")
-    """Whether this chunk is relevant to the query"""
-
-    score: float
-    """Similarity score for this chunk"""
-
-
-class Result(BaseModel):
-    chunks: List[ResultChunk]
-    """Matching content chunks from the document"""
-
-    created_at: datetime = FieldInfo(alias="createdAt")
-    """Document creation date"""
-
-    document_id: str = FieldInfo(alias="documentId")
-    """ID of the matching document"""
-
-    metadata: Optional[Dict[str, object]] = None
-    """Document metadata"""
-
-    score: float
-    """Relevance score of the match"""
-
-    title: str
-    """Document title"""
-
-    updated_at: datetime = FieldInfo(alias="updatedAt")
-    """Document last update date"""
-
-
-class SearchExecuteResponse(BaseModel):
-    results: List[Result]
-
-    timing: float
-
-    total: float