Skip to content

Commit

Permalink
Fix syntax errors
Browse files Browse the repository at this point in the history
  • Loading branch information
kaancayli committed Feb 19, 2024
1 parent c24439c commit dbb189f
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions app/llm/basic_request_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

0 comments on commit dbb189f

Please sign in to comment.