Provide defaults

Author: tconley1428

temporalio/contrib/openai_agents/_openai_runner.py

@@ -7,6 +7,7 @@
     RunConfig,
     RunResult,
     RunResultStreaming,
+    SQLiteSession,
     TContext,
     Tool,
     TResponseInputItem,
@@ -61,6 +62,10 @@ async def run(
         hooks = kwargs.get("hooks")
         run_config = kwargs.get("run_config")
         previous_response_id = kwargs.get("previous_response_id")
+        session = kwargs.get("session")
+
+        if isinstance(session, SQLiteSession):
+            raise ValueError("Temporal workflows don't support SQLite sessions.")
 
         if run_config is None:
             run_config = RunConfig()
@@ -85,6 +90,7 @@ async def run(
             hooks=hooks,
             run_config=updated_run_config,
             previous_response_id=previous_response_id,
+            session=session,
         )
 
     def run_sync(

temporalio/contrib/pydantic.py

@@ -53,7 +53,7 @@ class PydanticJSONPlainPayloadConverter(EncodingPayloadConverter):
     See https://docs.pydantic.dev/latest/api/standard_library_types/
     """
 
-    def __init__(self, to_json_options: Optional[ToJsonOptions]):
+    def __init__(self, to_json_options: Optional[ToJsonOptions] = None):
         """Create a new payload converter."""
         self._schema_serializer = SchemaSerializer(any_schema())
         self._to_json_options = to_json_options
@@ -106,7 +106,7 @@ class PydanticPayloadConverter(CompositePayloadConverter):
     :py:class:`PydanticJSONPlainPayloadConverter`.
     """
 
-    def __init__(self, to_json_options: Optional[ToJsonOptions]) -> None:
+    def __init__(self, to_json_options: Optional[ToJsonOptions] = None) -> None:
         """Initialize object"""
         json_payload_converter = PydanticJSONPlainPayloadConverter(to_json_options)
         super().__init__(

tests/contrib/openai_agents/test_openai.py

@@ -34,6 +34,7 @@
     OutputGuardrailTripwireTriggered,
     RunContextWrapper,
     Runner,
+    SQLiteSession,
     Tool,
     TResponseInputItem,
     Usage,
@@ -90,7 +91,7 @@
 from tests.contrib.openai_agents.research_agents.research_manager import (
     ResearchManager,
 )
-from tests.helpers import new_worker
+from tests.helpers import assert_task_fail_eventually, new_worker
 from tests.helpers.nexus import create_nexus_endpoint, make_nexus_endpoint_name
 
 
@@ -1996,6 +1997,47 @@ def test_summary_extraction():
     assert _extract_summary(input) == "Second message"
 
 
+@workflow.defn
+class SessionWorkflow:
+    @workflow.run
+    async def run(self) -> None:
+        agent: Agent = Agent(
+            name="Assistant",
+            instructions="You are a helpful assistant.",
+        )
+        await Runner.run(
+            agent,
+            "My phone number is 650-123-4567. Where do you think I live?",
+            session=SQLiteSession(session_id="id"),
+        )
+
+
+async def test_session(client: Client):
+    new_config = client.config()
+    new_config["plugins"] = [
+        openai_agents.OpenAIAgentsPlugin(
+            model_provider=TestModelProvider(TestHelloModel()),
+        )
+    ]
+    client = Client(**new_config)
+
+    async with new_worker(
+        client,
+        SessionWorkflow,
+    ) as worker:
+        workflow_handle = await client.start_workflow(
+            SessionWorkflow.run,
+            id=f"session-{uuid.uuid4()}",
+            task_queue=worker.task_queue,
+            execution_timeout=timedelta(seconds=1.0),
+            retry_policy=RetryPolicy(maximum_attempts=1),
+        )
+        await assert_task_fail_eventually(
+            workflow_handle,
+            message_contains="Temporal workflows don't support SQLite sessions",
+        )
+
+
 async def test_lite_llm(client: Client, env: WorkflowEnvironment):
     if not os.environ.get("OPENAI_API_KEY"):
         pytest.skip("No openai API key")