Add Ollama as a supported embedding provider (#1344)

This commit is contained in:
Colin O'Brien
2024-05-01 22:08:47 -07:00
committed by GitHub
parent 1a66f961f4
commit a795798156
9 changed files with 46 additions and 2 deletions
+1
View File
@@ -6,6 +6,7 @@ from .base_config import BaseConfig
from .cache_config import CacheConfig
from .embedder.base import BaseEmbedderConfig
from .embedder.base import BaseEmbedderConfig as EmbedderConfig
from .embedder.ollama import OllamaEmbedderConfig
from .llm.base import BaseLlmConfig
from .vectordb.chroma import ChromaDbConfig
from .vectordb.elasticsearch import ElasticsearchDBConfig
+15
View File
@@ -0,0 +1,15 @@
from typing import Optional
from embedchain.config.embedder.base import BaseEmbedderConfig
from embedchain.helpers.json_serializable import register_deserializable
@register_deserializable
class OllamaEmbedderConfig(BaseEmbedderConfig):
def __init__(
self,
model: Optional[str] = None,
base_url: Optional[str] = None,
):
super().__init__(model)
self.base_url = base_url or "http://127.0.0.1:11434"