Add ollama embeddings (#1634)

This commit is contained in:
Dev Khant
2024-08-03 10:55:40 +05:30
committed by GitHub
parent 1c46fddce1
commit 024089d33e
7 changed files with 57 additions and 16 deletions

View File

@@ -12,6 +12,8 @@ class BaseLlmConfig(ABC):
temperature: float = 0,
max_tokens: int = 3000,
top_p: float = 1,
# Ollama specific
base_url: Optional[str] = None
):
"""
@@ -35,4 +37,6 @@ class BaseLlmConfig(ABC):
self.temperature = temperature
self.max_tokens = max_tokens
self.top_p = top_p
self.base_url = base_url
# Ollama specific
self.base_url = base_url