diff --git a/myllm/main.py b/myllm/main.py index f64e0870..e2b2044a 100644 --- a/myllm/main.py +++ b/myllm/main.py @@ -53,6 +53,7 @@ def __init__(self): provider_module = importlib.import_module(provider_module_name) provider_class = getattr(provider_module, provider_module_name.split(".")[-1]) self.provider = provider_class() + self.llm_model = settings.llm_model self.conversation = Conversation() async def get_myllm_help(self): @@ -88,7 +89,7 @@ async def chat(self, prompt): try: self.conversation.add_message("user", prompt) response = await self.provider.create_async( - model=settings.llm_model, + model=self.llm_model, messages=self.conversation.get_messages(), ) sleep(settings.lag)