Update dynamic_cheatsheet/language_model.py

dynamic_cheatsheet/language_model.py

@@ -6,6 +6,7 @@ from .utils.execute_code import extract_and_run_python_code
 from .utils.extractor import extract_answer, extract_cheatsheet
 from litellm import completion
 from functools import partial
+import litellm
 import os # Added for SAMBANOVA env vars
 
 litellm._turn_on_debug()