This commit is contained in:
43
packages/elixir/lib/kreuzberg/embedding_config.ex
generated
Normal file
43
packages/elixir/lib/kreuzberg/embedding_config.ex
generated
Normal file
@@ -0,0 +1,43 @@
|
||||
# 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
|
||||
Reference in New Issue
Block a user