First test

Author: dargilco

sdk/ai/azure-ai-projects/.env.template

@@ -46,6 +46,8 @@ SHAREPOINT_USER_INPUT=
 FABRIC_USER_INPUT=
 BING_CUSTOM_USER_INPUT=
 A2A_USER_INPUT=
+WORK_IQ_PROJECT_CONNECTION_ID=
+WORK_IQ_USER_INPUT=
 
 #######################################################################
 #

sdk/ai/azure-ai-projects/tests/agents/test_agents_crud.py

@@ -7,7 +7,7 @@
 import json
 import io
 from test_base import TestBase, servicePreparer
-from devtools_testutils import recorded_by_proxy
+from devtools_testutils import recorded_by_proxy, RecordedTransport
 from azure.ai.projects.models import PromptAgentDefinition, AgentDetails, AgentVersionDetails
 
 
@@ -111,3 +111,98 @@ def test_agents_crud(self, **kwargs):
             agent_name=second_agent_name, agent_version=agent2_version1.version
         )
         assert result.deleted
+
+    # To run this test:
+    # pytest tests\agents\test_agents_crud.py::TestAgentCrud::test_agent_disable_enable -s
+    @servicePreparer()
+    @recorded_by_proxy(RecordedTransport.AZURE_CORE, RecordedTransport.HTTPX)
+    def test_agent_disable_enable(self, **kwargs):
+        """
+        Test disable and enable operations for Agents.
+
+        This test creates an agent, verifies it can respond to requests,
+        disables it and verifies requests fail, then enables it and
+        verifies requests work again.
+
+        Routes used in this test:
+
+        Action REST API Route                                Client Method
+        ------+---------------------------------------------+-----------------------------------
+        POST   /agents/{agent_name}/versions                 project_client.agents.create_version()
+        POST   /openai/conversations                         openai_client.conversations.create()
+        POST   /openai/responses                             openai_client.responses.create()
+        POST   /agents/{agent_name}:disable                  project_client.agents.disable()
+        POST   /agents/{agent_name}:enable                   project_client.agents.enable()
+        DELETE /agents/{agent_name}/versions/{agent_version} project_client.agents.delete_version()
+        """
+        print("\n")
+        model = kwargs.get("foundry_model_name")
+        agent_name = "DisableEnableTestAgent"
+
+        # Setup
+        project_client = self.create_client(operation_group="agents", **kwargs)
+        openai_client = project_client.get_openai_client()
+
+        # Create an Agent
+        agent = project_client.agents.create_version(
+            agent_name=agent_name,
+            definition=PromptAgentDefinition(
+                model=model,
+                instructions="You are a helpful assistant that answers general questions",
+            ),
+        )
+        print(f"Agent created (id: {agent.id}, name: {agent.name}, version: {agent.version})")
+        self._validate_agent_version(agent)
+
+        # Create a conversation
+        conversation = openai_client.conversations.create(
+            items=[{"type": "message", "role": "user", "content": "How many feet in a mile?"}]
+        )
+        print(f"Created conversation with initial user message (id: {conversation.id})")
+
+        # Verify the agent can respond to requests
+        response = openai_client.responses.create(
+            conversation=conversation.id,
+            extra_body={"agent_reference": {"name": agent.name, "type": "agent_reference"}},
+        )
+        print(f"Response id: {response.id}, output text: {response.output_text}")
+        assert "5280" in response.output_text or "5,280" in response.output_text
+
+        # Disable the agent
+        project_client.agents.disable(agent_name=agent_name)
+        print(f"Agent disabled")
+
+        # Verify requests fail when agent is disabled
+        error_raised = False
+        try:
+            _ = openai_client.responses.create(
+                conversation=conversation.id,
+                extra_body={"agent_reference": {"name": agent.name, "type": "agent_reference"}},
+            )
+        except Exception as e:
+            error_raised = True
+            print(f"Expected error when calling disabled agent: {e}")
+        assert error_raised, "Expected an error when calling a disabled agent"
+
+        # Enable the agent
+        project_client.agents.enable(agent_name=agent_name)
+        print(f"Agent enabled")
+
+        # Add a new message to the conversation for the next request
+        _ = openai_client.conversations.items.create(
+            conversation.id,
+            items=[{"type": "message", "role": "user", "content": "And how many meters?"}],
+        )
+
+        # Verify the agent can respond to requests again
+        response = openai_client.responses.create(
+            conversation=conversation.id,
+            extra_body={"agent_reference": {"name": agent.name, "type": "agent_reference"}},
+        )
+        print(f"Response id: {response.id}, output text: {response.output_text}")
+        assert "1609" in response.output_text or "1,609" in response.output_text
+
+        # Cleanup - delete the agent
+        result = project_client.agents.delete_version(agent_name=agent_name, agent_version=agent.version)
+        assert result.deleted
+        print(f"Agent deleted")