Fix directory structure

Author: danielmillerp

src/agentex/lib/core/temporal/plugins/openai_agents/__init__.py

@@ -51,24 +51,23 @@
 - No forked plugin required - uses standard OpenAIAgentsPlugin
 """
 
-from agentex.lib.core.temporal.plugins.openai_agents.streaming_model import (
+from agentex.lib.core.temporal.plugins.openai_agents.models.streaming_model import (
     StreamingModel,
     StreamingModelProvider,
 )
-# Import TracingModelProvider from the adk providers module
-from agentex.lib.adk.providers._modules.temporal_tracing import (
+from agentex.lib.core.temporal.plugins.openai_agents.models.tracing_model import (
     TracingModelProvider,
 )
-from agentex.lib.core.temporal.plugins.openai_agents.context_interceptor import (
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import (
     ContextInterceptor,
     streaming_task_id,
     streaming_trace_id,
     streaming_parent_span_id,
 )
-from agentex.lib.core.temporal.plugins.openai_agents.hooks import (
+from agentex.lib.core.temporal.plugins.openai_agents.hooks.hooks import (
     TemporalStreamingHooks,
 )
-from agentex.lib.core.temporal.plugins.openai_agents.activities import (
+from agentex.lib.core.temporal.plugins.openai_agents.hooks.activities import (
     stream_lifecycle_content,
 )
 

src/agentex/lib/core/temporal/plugins/openai_agents/hooks/__init__.py

@@ -0,0 +1,17 @@
+"""Temporal streaming hooks and activities for OpenAI Agents SDK.
+
+This module provides hooks for streaming agent lifecycle events and
+activities for streaming content to the AgentEx UI.
+"""
+
+from agentex.lib.core.temporal.plugins.openai_agents.hooks.hooks import (
+    TemporalStreamingHooks,
+)
+from agentex.lib.core.temporal.plugins.openai_agents.hooks.activities import (
+    stream_lifecycle_content,
+)
+
+__all__ = [
+    "TemporalStreamingHooks",
+    "stream_lifecycle_content",
+]

src/agentex/lib/core/temporal/plugins/openai_agents/activities.py renamed to src/agentex/lib/core/temporal/plugins/openai_agents/hooks/activities.py

@@ -13,7 +13,7 @@
 
 from agentex.types.text_content import TextContent
 from agentex.types.task_message_content import ToolRequestContent, ToolResponseContent
-from agentex.lib.core.temporal.plugins.openai_agents.activities import stream_lifecycle_content
+from agentex.lib.core.temporal.plugins.openai_agents.hooks.activities import stream_lifecycle_content
 
 
 class TemporalStreamingHooks(RunHooks):

src/agentex/lib/core/temporal/plugins/openai_agents/hooks.py renamed to src/agentex/lib/core/temporal/plugins/openai_agents/hooks/hooks.py

@@ -0,0 +1,19 @@
+"""Temporal interceptors for OpenAI Agents SDK integration.
+
+This module provides interceptors for threading context (task_id, trace_id, parent_span_id)
+from workflows to activities in Temporal.
+"""
+
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import (
+    ContextInterceptor,
+    streaming_task_id,
+    streaming_trace_id,
+    streaming_parent_span_id,
+)
+
+__all__ = [
+    "ContextInterceptor",
+    "streaming_task_id",
+    "streaming_trace_id",
+    "streaming_parent_span_id",
+]

src/agentex/lib/core/temporal/plugins/openai_agents/interceptors/__init__.py

@@ -0,0 +1,23 @@
+"""Model providers for Temporal OpenAI Agents SDK integration.
+
+This module provides model implementations that add streaming and tracing
+capabilities to standard OpenAI models when running in Temporal workflows/activities.
+"""
+
+from agentex.lib.core.temporal.plugins.openai_agents.models.streaming_model import (
+    StreamingModel,
+    StreamingModelProvider,
+)
+from agentex.lib.core.temporal.plugins.openai_agents.models.tracing_model import (
+    TracingModelProvider,
+    TracingResponsesModel,
+    TracingChatCompletionsModel,
+)
+
+__all__ = [
+    "StreamingModel",
+    "StreamingModelProvider",
+    "TracingModelProvider",
+    "TracingResponsesModel",
+    "TracingChatCompletionsModel",
+]

src/agentex/lib/core/temporal/plugins/openai_agents/context_interceptor.py renamed to src/agentex/lib/core/temporal/plugins/openai_agents/interceptors/context_interceptor.py

@@ -59,9 +59,9 @@
 from agentex.types.task_message_delta import ReasoningSummaryDelta, TextDelta, ReasoningContentDelta
 from agentex.types.task_message_content import TextContent, ReasoningContent
 from agentex.types.task_message_update import StreamTaskMessageDelta, StreamTaskMessageFull
-from agentex.lib.core.temporal.plugins.openai_agents.context_interceptor import streaming_task_id
-from agentex.lib.core.temporal.plugins.openai_agents.context_interceptor import streaming_trace_id
-from agentex.lib.core.temporal.plugins.openai_agents.context_interceptor import streaming_parent_span_id
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import streaming_task_id
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import streaming_trace_id
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import streaming_parent_span_id
 # Create logger for this module
 logger = logging.getLogger("agentex.temporal.streaming")
 

src/agentex/lib/core/temporal/plugins/openai_agents/models/__init__.py

@@ -34,7 +34,7 @@
 from agentex.lib.core.tracing.tracer import AsyncTracer
 
 # Import context variables from the interceptor
-from agentex.lib.core.temporal.plugins.openai_agents.context_interceptor import (
+from agentex.lib.core.temporal.plugins.openai_agents.interceptors.context_interceptor import (
     streaming_task_id,
     streaming_trace_id,
     streaming_parent_span_id,