Add Ollama as a supported embedding provider (#1344)

This commit is contained in:
Colin O'Brien
2024-05-02 01:08:47 -04:00
committed by GitHub
parent 1a66f961f4
commit a795798156
9 changed files with 46 additions and 2 deletions

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

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"