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

@@ -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"