From 12d59f4379663d8786c56daa9713e2599edeeacf Mon Sep 17 00:00:00 2001
From: Bryce <bryce@brycecovertoperations.com>
Date: Sun, 26 Jan 2025 20:24:26 -0800
Subject: [PATCH] changes

---
 code/main.py |  9 +++++++--
 code/test.py | 10 ++++++++++
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/code/main.py b/code/main.py
index c0cf752..49330d6 100755
--- a/code/main.py
+++ b/code/main.py
@@ -17,6 +17,7 @@ BASE_PROMPT="""You extract invoice details from pdfs. Some pdfs are invoices, so
 """.format(slurp_file('schema.json'))
 
 import sys
+import pprint
 # context = sys.argv[1]
 # problem = sys.argv[2]
 # other = sys.argv[3]
@@ -61,9 +62,13 @@ def analyze_url(url):
     )
 
     messages = list(client.beta.threads.messages.list(thread_id=thread.id, run_id=run.id))
-    print("messages", messages)
+    print("MESSAGES")
+    pprint.pprint(messages)
+    print("\n\n")
     print("status", run.status)
-    print("full run", run)
+    print ("\n\n")
+    print("full run")
+    pprint.pprint(run)
     return json.loads(messages[0].content[0].text.value)
 
 def handler(event, context):
diff --git a/code/test.py b/code/test.py
index edafebe..c8d223b 100644
--- a/code/test.py
+++ b/code/test.py
@@ -6,6 +6,16 @@ class TestAnalyzeUrl(unittest.TestCase):
 
     def test_analyze_url(self):
         result = analyze_url('https://data.prod.app.integreatconsult.com/glimpse2/import/c9c5b9af-bfb5-475d-820d-20a6a00f006b.pdf')
+        print(result)
+
+
+        self.assertEqual(result, {})
+        
+        # Add more assertions depending on how you expect your function to behave.
+        
+    def test_analyze_url3(self):
+        result = analyze_url('https://data.prod.app.integreatconsult.com/bryce/glimpse2-testing/SWP.pdf')
+        print(result)
 
         self.assertEqual(result, {})