20 lines
675 B
Python
20 lines
675 B
Python
from typing import Optional
|
|
|
|
from embedchain.config.embedder.base import BaseEmbedderConfig
|
|
from embedchain.helpers.json_serializable import register_deserializable
|
|
|
|
|
|
@register_deserializable
|
|
class GoogleAIEmbedderConfig(BaseEmbedderConfig):
|
|
def __init__(
|
|
self,
|
|
model: Optional[str] = None,
|
|
deployment_name: Optional[str] = None,
|
|
vector_dimension: Optional[int] = None,
|
|
task_type: Optional[str] = None,
|
|
title: Optional[str] = None,
|
|
):
|
|
super().__init__(model, deployment_name, vector_dimension)
|
|
self.task_type = task_type or "retrieval_document"
|
|
self.title = title or "Embeddings for Embedchain"
|