From afc35e5fca6832c80511bea51a27f55ed812c3a4 Mon Sep 17 00:00:00 2001 From: Emmett McFaralne Date: Fri, 26 Jul 2024 01:24:00 -0400 Subject: [PATCH] fixed path testing in scraper --- setup.py | 2 +- tests/test_core.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 57258e1..4c5923c 100644 --- 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='emmett@thepi.pe', description='AI-native extractor, powered by multimodal LLMs.', diff --git a/tests/test_core.py b/tests/test_core.py index a74c5f1..99d62c8 100644 --- 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")