diff --git a/llm_mistral.py b/llm_mistral.py index f66b6db..c5ff86c 100644 --- a/llm_mistral.py +++ b/llm_mistral.py @@ -81,7 +81,7 @@ def build_messages(self, prompt, conversation): return messages def execute(self, prompt, stream, response, conversation): - key = llm.get_key("", "mistral", "LLM_MISTRAL_KEY") + key = llm.get_key("", "mistral", "LLM_MISTRAL_KEY") or getattr(self, "key", None) messages = self.build_messages(prompt, conversation) response._prompt_json = {"messages": messages} body = {