Add tests for azure

libs/agno/tests/integration/models/azure/ai_foundry/test_basic.py

@@ -0,0 +1,156 @@
+import pytest
+from pydantic import BaseModel, Field
+
+from agno.agent import Agent, RunResponse
+from agno.models.azure import AzureAIFoundry
+from agno.storage.agent.postgres import PostgresAgentStorage
+
+
+def _assert_metrics(response: RunResponse):
+    input_tokens = response.metrics.get("input_tokens", [])
+    output_tokens = response.metrics.get("output_tokens", [])
+    total_tokens = response.metrics.get("total_tokens", [])
+
+    assert sum(input_tokens) > 0
+    assert sum(output_tokens) > 0
+    assert sum(total_tokens) > 0
+    assert sum(total_tokens) == sum(input_tokens) + sum(output_tokens)
+
+
+def test_basic():
+    agent = Agent(model=AzureAIFoundry(id="Phi-4"), markdown=True, telemetry=False, monitoring=False)
+
+    # Print the response in the terminal
+    response: RunResponse = agent.run("Share a 2 sentence horror story")
+
+    assert response.content is not None
+    assert len(response.messages) == 3
+    assert [m.role for m in response.messages] == ["system", "user", "assistant"]
+
+    _assert_metrics(response)
+
+
+def test_basic_stream():
+    agent = Agent(
+        model=AzureAIFoundry(id="Phi-4"),
+        instructions="You tell ghost stories",
+        markdown=True,
+        telemetry=False,
+        monitoring=False,
+    )
+
+    response_stream = agent.run("Share a 2 sentence horror story", stream=True)
+
+    # Verify it's an iterator
+    assert hasattr(response_stream, "__iter__")
+
+    responses = list(response_stream)
+    assert len(responses) > 0
+    for response in responses:
+        assert isinstance(response, RunResponse)
+        assert response.content is not None
+
+    _assert_metrics(agent.run_response)
+
+
+@pytest.mark.asyncio
+async def test_async_basic():
+    agent = Agent(model=AzureAIFoundry(id="Phi-4"), markdown=True, telemetry=False, monitoring=False)
+
+    response = await agent.arun("Share a 2 sentence horror story")
+
+    assert response.content is not None
+    assert len(response.messages) == 3
+    assert [m.role for m in response.messages] == ["system", "user", "assistant"]
+    _assert_metrics(response)
+
+
+@pytest.mark.asyncio
+async def test_async_basic_stream():
+    agent = Agent(model=AzureAIFoundry(id="Phi-4"), markdown=True, telemetry=False, monitoring=False)
+
+    response_stream = await agent.arun("Share a 2 sentence horror story", stream=True)
+
+    async for response in response_stream:
+        assert isinstance(response, RunResponse)
+        assert response.content is not None
+
+    _assert_metrics(agent.run_response)
+
+
+def test_with_memory():
+    agent = Agent(
+        model=AzureAIFoundry(id="Phi-4"),
+        add_history_to_messages=True,
+        num_history_responses=5,
+        markdown=True,
+        telemetry=False,
+        monitoring=False,
+    )
+
+    # First interaction
+    response1 = agent.run("My name is John Smith")
+    assert response1.content is not None
+
+    # Second interaction should remember the name
+    response2 = agent.run("What's my name?")
+    assert "John Smith" in response2.content
+
+    # Verify memories were created
+    assert len(agent.memory.messages) == 5
+    assert [m.role for m in agent.memory.messages] == ["system", "user", "assistant", "user", "assistant"]
+
+    # Test metrics structure and types
+    input_tokens = response2.metrics["input_tokens"]
+    output_tokens = response2.metrics["output_tokens"]
+    total_tokens = response2.metrics["total_tokens"]
+
+    assert isinstance(input_tokens[0], int)
+    assert input_tokens[0] > 0
+    assert isinstance(output_tokens[0], int)
+    assert output_tokens[0] > 0
+    assert isinstance(total_tokens[0], int)
+    assert total_tokens[0] > 0
+    assert total_tokens[0] == input_tokens[0] + output_tokens[0]
+
+
+def test_structured_output():
+    class MovieScript(BaseModel):
+        title: str = Field(..., description="Movie title")
+        genre: str = Field(..., description="Movie genre")
+        plot: str = Field(..., description="Brief plot summary")
+
+    agent = Agent(
+        model=AzureAIFoundry(id="Phi-4"),
+        response_model=MovieScript,
+        telemetry=False,
+        monitoring=False,
+    )
+
+    response = agent.run("Create a movie about time travel")
+
+    # Verify structured output
+    assert isinstance(response.content, MovieScript)
+    assert response.content.title is not None
+    assert response.content.genre is not None
+    assert response.content.plot is not None
+
+
+def test_history():
+    db_url = "postgresql+psycopg://ai:ai@localhost:5532/ai"
+    agent = Agent(
+        model=AzureAIFoundry(id="Phi-4"),
+        storage=PostgresAgentStorage(table_name="agent_sessions", db_url=db_url),
+        add_history_to_messages=True,
+        telemetry=False,
+        monitoring=False,
+        markdown=True,
+    )
+    agent.run("Hello")
+    assert len(agent.run_response.messages) == 2
+    agent.run("Hello 2")
+    assert len(agent.run_response.messages) == 4
+    agent.run("Hello 3")
+    assert len(agent.run_response.messages) == 6
+    agent.run("Hello 4")
+    assert len(agent.run_response.messages) == 8

libs/agno/tests/integration/models/azure/ai_foundry/test_multimodal.py

@@ -0,0 +1,17 @@
+from agno.agent.agent import Agent
+from agno.media import Image
+from agno.models.azure import AzureAIFoundry
+
+
+def test_image_input():
+    agent = Agent(
+        model=AzureAIFoundry(id="Phi-4"), markdown=True, telemetry=False, monitoring=False
+    )
+
+    response = agent.run(
+        "Tell me about this image.",
+        images=[Image(url="https://upload.wikimedia.org/wikipedia/commons/0/0c/GoldenGateBridge-001.jpg")],
+    )
+
+    assert "golden" in response.content.lower()
+    assert "bridge" in response.content.lower()