Skip to content

Commit b408d8c

Browse files
Update with redesigned agent environement
1 parent ad49b99 commit b408d8c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

temporalio/contrib/openai_agents/testing.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,13 @@
2929
from temporalio.contrib.openai_agents._model_parameters import ModelActivityParameters
3030
from temporalio.contrib.openai_agents._temporal_openai_agents import OpenAIAgentsPlugin
3131

32+
__all__ = [
33+
"AgentEnvironment",
34+
"ResponseBuilders",
35+
"TestModel",
36+
"TestModelProvider",
37+
]
38+
3239

3340
class ResponseBuilders:
3441
"""Builders for creating model responses for testing.
@@ -289,7 +296,7 @@ def applied_on_client(self, client: Client) -> Client:
289296
return Client(**new_config)
290297

291298
@property
292-
def plugin(self) -> OpenAIAgentsPlugin:
299+
def openai_agents_plugin(self) -> OpenAIAgentsPlugin:
293300
"""Get the underlying OpenAI agents plugin.
294301
295302
.. warning::

0 commit comments

Comments
 (0)