Test the hello world agent, and setup fixture for loading the plugin

Author: donald-pinckney

tests/conftest.py

@@ -36,9 +36,14 @@ def event_loop():
     yield loop
     loop.close()
 
+@pytest.fixture(scope="session")
+def plugins():
+    # By default, no plugins.
+    # Other tests can override this fixture, such as in tests/openai_agents/conftest.py
+    return []
 
 @pytest_asyncio.fixture(scope="session")
-async def env(request) -> AsyncGenerator[WorkflowEnvironment, None]:
+async def env(request, plugins) -> AsyncGenerator[WorkflowEnvironment, None]:
     env_type = request.config.getoption("--workflow-environment")
     if env_type == "local":
         env = await WorkflowEnvironment.start_local(
@@ -47,12 +52,13 @@ async def env(request) -> AsyncGenerator[WorkflowEnvironment, None]:
                 "frontend.enableExecuteMultiOperation=true",
                 "--dynamic-config-value",
                 "system.enableEagerWorkflowStart=true",
-            ]
+            ],
+            plugins=plugins
         )
     elif env_type == "time-skipping":
         env = await WorkflowEnvironment.start_time_skipping()
     else:
-        env = WorkflowEnvironment.from_client(await Client.connect(env_type))
+        env = WorkflowEnvironment.from_client(await Client.connect(env_type, plugins=plugins))
     yield env
     await env.shutdown()
 

tests/openai_agents/__init__.py

@@ -0,0 +1,15 @@
+from datetime import timedelta
+
+import pytest
+from temporalio.contrib.openai_agents import ModelActivityParameters, OpenAIAgentsPlugin
+
+
+@pytest.fixture(scope="session")
+def plugins():
+    return [
+        OpenAIAgentsPlugin(
+            model_params=ModelActivityParameters(
+                start_to_close_timeout=timedelta(seconds=30)
+            )
+        )
+    ]

tests/openai_agents/conftest.py

@@ -0,0 +1,38 @@
+import uuid
+from concurrent.futures import ThreadPoolExecutor
+
+import pytest
+from temporalio import activity
+from temporalio.client import Client
+from temporalio.worker import Worker
+
+from openai_agents.basic.activities.get_weather_activity import get_weather
+from openai_agents.basic.activities.math_activities import multiply_by_two, random_number
+from openai_agents.basic.activities.image_activities import read_image_as_base64
+
+from openai_agents.basic.workflows.hello_world_workflow import HelloWorldAgent
+
+
+@pytest.mark.fixt_data(42)
+async def test_execute_workflow(client: Client):
+    task_queue_name = str(uuid.uuid4())
+
+    async with Worker(
+        client,
+        task_queue=task_queue_name,
+        workflows=[HelloWorldAgent],
+        activity_executor=ThreadPoolExecutor(5),
+        activities=[
+            get_weather,
+            multiply_by_two,
+            random_number,
+            read_image_as_base64,
+        ],
+    ):
+        await client.execute_workflow(
+            HelloWorldAgent.run,
+            "Write a recursive haiku about recursive haikus.",
+            id=str(uuid.uuid4()),
+            task_queue=task_queue_name,
+        )
+