klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add vector dimension to Ollama embedder (#1435)

Browse Source
This commit is contained in:
Dev Khant
2024-06-21 21:26:46 +05:30
committed by GitHub
parent 50512a5f03
commit b43a116b3c
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
embedchain/config/embedder/google.py
View File

@@ -10,9 +10,10 @@ class GoogleAIEmbedderConfig(BaseEmbedderConfig):
self, self,
model: Optional[str] = None, model: Optional[str] = None,
deployment_name: Optional[str] = None, deployment_name: Optional[str] = None,
vector_dimension: Optional[int] = None,
task_type: Optional[str] = None, task_type: Optional[str] = None,
title: Optional[str] = None, title: Optional[str] = None,
): ):
super().__init__(model, deployment_name) super().__init__(model, deployment_name, vector_dimension)
self.task_type = task_type or "retrieval_document" self.task_type = task_type or "retrieval_document"
self.title = title or "Embeddings for Embedchain" self.title = title or "Embeddings for Embedchain"

3
embedchain/config/embedder/ollama.py
View File

@@ -10,6 +10,7 @@ class OllamaEmbedderConfig(BaseEmbedderConfig):
self, self,
model: Optional[str] = None, model: Optional[str] = None,
base_url: Optional[str] = None, base_url: Optional[str] = None,
vector_dimension: Optional[int] = None,
): ):
super().__init__(model) super().__init__(model=model, vector_dimension=vector_dimension)
self.base_url = base_url or "http://localhost:11434" self.base_url = base_url or "http://localhost:11434"