tests: Enable xai integration tests. (#709)

* tests: Enable `xai` integration tests.

* Set drop_params also for `xai`

Author: daavoo

.github/workflows/tests-integration.yaml

@@ -31,6 +31,7 @@ jobs:
           GEMINI_API_KEY: ${{ secrets.GEMINI_API_KEY }}
           MISTRAL_API_KEY: ${{ secrets.MISTRAL_API_KEY }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+          XAI_API_KEY: ${{ secrets.XAI_API_KEY }}
         run: pytest tests/integration/frameworks -v -k "TINYAGENT" --cov --cov-report=xml --cov-append
 
       - name: Upload coverage reports to Codecov

tests/integration/frameworks/test_agent.py

@@ -114,6 +114,7 @@ class Steps(BaseModel):
         "anthropic/claude-3-5-haiku-latest",
         "gemini/gemini-2.5-flash",
         "openai/gpt-4.1-nano",
+        "xai/grok-3-mini-latest",
         DEFAULT_SMALL_MODEL_ID,
     ],
 )
@@ -152,7 +153,7 @@ def write_file(text: str) -> None:
         pytest.skip(f"{env_check['missing_keys']} needed for {model_id}")
 
     model_args = get_default_agent_model_args(agent_framework)
-    if "gemini" in model_id:
+    if "gemini" in model_id or "xai" in model_id:
         model_args["drop_params"] = True
 
     tools = [