Update dynamic_cheatsheet/language_model.py

dynamic_cheatsheet/language_model.py

@@ -71,6 +71,7 @@ class LanguageModel:
             # Or, if SambaNova is a recognized provider by a different name in litellm, use that.
             # Given the error, litellm is not recognizing "samba/" as a provider directly.
             # The simp            actual_model_name = self.model_name.split("samba/", 1)[1] if "samba/" in self.model_name else self.model_name
+            actual_model_name = self.model_name.split("sambanova/", 1)[1] if "sambanova/" in self.model_name else self.model_name
             self.client = partial(completion,
                                   model=actual_model_name,
                                   api_key=samba_api_key,