Update dynamic_cheatsheet/language_model.py

dynamic_cheatsheet/language_model.py

@@ -50,7 +50,7 @@ class LanguageModel:
         ]
 
         # Load the client for the model based on the model name
-        if self.model_name.startswith("samba/"):
+        if self.model_name.startswith("sambanova/"):
             samba_api_key = os.environ.get("SAMBANOVA_API_KEY")
             samba_base_url = os.environ.get("SAMBANOVA_BASE_URL", "https://api.sambanova.ai/v1") # Default if not set
             if not samba_api_key: