Update pytests

.github/workflows/test.yml

@@ -24,6 +24,10 @@ jobs:
           - 11434:11434
         volumes:
           - ollama:/root/.ollama
+
+    strategy:
+      matrix:
+        model: [gemini/gemini-1.5-flash-001, openai/gpt-4o] # List of models
 
     steps:
       - uses: actions/checkout@v4
@@ -73,4 +77,5 @@ jobs:
           REGION: ${{ vars.REGION }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
           GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
+          TEST_MODEL: ${{ matrix.model }} # Pass the model as an environment variable
         run: poetry run pytest