From 13ede0092d4c157ddde792f64927563b91c01965 Mon Sep 17 00:00:00 2001
From: Csaba Henk <csaba@lowlife.hu>
Date: Thu, 6 Feb 2025 17:05:08 +0100
Subject: [PATCH] fix setting API key in environment

---
 llm_fireworks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/llm_fireworks.py b/llm_fireworks.py
index ece49d4..005dbc9 100644
--- a/llm_fireworks.py
+++ b/llm_fireworks.py
@@ -50,6 +50,7 @@
 
 class FireworksChat(Chat):
     needs_key = "fireworks"
+    key_env_var = "LLM_FIREWORKS_KEY"
 
     def __init__(self, model_name):
         super().__init__(
@@ -69,7 +70,7 @@ def __str__(self):
 @llm.hookimpl
 def register_models(register):
     # Only do this if the key is set
-    key = llm.get_key("", "fireworks", "LLM_FIREWORKS_KEY")
+    key = llm.get_key("", "fireworks", FireworksChat.key_env_var)
     if not key:
         return