Update dynamic_cheatsheet/language_model.py

dynamic_cheatsheet/language_model.py

@@ -111,7 +111,8 @@ class LanguageModel:
         """
         if len(history) == 0:
             raise ValueError("History must contain at least one message.")
-        
+
+        print('history\n', history)
         # The self.client is already a partial function with model, api_key, base_url, etc., pre-filled for SambaNova
         response = self.client(
             messages=history,
@@ -121,6 +122,8 @@ class LanguageModel:
         )
         output = response.choices[0].message.content # Corrected access to content
 
+        print('output\n', output)
+
         pre_code_execution_flag = output.split(code_execution_flag)[0].strip()
         if allow_code_execution and code_execution_flag in output and pre_code_execution_flag.endswith("```"):
             output_prefix = output.split(code_execution_flag)[0].strip()