[Bug fix] Fix typos, static methods and other sanity improvements in the package (#1129)

2024-01-08 19:47:46 +01:00
parent 62c0c52e31
commit 2496ed133e
41 changed files with 133 additions and 103 deletions
--- a/embedchain/llm/ollama.py
+++ b/embedchain/llm/ollama.py
@@ -20,7 +20,8 @@ class OllamaLlm(BaseLlm):
    def get_llm_model_answer(self, prompt):
        return self._get_answer(prompt=prompt, config=self.config)

-    def _get_answer(self, prompt: str, config: BaseLlmConfig) -> Union[str, Iterable]:
+    @staticmethod
+    def _get_answer(prompt: str, config: BaseLlmConfig) -> Union[str, Iterable]:
        callback_manager = [StreamingStdOutCallbackHandler()] if config.stream else [StdOutCallbackHandler()]

        llm = Ollama(