Skip to content

Commit c7b2ea5

Browse files
docs: sync Core Integrations API reference (fastembed) on Docusaurus (#12194)
Co-authored-by: julian-risch <4181769+julian-risch@users.noreply.github.com>
1 parent efc8bd1 commit c7b2ea5

16 files changed

Lines changed: 128 additions & 32 deletions

File tree

docs-website/reference/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.18/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.19/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.20/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.21/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.22/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

docs-website/reference_versioned_docs/version-2.23/integrations-api/fastembed.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ __init__(
7171
local_files_only: bool = False,
7272
meta_fields_to_embed: list[str] | None = None,
7373
embedding_separator: str = "\n",
74+
model_kwargs: dict[str, Any] | None = None,
7475
) -> None
7576
```
7677

@@ -94,6 +95,8 @@ Create an FastembedDocumentEmbedder component.
9495
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
9596
- **meta_fields_to_embed** (<code>list\[str\] | None</code>) – List of meta fields that should be embedded along with the Document content.
9697
- **embedding_separator** (<code>str</code>) – Separator used to concatenate the meta fields to the Document content.
98+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
99+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
97100

98101
#### to_dict
99102

@@ -198,7 +201,7 @@ __init__(
198201
) -> None
199202
```
200203

201-
Create an FastembedDocumentEmbedder component.
204+
Create a FastembedSparseDocumentEmbedder component.
202205

203206
**Parameters:**
204207

@@ -346,7 +349,7 @@ Embeds text using the Fastembed model.
346349
**Returns:**
347350

348351
- <code>dict\[str, SparseEmbedding\]</code> – A dictionary with the following keys:
349-
- `embedding`: A list of floats representing the embedding of the input text.
352+
- `sparse_embedding`: The `SparseEmbedding` of the input text.
350353

351354
**Raises:**
352355

@@ -385,6 +388,7 @@ __init__(
385388
progress_bar: bool = True,
386389
parallel: int | None = None,
387390
local_files_only: bool = False,
391+
model_kwargs: dict[str, Any] | None = None,
388392
) -> None
389393
```
390394

@@ -404,6 +408,8 @@ Create a FastembedTextEmbedder component.
404408
If 0, use all available cores.
405409
If None, don't use data-parallel processing, use default onnxruntime threading instead.
406410
- **local_files_only** (<code>bool</code>) – If `True`, only use the model files in the `cache_dir`.
411+
- **model_kwargs** (<code>dict\[str, Any\] | None</code>) – Dictionary containing additional keyword arguments to pass to the Fastembed model,
412+
such as `providers` (e.g. `["CUDAExecutionProvider"]` to run on GPU), `cuda`, or `device_ids`.
407413

408414
#### to_dict
409415

0 commit comments

Comments
 (0)