Merge pull request #78 from scaleapi/fs/reasoning

Intermediate Streaming Reasoning

Author: IdkwhatImD0ing

examples/tutorials/00_sync/000_hello_acp/project/acp.py

@@ -2,7 +2,7 @@
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 
-from agentex.lib.types.task_message_updates import TaskMessageUpdate
+from agentex.types.task_message_update import TaskMessageUpdate
 from agentex.types.task_message import TaskMessageContent
 from agentex.types.task_message_content import TextContent
 from agentex.lib.utils.logging import make_logger

examples/tutorials/00_sync/010_multiturn/project/acp.py

@@ -5,7 +5,7 @@
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 from agentex.lib.types.llm_messages import AssistantMessage, LLMConfig, SystemMessage, UserMessage
-from agentex.lib.types.task_message_updates import TaskMessageUpdate
+from agentex.types.task_message_update import TaskMessageUpdate
 from agentex.types.task_message import TaskMessageContent
 from agentex.types.task_message_content import TextContent
 from agentex.lib.utils.model_utils import BaseModel

examples/tutorials/00_sync/020_streaming/project/acp.py

@@ -5,7 +5,8 @@
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 from agentex.lib.types.llm_messages import AssistantMessage, LLMConfig, SystemMessage, UserMessage
-from agentex.lib.types.task_message_updates import StreamTaskMessageDelta, StreamTaskMessageDone, StreamTaskMessageFull, TaskMessageUpdate, TextDelta
+from agentex.types.task_message_update import StreamTaskMessageDelta, StreamTaskMessageDone, StreamTaskMessageFull, TaskMessageUpdate
+from agentex.types.task_message_delta import TextDelta
 from agentex.lib.utils.model_utils import BaseModel
 from agentex.types.task_message_content import TaskMessageContent, TextContent
 

examples/tutorials/10_agentic/00_base/040_other_sdks/project/acp.py

@@ -8,11 +8,11 @@
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import CancelTaskParams, CreateTaskParams, SendEventParams
 from agentex.lib.types.fastacp import AgenticACPConfig
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     StreamTaskMessageDelta, 
     StreamTaskMessageFull,
-    TextDelta,
 )
+from agentex.types.task_message_delta import TextDelta
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
 from agentex.types.text_content import TextContent

examples/tutorials/10_agentic/10_temporal/010_agent_chat/dev.ipynb

@@ -234,14 +234,14 @@ async def on_task_event_send(self, params: SendEventParams) -> None:
                     "to provide accurate and well-reasoned responses."
                 ),
                 parent_span_id=span.id if span else None,
-                model="o4-mini",
+                model="gpt-5-mini",
                 model_settings=ModelSettings(
                     # Include reasoning items in the response (IDs, summaries)
                     # response_include=["reasoning.encrypted_content"],
                     # Ask the model to include a short reasoning summary
-                    reasoning=Reasoning(effort="medium", summary="auto"),
+                    reasoning=Reasoning(effort="medium", summary="detailed"),
                 ),
-                tools=[CALCULATOR_TOOL],
+                # tools=[CALCULATOR_TOOL],
             )
             if self._state:
                 # Update the state with the final input list if available

examples/tutorials/10_agentic/10_temporal/010_agent_chat/project/workflow.py

@@ -2,7 +2,7 @@ from typing import AsyncGenerator, Union
 from agentex.lib.sdk.fastacp.fastacp import FastACP
 from agentex.lib.types.acp import SendMessageParams
 
-from agentex.lib.types.task_message_updates import TaskMessageUpdate
+from agentex.types.task_message_update import TaskMessageUpdate
 from agentex.types.task_message_content import TaskMessageContent
 from agentex.types.text_content import TextContent
 from agentex.lib.utils.logging import make_logger

examples/tutorials/10_agentic/10_temporal/010_agent_chat/uv.lock

@@ -4,7 +4,7 @@
 from agentex import AsyncAgentex
 from agentex.lib.core.services.adk.streaming import StreamingService
 from agentex.lib.core.tracing.tracer import AsyncTracer
-from agentex.lib.types.task_message_updates import StreamTaskMessageFull, TaskMessageUpdate
+from agentex.types.task_message_update import StreamTaskMessageFull, TaskMessageUpdate
 from agentex.types.task_message import TaskMessage, TaskMessageContent
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.temporal import heartbeat_if_in_workflow

src/agentex/lib/cli/templates/sync/project/acp.py.j2

@@ -8,11 +8,11 @@
     Completion,
     LLMConfig,
 )
-from agentex.lib.types.task_message_updates import (
+from agentex.types.task_message_update import (
     StreamTaskMessageDelta,
     StreamTaskMessageFull,
-    TextDelta,
 )
+from agentex.types.task_message_delta import TextDelta
 from agentex.types.task_message import TaskMessage
 from agentex.types.task_message_content import TextContent
 from agentex.lib.utils import logging
@@ -112,6 +112,7 @@ async def chat_completion_auto_send(
                         update=StreamTaskMessageFull(
                             parent_task_message=streaming_context.task_message,
                             content=final_content,
+                            type="full",
                         ),
                     )
                 else:
@@ -221,7 +222,8 @@ async def chat_completion_stream_auto_send(
                             await streaming_context.stream_update(
                                 update=StreamTaskMessageDelta(
                                     parent_task_message=streaming_context.task_message,
-                                    delta=TextDelta(text_delta=delta),
+                                    delta=TextDelta(text_delta=delta, type="text"),
+                                    type="delta",
                                 ),
                             )
                             heartbeat_if_in_workflow("content chunk streamed")
@@ -244,6 +246,7 @@ async def chat_completion_stream_auto_send(
                         update=StreamTaskMessageFull(
                             parent_task_message=streaming_context.task_message,
                             content=final_content,
+                            type="full",
                         ),
                     )