Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions tests/functional/model/run_connect_model_test.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,8 @@
import os
from uuid import uuid4

import pytest

from aixplain.enums import ResponseStatus
from aixplain.factories import ModelFactory
from aixplain.modules.model.integration import Integration, AuthenticationSchema
Expand Down Expand Up @@ -39,6 +41,11 @@ def test_run_connect_model():
connection.delete()


@pytest.mark.skip(
reason="The hardcoded Zapier MCP share URL (token committed in plaintext) is "
"dead — every POST fails with 'Streamable HTTP error'. Re-enable with a "
"fresh, secret-managed endpoint."
)
def test_run_mcp_connect_model():
# get slack connector
connector = ModelFactory.get("686eb9cd26480723d0634d3e")
Expand Down
1 change: 1 addition & 0 deletions tests/functional/team_agent/team_agent_functional_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -255,6 +255,7 @@ def test_add_remove_agents_from_team_agent(run_input_map, resource_tracker, Team
assert len(team_agent.agents) == len(agents)


@pytest.mark.flaky(reruns=2, reruns_delay=5)
def test_team_agent_tasks(resource_tracker):
agent_name = f"TSA {str(uuid4())[:8]}"
agent = AgentFactory.create(
Expand Down
3 changes: 3 additions & 0 deletions tests/functional/v2/test_arabic_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -269,6 +269,7 @@ def _make_output_inspector(llm_id: str, model_name: str):
)


@pytest.mark.flaky(reruns=1, reruns_delay=5)
@pytest.mark.parametrize(("model_name", "llm_id"), MODELS)
def test_arabic_single_agent_variants_across_llms(client, resource_tracker, model_name, llm_id):
"""Keep broad LLM coverage, but run only one representative query per agent variant."""
Expand All @@ -288,6 +289,7 @@ def test_arabic_single_agent_variants_across_llms(client, resource_tracker, mode
_assert_query_expectations(query_key, output)


@pytest.mark.flaky(reruns=1, reruns_delay=5)
@pytest.mark.parametrize(("model_name", "llm_id"), MODELS)
def test_arabic_team_agent_across_llms(client, resource_tracker, model_name, llm_id):
"""Use one contract-heavy query to cover the team-agent serialization path."""
Expand All @@ -301,6 +303,7 @@ def test_arabic_team_agent_across_llms(client, resource_tracker, model_name, llm
assert steps, "Expected team-agent execution steps for Arabic team flow"


@pytest.mark.flaky(reruns=1, reruns_delay=5)
@pytest.mark.parametrize(("model_name", "llm_id"), MODELS)
def test_arabic_inspector_agent_across_llms(client, resource_tracker, model_name, llm_id):
"""Verify the inspector actually executes on the Arabic runtime path."""
Expand Down
Loading