Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .stats.yml
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
configured_endpoints: 33
configured_endpoints: 24
openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/mixedbread%2Fmixedbread-43906e7132098c54e8b761fa3bd67fbd9174894faeb22367a681e92f60f6445f.yml
77 changes: 0 additions & 77 deletions api.md
Original file line number Diff line number Diff line change
Expand Up @@ -16,30 +16,6 @@ Methods:

- <code title="get /">client.<a href="./src/mixedbread/_client.py">info</a>() -> <a href="./src/mixedbread/types/info_response.py">InfoResponse</a></code>

# Embeddings

Types:

```python
from mixedbread.types import Embedding, EmbeddingCreateResponse
```

Methods:

- <code title="post /v1/embeddings">client.embeddings.<a href="./src/mixedbread/resources/embeddings.py">create</a>(\*\*<a href="src/mixedbread/types/embedding_create_params.py">params</a>) -> <a href="./src/mixedbread/types/embedding_create_response.py">EmbeddingCreateResponse</a></code>

# Reranking

Types:

```python
from mixedbread.types import RerankingCreateResponse
```

Methods:

- <code title="post /v1/reranking">client.reranking.<a href="./src/mixedbread/resources/reranking.py">create</a>(\*\*<a href="src/mixedbread/types/reranking_create_params.py">params</a>) -> <a href="./src/mixedbread/types/reranking_create_response.py">RerankingCreateResponse</a></code>

# Parsing

## Jobs
Expand Down Expand Up @@ -122,56 +98,3 @@ Methods:
- <code title="get /v1/vector_stores/{vector_store_id}/files">client.vector_stores.files.<a href="./src/mixedbread/resources/vector_stores/files.py">list</a>(vector_store_id, \*\*<a href="src/mixedbread/types/vector_stores/file_list_params.py">params</a>) -> <a href="./src/mixedbread/types/vector_stores/vector_store_file.py">SyncLimitOffset[VectorStoreFile]</a></code>
- <code title="delete /v1/vector_stores/{vector_store_id}/files/{file_id}">client.vector_stores.files.<a href="./src/mixedbread/resources/vector_stores/files.py">delete</a>(file_id, \*, vector_store_id) -> <a href="./src/mixedbread/types/vector_stores/file_delete_response.py">FileDeleteResponse</a></code>
- <code title="post /v1/vector_stores/files/search">client.vector_stores.files.<a href="./src/mixedbread/resources/vector_stores/files.py">search</a>(\*\*<a href="src/mixedbread/types/vector_stores/file_search_params.py">params</a>) -> <a href="./src/mixedbread/types/vector_stores/file_search_response.py">FileSearchResponse</a></code>

# Completions

Types:

```python
from mixedbread.types import CompletionCreateResponse
```

Methods:

- <code title="post /v1/chat/completions">client.completions.<a href="./src/mixedbread/resources/completions.py">create</a>() -> <a href="./src/mixedbread/types/completion_create_response.py">object</a></code>

# Extractions

## Jobs

Types:

```python
from mixedbread.types.extractions import ExtractionJob
```

Methods:

- <code title="post /v1/extractions/jobs">client.extractions.jobs.<a href="./src/mixedbread/resources/extractions/jobs.py">create</a>(\*\*<a href="src/mixedbread/types/extractions/job_create_params.py">params</a>) -> <a href="./src/mixedbread/types/extractions/extraction_job.py">ExtractionJob</a></code>
- <code title="get /v1/extractions/jobs/{job_id}">client.extractions.jobs.<a href="./src/mixedbread/resources/extractions/jobs.py">retrieve</a>(job_id) -> <a href="./src/mixedbread/types/extractions/extraction_job.py">ExtractionJob</a></code>

## Schema

Types:

```python
from mixedbread.types.extractions import CreatedJsonSchema, EnhancedJsonSchema, ValidatedJsonSchema
```

Methods:

- <code title="post /v1/extractions/schema">client.extractions.schema.<a href="./src/mixedbread/resources/extractions/schema.py">create</a>(\*\*<a href="src/mixedbread/types/extractions/schema_create_params.py">params</a>) -> <a href="./src/mixedbread/types/extractions/created_json_schema.py">CreatedJsonSchema</a></code>
- <code title="post /v1/extractions/schema/enhance">client.extractions.schema.<a href="./src/mixedbread/resources/extractions/schema.py">enhance</a>(\*\*<a href="src/mixedbread/types/extractions/schema_enhance_params.py">params</a>) -> <a href="./src/mixedbread/types/extractions/enhanced_json_schema.py">EnhancedJsonSchema</a></code>
- <code title="post /v1/extractions/schema/validate">client.extractions.schema.<a href="./src/mixedbread/resources/extractions/schema.py">validate</a>(\*\*<a href="src/mixedbread/types/extractions/schema_validate_params.py">params</a>) -> <a href="./src/mixedbread/types/extractions/validated_json_schema.py">ValidatedJsonSchema</a></code>

## Content

Types:

```python
from mixedbread.types.extractions import ExtractionResult
```

Methods:

- <code title="post /v1/extractions/content">client.extractions.content.<a href="./src/mixedbread/resources/extractions/content.py">create</a>(\*\*<a href="src/mixedbread/types/extractions/content_create_params.py">params</a>) -> <a href="./src/mixedbread/types/extractions/extraction_result.py">ExtractionResult</a></code>
35 changes: 1 addition & 34 deletions src/mixedbread/_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
async_to_raw_response_wrapper,
async_to_streamed_response_wrapper,
)
from .resources import files, reranking, embeddings, completions
from .resources import files
from ._streaming import Stream as Stream, AsyncStream as AsyncStream
from ._exceptions import APIStatusError, MixedbreadError
from ._base_client import (
Expand All @@ -44,7 +44,6 @@
)
from .resources.parsing import parsing
from .types.info_response import InfoResponse
from .resources.extractions import extractions
from .resources.vector_stores import vector_stores

__all__ = [
Expand All @@ -66,13 +65,9 @@


class Mixedbread(SyncAPIClient):
embeddings: embeddings.EmbeddingsResource
reranking: reranking.RerankingResource
parsing: parsing.ParsingResource
files: files.FilesResource
vector_stores: vector_stores.VectorStoresResource
completions: completions.CompletionsResource
extractions: extractions.ExtractionsResource
with_raw_response: MixedbreadWithRawResponse
with_streaming_response: MixedbreadWithStreamedResponse

Expand Down Expand Up @@ -154,13 +149,9 @@ def __init__(
_strict_response_validation=_strict_response_validation,
)

self.embeddings = embeddings.EmbeddingsResource(self)
self.reranking = reranking.RerankingResource(self)
self.parsing = parsing.ParsingResource(self)
self.files = files.FilesResource(self)
self.vector_stores = vector_stores.VectorStoresResource(self)
self.completions = completions.CompletionsResource(self)
self.extractions = extractions.ExtractionsResource(self)
self.with_raw_response = MixedbreadWithRawResponse(self)
self.with_streaming_response = MixedbreadWithStreamedResponse(self)

Expand Down Expand Up @@ -295,13 +286,9 @@ def _make_status_error(


class AsyncMixedbread(AsyncAPIClient):
embeddings: embeddings.AsyncEmbeddingsResource
reranking: reranking.AsyncRerankingResource
parsing: parsing.AsyncParsingResource
files: files.AsyncFilesResource
vector_stores: vector_stores.AsyncVectorStoresResource
completions: completions.AsyncCompletionsResource
extractions: extractions.AsyncExtractionsResource
with_raw_response: AsyncMixedbreadWithRawResponse
with_streaming_response: AsyncMixedbreadWithStreamedResponse

Expand Down Expand Up @@ -383,13 +370,9 @@ def __init__(
_strict_response_validation=_strict_response_validation,
)

self.embeddings = embeddings.AsyncEmbeddingsResource(self)
self.reranking = reranking.AsyncRerankingResource(self)
self.parsing = parsing.AsyncParsingResource(self)
self.files = files.AsyncFilesResource(self)
self.vector_stores = vector_stores.AsyncVectorStoresResource(self)
self.completions = completions.AsyncCompletionsResource(self)
self.extractions = extractions.AsyncExtractionsResource(self)
self.with_raw_response = AsyncMixedbreadWithRawResponse(self)
self.with_streaming_response = AsyncMixedbreadWithStreamedResponse(self)

Expand Down Expand Up @@ -525,13 +508,9 @@ def _make_status_error(

class MixedbreadWithRawResponse:
def __init__(self, client: Mixedbread) -> None:
self.embeddings = embeddings.EmbeddingsResourceWithRawResponse(client.embeddings)
self.reranking = reranking.RerankingResourceWithRawResponse(client.reranking)
self.parsing = parsing.ParsingResourceWithRawResponse(client.parsing)
self.files = files.FilesResourceWithRawResponse(client.files)
self.vector_stores = vector_stores.VectorStoresResourceWithRawResponse(client.vector_stores)
self.completions = completions.CompletionsResourceWithRawResponse(client.completions)
self.extractions = extractions.ExtractionsResourceWithRawResponse(client.extractions)

self.info = to_raw_response_wrapper(
client.info,
Expand All @@ -540,13 +519,9 @@ def __init__(self, client: Mixedbread) -> None:

class AsyncMixedbreadWithRawResponse:
def __init__(self, client: AsyncMixedbread) -> None:
self.embeddings = embeddings.AsyncEmbeddingsResourceWithRawResponse(client.embeddings)
self.reranking = reranking.AsyncRerankingResourceWithRawResponse(client.reranking)
self.parsing = parsing.AsyncParsingResourceWithRawResponse(client.parsing)
self.files = files.AsyncFilesResourceWithRawResponse(client.files)
self.vector_stores = vector_stores.AsyncVectorStoresResourceWithRawResponse(client.vector_stores)
self.completions = completions.AsyncCompletionsResourceWithRawResponse(client.completions)
self.extractions = extractions.AsyncExtractionsResourceWithRawResponse(client.extractions)

self.info = async_to_raw_response_wrapper(
client.info,
Expand All @@ -555,13 +530,9 @@ def __init__(self, client: AsyncMixedbread) -> None:

class MixedbreadWithStreamedResponse:
def __init__(self, client: Mixedbread) -> None:
self.embeddings = embeddings.EmbeddingsResourceWithStreamingResponse(client.embeddings)
self.reranking = reranking.RerankingResourceWithStreamingResponse(client.reranking)
self.parsing = parsing.ParsingResourceWithStreamingResponse(client.parsing)
self.files = files.FilesResourceWithStreamingResponse(client.files)
self.vector_stores = vector_stores.VectorStoresResourceWithStreamingResponse(client.vector_stores)
self.completions = completions.CompletionsResourceWithStreamingResponse(client.completions)
self.extractions = extractions.ExtractionsResourceWithStreamingResponse(client.extractions)

self.info = to_streamed_response_wrapper(
client.info,
Expand All @@ -570,13 +541,9 @@ def __init__(self, client: Mixedbread) -> None:

class AsyncMixedbreadWithStreamedResponse:
def __init__(self, client: AsyncMixedbread) -> None:
self.embeddings = embeddings.AsyncEmbeddingsResourceWithStreamingResponse(client.embeddings)
self.reranking = reranking.AsyncRerankingResourceWithStreamingResponse(client.reranking)
self.parsing = parsing.AsyncParsingResourceWithStreamingResponse(client.parsing)
self.files = files.AsyncFilesResourceWithStreamingResponse(client.files)
self.vector_stores = vector_stores.AsyncVectorStoresResourceWithStreamingResponse(client.vector_stores)
self.completions = completions.AsyncCompletionsResourceWithStreamingResponse(client.completions)
self.extractions = extractions.AsyncExtractionsResourceWithStreamingResponse(client.extractions)

self.info = async_to_streamed_response_wrapper(
client.info,
Expand Down
56 changes: 0 additions & 56 deletions src/mixedbread/resources/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,38 +16,6 @@
ParsingResourceWithStreamingResponse,
AsyncParsingResourceWithStreamingResponse,
)
from .reranking import (
RerankingResource,
AsyncRerankingResource,
RerankingResourceWithRawResponse,
AsyncRerankingResourceWithRawResponse,
RerankingResourceWithStreamingResponse,
AsyncRerankingResourceWithStreamingResponse,
)
from .embeddings import (
EmbeddingsResource,
AsyncEmbeddingsResource,
EmbeddingsResourceWithRawResponse,
AsyncEmbeddingsResourceWithRawResponse,
EmbeddingsResourceWithStreamingResponse,
AsyncEmbeddingsResourceWithStreamingResponse,
)
from .completions import (
CompletionsResource,
AsyncCompletionsResource,
CompletionsResourceWithRawResponse,
AsyncCompletionsResourceWithRawResponse,
CompletionsResourceWithStreamingResponse,
AsyncCompletionsResourceWithStreamingResponse,
)
from .extractions import (
ExtractionsResource,
AsyncExtractionsResource,
ExtractionsResourceWithRawResponse,
AsyncExtractionsResourceWithRawResponse,
ExtractionsResourceWithStreamingResponse,
AsyncExtractionsResourceWithStreamingResponse,
)
from .vector_stores import (
VectorStoresResource,
AsyncVectorStoresResource,
Expand All @@ -58,18 +26,6 @@
)

__all__ = [
"EmbeddingsResource",
"AsyncEmbeddingsResource",
"EmbeddingsResourceWithRawResponse",
"AsyncEmbeddingsResourceWithRawResponse",
"EmbeddingsResourceWithStreamingResponse",
"AsyncEmbeddingsResourceWithStreamingResponse",
"RerankingResource",
"AsyncRerankingResource",
"RerankingResourceWithRawResponse",
"AsyncRerankingResourceWithRawResponse",
"RerankingResourceWithStreamingResponse",
"AsyncRerankingResourceWithStreamingResponse",
"ParsingResource",
"AsyncParsingResource",
"ParsingResourceWithRawResponse",
Expand All @@ -88,16 +44,4 @@
"AsyncVectorStoresResourceWithRawResponse",
"VectorStoresResourceWithStreamingResponse",
"AsyncVectorStoresResourceWithStreamingResponse",
"CompletionsResource",
"AsyncCompletionsResource",
"CompletionsResourceWithRawResponse",
"AsyncCompletionsResourceWithRawResponse",
"CompletionsResourceWithStreamingResponse",
"AsyncCompletionsResourceWithStreamingResponse",
"ExtractionsResource",
"AsyncExtractionsResource",
"ExtractionsResourceWithRawResponse",
"AsyncExtractionsResourceWithRawResponse",
"ExtractionsResourceWithStreamingResponse",
"AsyncExtractionsResourceWithStreamingResponse",
]
Loading