diff --git a/libs/partners/mistralai/Makefile b/libs/partners/mistralai/Makefile
index 50d47f2abb2af..5b025fe76cd75 100644
--- a/libs/partners/mistralai/Makefile
+++ b/libs/partners/mistralai/Makefile
@@ -5,11 +5,14 @@ all: help
 
 # Define a variable for the test file path.
 TEST_FILE ?= tests/unit_tests/
+INTEGRATION_TEST_FILE ?= tests/integration_tests/
 
-test:
+integration_test integration_tests: TEST_FILE=$(INTEGRATION_TEST_FILE)
+
+test tests:
 	poetry run pytest $(TEST_FILE)
 
-tests:
+integration_test integration_tests:
 	poetry run pytest $(TEST_FILE)