From dbb189f381a29648e6a105e1981f6d209272143c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Kaan=20=C3=87ayl=C4=B1?= Date: Mon, 19 Feb 2024 12:43:40 +0100 Subject: [PATCH] Fix syntax errors --- app/llm/basic_request_handler.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/app/llm/basic_request_handler.py b/app/llm/basic_request_handler.py index a5d2ca15..14227997 100644 --- a/app/llm/basic_request_handler.py +++ b/app/llm/basic_request_handler.py @@ -12,15 +12,15 @@ def __init__(self, model_id: str): self.llm_manager = LlmManager() def complete(self, prompt: str, arguments: CompletionArguments) -> str: - llm = self.llm_manager.get_by_id(self.model_id) + llm = self.llm_manager.get_llm_by_id(self.model_id) return llm.complete(prompt, arguments) def chat( self, messages: list[IrisMessage], arguments: CompletionArguments ) -> IrisMessage: - llm = self.llm_manager.get_by_id(self.model_id) + llm = self.llm_manager.get_llm_by_id(self.model_id) return llm.chat(messages, arguments) def embed(self, text: str) -> list[float]: - llm = self.llm_manager.get_by_id(self.model_id) + llm = self.llm_manager.get_llm_by_id(self.model_id) return llm.embed(text)