diff --git a/.stats.yml b/.stats.yml index c2ca6ff0..155f8534 100644 --- a/.stats.yml +++ b/.stats.yml @@ -1,4 +1,4 @@ configured_endpoints: 32 openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/mixedbread%2Fmixedbread-82c2c1c322149cd73b2e8e45f475919b941752a89e74464ccecd1aee9352e9be.yml openapi_spec_hash: dbd7616a32c90fd25b32994830fb12f6 -config_hash: 20a40be2c85d83a4bb34aee53cefd856 +config_hash: f87f729f63f3b34364d1c144753b920d diff --git a/api.md b/api.md index a7d820d8..b1cf6f5a 100644 --- a/api.md +++ b/api.md @@ -1,7 +1,7 @@ # Shared Types ```python -from mixedbread.types import SearchFilter, SearchFilterCondition +from mixedbread.types import SearchFilter, SearchFilterCondition, Usage ``` # Mixedbread diff --git a/src/mixedbread/types/__init__.py b/src/mixedbread/types/__init__.py index ae8ce3a4..5d068c8e 100644 --- a/src/mixedbread/types/__init__.py +++ b/src/mixedbread/types/__init__.py @@ -2,7 +2,7 @@ from __future__ import annotations -from .shared import SearchFilter as SearchFilter, SearchFilterCondition as SearchFilterCondition +from .shared import Usage as Usage, SearchFilter as SearchFilter, SearchFilterCondition as SearchFilterCondition from .embedding import Embedding as Embedding from .file_counts import FileCounts as FileCounts from .file_object import FileObject as FileObject diff --git a/src/mixedbread/types/embedding_create_response.py b/src/mixedbread/types/embedding_create_response.py index 8f3605cb..c7374f21 100644 --- a/src/mixedbread/types/embedding_create_response.py +++ b/src/mixedbread/types/embedding_create_response.py @@ -5,20 +5,10 @@ from .._models import BaseModel from .embedding import Embedding +from .shared.usage import Usage from .multi_encoding_embedding import MultiEncodingEmbedding -__all__ = ["EmbeddingCreateResponse", "Usage"] - - -class Usage(BaseModel): - prompt_tokens: int - """The number of tokens used for the prompt""" - - total_tokens: int - """The total number of tokens used""" - - completion_tokens: Optional[int] = None - """The number of tokens used for the completion""" +__all__ = ["EmbeddingCreateResponse"] class EmbeddingCreateResponse(BaseModel): diff --git a/src/mixedbread/types/rerank_response.py b/src/mixedbread/types/rerank_response.py index 5cf151fc..1b4dd0a9 100644 --- a/src/mixedbread/types/rerank_response.py +++ b/src/mixedbread/types/rerank_response.py @@ -4,19 +4,9 @@ from typing_extensions import Literal from .._models import BaseModel +from .shared.usage import Usage -__all__ = ["RerankResponse", "Usage", "Data"] - - -class Usage(BaseModel): - prompt_tokens: int - """The number of tokens used for the prompt""" - - total_tokens: int - """The total number of tokens used""" - - completion_tokens: Optional[int] = None - """The number of tokens used for the completion""" +__all__ = ["RerankResponse", "Data"] class Data(BaseModel): diff --git a/src/mixedbread/types/shared/__init__.py b/src/mixedbread/types/shared/__init__.py index c91e740d..66d5dcf9 100644 --- a/src/mixedbread/types/shared/__init__.py +++ b/src/mixedbread/types/shared/__init__.py @@ -1,4 +1,5 @@ # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. +from .usage import Usage as Usage from .search_filter import SearchFilter as SearchFilter from .search_filter_condition import SearchFilterCondition as SearchFilterCondition diff --git a/src/mixedbread/types/shared/usage.py b/src/mixedbread/types/shared/usage.py new file mode 100644 index 00000000..eab91a6c --- /dev/null +++ b/src/mixedbread/types/shared/usage.py @@ -0,0 +1,18 @@ +# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details. + +from typing import Optional + +from ..._models import BaseModel + +__all__ = ["Usage"] + + +class Usage(BaseModel): + prompt_tokens: int + """The number of tokens used for the prompt""" + + total_tokens: int + """The total number of tokens used""" + + completion_tokens: Optional[int] = None + """The number of tokens used for the completion"""