Update custom_llm.py

custom_llm.py

@@ -22,7 +22,7 @@ from langchain_community.embeddings import HuggingFaceInferenceAPIEmbeddings
 
 from langchain.vectorstores import FAISS
 
-import pickle, asyncio
+import pickle, asyncio, traceback
 
 # os.environ['FAISS_NO_AVX2'] = '1'
 import pandas as pd
@@ -191,9 +191,9 @@ def out_format(text, llm, df):
             
             return text
         except Exception as e:
-            print(f"ERORRR! ATTEMPT : {i}\n",str(e))
-            text = err_chain.invoke({"code":text, "err":str(e)})
-            e_ = e
+            print(f"ERORRR! ATTEMPT : {i}\n",str(traceback.format_exc(limit=2)))
+            text = err_chain.invoke({"code":text, "err":str(traceback.format_exc(limit=2))})
+            e_ = traceback.format_exc(limit=2)
     # exec(text)
     
     return "Bad Python Code, Error Message : " + str(e_)