changes

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):

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, {})