mirror of
https://github.com/discourse/discourse-ai.git
synced 2025-03-09 11:48:47 +00:00
e2e753d73c
We have a flag to signal we are shortening the embeddings of a model. Only used in Open AI's text-embedding-3-*, but we plan to use it for other services.
33 lines
697 B
Ruby
33 lines
697 B
Ruby
# frozen_string_literal: true
|
|
|
|
class AiEmbeddingDefinitionSerializer < ApplicationSerializer
|
|
root "ai_embedding"
|
|
|
|
attributes :id,
|
|
:display_name,
|
|
:dimensions,
|
|
:max_sequence_length,
|
|
:pg_function,
|
|
:provider,
|
|
:url,
|
|
:api_key,
|
|
:seeded,
|
|
:tokenizer_class,
|
|
:embed_prompt,
|
|
:search_prompt,
|
|
:matryoshka_dimensions,
|
|
:provider_params
|
|
|
|
def api_key
|
|
object.seeded? ? "********" : object.api_key
|
|
end
|
|
|
|
def url
|
|
object.seeded? ? "********" : object.url
|
|
end
|
|
|
|
def provider
|
|
object.seeded? ? "CDCK" : object.provider
|
|
end
|
|
end
|