add llm

IMMM-SFA · Jan 14, 2025 · 0b6f373 · 0b6f373
1 parent 3563bc4
commit 0b6f373
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/format_file.py b/format_file.py
@@ -31,11 +31,13 @@
 )
 
 chain = prompt | llm
-chain.invoke(
+ai_msg = chain.invoke(
     {
         "language": "Python",
         "input": "Go through all the scripts in this repository and provide detailed feedback and suggestions on how to make the scripts easier to read, cleaner, and more reproducible.",
     }
 )
 
-print(chain)
+print(chain)
+print(ai_msg)
+print(ai_msg.content)