ollama.py 449 B

123456789101112131415
  1. from typing import Optional
  2. from embedchain.config.embedder.base import BaseEmbedderConfig
  3. from embedchain.helpers.json_serializable import register_deserializable
  4. @register_deserializable
  5. class OllamaEmbedderConfig(BaseEmbedderConfig):
  6. def __init__(
  7. self,
  8. model: Optional[str] = None,
  9. base_url: Optional[str] = None,
  10. ):
  11. super().__init__(model)
  12. self.base_url = base_url or "http://127.0.0.1:11434"