Added Support for Ollama for local model inference. (#1045)

Co-authored-by: Deshraj Yadav <deshraj@gatech.edu>
2023-12-22 05:10:00 +05:30
parent 210fe9bb80
commit 0f73ba9677
11 changed files with 348 additions and 1 deletions
--- a/embedchain/models/providers.py
+++ b/embedchain/models/providers.py
@@ -6,4 +6,5 @@ class Providers(Enum):
    ANTHROPHIC = "ANTHPROPIC"
    VERTEX_AI = "VERTEX_AI"
    GPT4ALL = "GPT4ALL"
+    OLLAMA = "OLLAMA"
    AZURE_OPENAI = "AZURE_OPENAI"