Merge pull request #532 from 3rd-Son/testrefactor

updated mistral test
swarmauri · Sep 25, 2024 · 5a6bafe · 5a6bafe
2 parents 76df23f + abb3654
commit 5a6bafe
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pkgs/swarmauri/tests/integration/llms/MistralModel_i9n_test.py b/pkgs/swarmauri/tests/integration/llms/MistralModel_i9n_test.py
@@ -6,6 +6,9 @@
 from swarmauri.messages.concrete.AgentMessage import AgentMessage
 from swarmauri.messages.concrete.HumanMessage import HumanMessage
 from swarmauri.messages.concrete.SystemMessage import SystemMessage
+from dotenv import load_dotenv
+
+load_dotenv()
 
 
 @pytest.fixture(scope="module")