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

2024-01-09 00:17:46 +05:30
parent 62c0c52e31
commit 2496ed133e
41 changed files with 133 additions and 103 deletions
@@ -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(