From 0b6f3739a0e68ced1248d50462fde099cdaf8187 Mon Sep 17 00:00:00 2001
From: erexer <13180883+erexer@users.noreply.github.com>
Date: Tue, 14 Jan 2025 12:15:57 -0800
Subject: [PATCH] add llm

---
 format_file.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/format_file.py b/format_file.py
index de02500..7655a0e 100644
--- 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)
\ No newline at end of file
+print(chain)
+print(ai_msg)
+print(ai_msg.content)
\ No newline at end of file