Feature: Custom Ollama endpoint base_url (#1301)

2024-03-04 23:09:45 -03:00
parent faacfeb891
commit 6078738d34
4 changed files with 5 additions and 0 deletions
--- a/embedchain/llm/ollama.py
+++ b/embedchain/llm/ollama.py
@@ -31,6 +31,7 @@ class OllamaLlm(BaseLlm):
            temperature=config.temperature,
            top_p=config.top_p,
            callback_manager=CallbackManager(callback_manager),
+            base_url=config.base_url,
        )

        return llm.invoke(prompt)