Feature/bedrock embedder (#1470)

This commit is contained in:
andrewghlee
2024-08-01 23:25:28 +05:30
committed by GitHub
parent 80945df4ca
commit 563a130141
15 changed files with 390 additions and 26 deletions
@@ -0,0 +1,21 @@
from typing import Any, Dict, Optional
from embedchain.config.embedder.base import BaseEmbedderConfig
from embedchain.helpers.json_serializable import register_deserializable
@register_deserializable
class AWSBedrockEmbedderConfig(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,
model_kwargs: Optional[Dict[str, Any]] = None,
):
super().__init__(model, deployment_name, vector_dimension)
self.task_type = task_type or "retrieval_document"
self.title = title or "Embeddings for Embedchain"
self.model_kwargs = model_kwargs or {}