44 lines
1.3 KiB
Elixir
44 lines
1.3 KiB
Elixir
|
# This file is auto-generated by alef — DO NOT EDIT.
|
||
|
|
# alef:hash:4e15143f4af1ae8bafbdb1506ef057da924484c66a19483966333558ad437e75
|
||
|
|
# To regenerate: alef generate
|
||
|
|
# To verify freshness: alef verify --exit-code
|
||
|
|
# Issues & docs: https://github.com/kreuzberg-dev/alef
|
||
|
|
defmodule Kreuzberg.EmbeddingConfig do
|
||
|
|
@moduledoc """
|
||
|
|
Embedding configuration for text chunks.
|
||
|
|
|
||
|
|
Configures embedding generation using ONNX models via the vendored embedding engine.
|
||
|
|
Requires the `embeddings` feature to be enabled.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@typedoc "Embedding configuration for text chunks."
|
||
|
|
@type t :: %__MODULE__{
|
||
|
|
model: String.t() | nil,
|
||
|
|
normalize: boolean(),
|
||
|
|
batch_size: non_neg_integer(),
|
||
|
|
show_download_progress: boolean(),
|
||
|
|
cache_dir: String.t() | nil,
|
||
|
|
acceleration: map() | nil,
|
||
|
|
max_embed_duration_secs: non_neg_integer() | nil
|
||
|
|
}
|
||
|
|
|
||
|
|
defstruct model: :preset,
|
||
|
|
normalize: true,
|
||
|
|
batch_size: 32,
|
||
|
|
show_download_progress: false,
|
||
|
|
cache_dir: nil,
|
||
|
|
acceleration: nil,
|
||
|
|
max_embed_duration_secs: nil
|
||
|
|
|
||
|
|
defimpl Jason.Encoder do
|
||
|
|
@doc false
|
||
|
|
def encode(value, opts) do
|
||
|
|
value
|
||
|
|
|> Map.from_struct()
|
||
|
|
|> Enum.reject(fn {_k, v} -> v == nil end)
|
||
|
|
|> Enum.into(%{})
|
||
|
|
|> Jason.Encoder.encode(opts)
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|