test/files/test_uploaded_as_input.py

Rama main · réplica de lectura del repositorio original.

# Copyright (c) 2026, AG2ai, Inc., AG2ai open-source projects maintainers and core contributors
#
# SPDX-License-Identifier: Apache-2.0

import pytest

from ag2 import Agent
from ag2.files import FileProvider, UploadedFile
from ag2.testing import TestConfig


@pytest.mark.asyncio
async def test_uploaded_file_passed_to_agent_ask() -> None:
    agent = Agent("test", config=TestConfig("done"))
    uploaded = UploadedFile(file_id="file-123", filename="doc.pdf", provider=FileProvider.OPENAI)

    reply = await agent.ask("Summarize this", uploaded)

    assert reply.body == "done"