fixed path testing in scraper

emcf · Jul 26, 2024 · afc35e5 · afc35e5
1 parent d4bb7d8
commit afc35e5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@ def read_git_requirements(file):
 
 setup(
     name='thepipe_api',
-    version='1.2.0',
+    version='1.2.1',
     author='Emmett McFarlane',
     author_email='[email protected]',
     description='AI-native extractor, powered by multimodal LLMs.',

diff --git a/tests/test_core.py b/tests/test_core.py
@@ -35,7 +35,7 @@ def test_chunks_to_messages(self):
             self.assertEqual(type(message), dict)
             self.assertIn('role', message)
             self.assertIn('content', message)
-            self.assertIn('example.md', message['content'])
+            self.assertIn('example.md', message['content'][0]['text'])
 
     def test_save_outputs(self):
         chunks = scraper.scrape_plaintext(file_path=self.files_directory+"/example.txt")