From 23d528a7fe41f87ade99cd7215661b2baff16faf Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Daniel=20Grie=C3=9Fhaber?= <griesshaber@hdm-stuttgart.de>
Date: Tue, 20 Aug 2024 09:34:08 +0200
Subject: [PATCH] fix mro for chat models
---
evoprompt/models.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/evoprompt/models.py b/evoprompt/models.py
index ce55258..f922169 100644
--- a/evoprompt/models.py
+++ b/evoprompt/models.py
@@ -296,7 +296,7 @@ class ChatModel:
return reponse, history + messages, usage
-class LlamaChat(Llama, ChatModel):
+class LlamaChat(ChatModel, Llama):
def _create_completion(
self,
@@ -314,7 +314,7 @@ class LlamaChat(Llama, ChatModel):
return response_text, usage
-class OpenAiChat(LLMModel, ChatModel):
+class OpenAiChat(ChatModel, LLMModel):
"""Queries an OpenAI model using its API."""
def __init__(
--
GitLab