From 46b9ff9e2ce94dca094fa3376937afe436c03507 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fl=C3=A1vio=20Code=C3=A7o=20Coelho?= Date: Sat, 4 May 2024 11:52:39 -0300 Subject: [PATCH] added support to phi3 model --- base_agent/llminterface.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/base_agent/llminterface.py b/base_agent/llminterface.py index c8901ba..bda5616 100644 --- a/base_agent/llminterface.py +++ b/base_agent/llminterface.py @@ -30,6 +30,9 @@ def get_response(self, question: str, context: str = None) -> str: elif 'wizard' in self.model: self.model = 'wizardlm2' return self.get_ollama_response(question, context) + elif 'phi' in self.model: + self.model = 'phi3' + return self.get_ollama_response(question, context) def get_gpt_response(self, question: str, context: str)->str: response = self.llm.chat.completions.create(