From 14946216a5c4c1b6092dd752eafb81ff546e321a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fl=C3=A1vio=20Code=C3=A7o=20Coelho?= <fccoelho@gmail.com>
Date: Fri, 20 Sep 2024 10:56:28 -0300
Subject: [PATCH] fixed tests

---
 tests/test_llms.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/test_llms.py b/tests/test_llms.py
index fa84a6a..bf98469 100644
--- a/tests/test_llms.py
+++ b/tests/test_llms.py
@@ -90,7 +90,8 @@ def test_get_strutured_output_ollama(self):
 """
 
         self.assertIsInstance(response, Character)
-        self.assertEqual('Harry James Potter', response.name)
+        assert response.name.startswith('Harry')
+        assert response.name.endswith('Potter')
 
 if __name__ == '__main__':
     unittest.main()