Skip to content
Snippets Groups Projects
Commit 23d528a7 authored by Grießhaber Daniel's avatar Grießhaber Daniel :squid:
Browse files

fix mro for chat models

parent 331a4901
No related branches found
No related tags found
2 merge requests!2remove is_chat argument,!1Refactor models
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
evoprompt/models.py
+ 2
− 2
View file @ 23d528a7
...@@ -296,7 +296,7 @@ class ChatModel: ...@@ -296,7 +296,7 @@ class ChatModel:
return reponse, history + messages, usage return reponse, history + messages, usage
class LlamaChat(Llama, ChatModel): class LlamaChat(ChatModel, Llama):
def _create_completion( def _create_completion(
self, self,
...@@ -314,7 +314,7 @@ class LlamaChat(Llama, ChatModel): ...@@ -314,7 +314,7 @@ class LlamaChat(Llama, ChatModel):
return response_text, usage return response_text, usage
class OpenAiChat(LLMModel, ChatModel): class OpenAiChat(ChatModel, LLMModel):
"""Queries an OpenAI model using its API.""" """Queries an OpenAI model using its API."""
def __init__( def __init__(
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment