fix: openrouter config

vanna-ai · Jul 8, 2024 · 4c87262 · 4c87262
1 parent 5336825
commit 4c87262
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/vanna/openrouter/openrouter_chat.py b/src/vanna/openrouter/openrouter_chat.py
@@ -23,9 +23,12 @@ def __init__(
         VannaBase.__init__(self, config=config)
         self.client = client
         self.aclient = aclient
-        if config is not None:
-            self.temperature = config["temperature"] or 0.7
-            self.max_tokens = config["max_tokens"] or 500
+        self.temperature = 0.7
+        self.max_tokens = 500
+        if config is not None and "temperature" in config:
+            self.temperature = config["temperature"]
+        if config is not None and "max_tokens" in config:
+            self.max_tokens = config["max_tokens"]
 
     def system_message(self, message: str) -> Any:
         return {"role": "system", "content": message}