agentex-sdk 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.3.0"  # x-release-please-version
+__version__ = "0.4.0"  # x-release-please-version

agentex/lib/core/services/adk/providers/openai.py

@@ -29,6 +29,7 @@ from agentex.lib.types.task_message_updates import (
 )
 from agentex.types.task_message_content import (
     TextContent,
+    ReasoningContent,
     ToolRequestContent,
     ToolResponseContent,
 )
@@ -647,6 +648,31 @@ class OpenAIService:
                                             content=tool_response_content,
                                         ),
                                     )
+                            
+                            elif event.item.type == "reasoning_item":
+                                # Handle reasoning items
+                                reasoning_item = event.item.raw_item
+                                
+                                reasoning_content = ReasoningContent(
+                                    author="agent",
+                                    summary=[summary.text for summary in reasoning_item.summary],
+                                    content=[content.text for content in reasoning_item.content] if reasoning_item.content else None,
+                                )
+
+                                # Create reasoning content using streaming context (immediate completion)
+                                async with (
+                                    self.streaming_service.streaming_task_message_context(
+                                        task_id=task_id,
+                                        initial_content=reasoning_content,
+                                    ) as streaming_context
+                                ):
+                                    # The message has already been persisted, but we still need to send an update
+                                    await streaming_context.stream_update(
+                                        update=StreamTaskMessageFull(
+                                            parent_task_message=streaming_context.task_message,
+                                            content=reasoning_content,
+                                        ),
+                                    )
 
                         elif event.type == "raw_response_event":
                             if isinstance(event.data, ResponseTextDeltaEvent):

agentex/types/__init__.py

@@ -51,7 +51,9 @@ from .message_update_params import MessageUpdateParams as MessageUpdateParams
 from .tool_response_content import ToolResponseContent as ToolResponseContent
 from .tracker_list_response import TrackerListResponse as TrackerListResponse
 from .tracker_update_params import TrackerUpdateParams as TrackerUpdateParams
+from .reasoning_content_delta import ReasoningContentDelta as ReasoningContentDelta
 from .reasoning_content_param import ReasoningContentParam as ReasoningContentParam
+from .reasoning_summary_delta import ReasoningSummaryDelta as ReasoningSummaryDelta
 from .agent_rpc_by_name_params import AgentRpcByNameParams as AgentRpcByNameParams
 from .task_message_content_param import TaskMessageContentParam as TaskMessageContentParam
 from .tool_request_content_param import ToolRequestContentParam as ToolRequestContentParam

agentex/types/reasoning_content_delta.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["ReasoningContentDelta"]
+
+
+class ReasoningContentDelta(BaseModel):
+    content_index: int
+
+    content_delta: Optional[str] = None
+
+    type: Optional[Literal["reasoning_content"]] = None

agentex/types/reasoning_summary_delta.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+
+__all__ = ["ReasoningSummaryDelta"]
+
+
+class ReasoningSummaryDelta(BaseModel):
+    summary_index: int
+
+    summary_delta: Optional[str] = None
+
+    type: Optional[Literal["reasoning_summary"]] = None

agentex/types/task_message_delta.py

@@ -8,9 +8,12 @@ from .data_delta import DataDelta
 from .text_delta import TextDelta
 from .tool_request_delta import ToolRequestDelta
 from .tool_response_delta import ToolResponseDelta
+from .reasoning_content_delta import ReasoningContentDelta
+from .reasoning_summary_delta import ReasoningSummaryDelta
 
 __all__ = ["TaskMessageDelta"]
 
 TaskMessageDelta: TypeAlias = Annotated[
-    Union[TextDelta, DataDelta, ToolRequestDelta, ToolResponseDelta], PropertyInfo(discriminator="type")
+    Union[TextDelta, DataDelta, ToolRequestDelta, ToolResponseDelta, ReasoningSummaryDelta, ReasoningContentDelta],
+    PropertyInfo(discriminator="type"),
 ]

{agentex_sdk-0.3.0.dist-info → agentex_sdk-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.3.0
+Version: 0.4.0
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python
@@ -32,6 +32,7 @@ Requires-Dist: kubernetes<29.0.0,>=25.0.0
 Requires-Dist: litellm<2,>=1.66.0
 Requires-Dist: mcp[cli]>=1.4.1
 Requires-Dist: openai-agents!=0.2.3,>=0.0.7
+Requires-Dist: openai>=1.99.9
 Requires-Dist: pydantic<3,>=2.0.0
 Requires-Dist: pytest-asyncio>=1.0.0
 Requires-Dist: pytest>=8.4.0

{agentex_sdk-0.3.0.dist-info → agentex_sdk-0.4.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ agentex/_resource.py,sha256=S1t7wmR5WUvoDIhZjo_x-E7uoTJBynJ3d8tPJMQYdjw,1106
 agentex/_response.py,sha256=Tb9zazsnemO2rTxWtBjAD5WBqlhli5ZaXGbiKgdu5DE,28794
 agentex/_streaming.py,sha256=FNGJExRCF-vTRUZHFKUfoAWFhDGOB3XbioVCF37Jr7E,10104
 agentex/_types.py,sha256=KyKYySGIfHPod2hho1fPxssk5NuVn8C4MeMTtA-lg80,6198
-agentex/_version.py,sha256=dqwMLdLjUtSWnQJ3iaGVlj64-uDW5tpyxJToQihPMIc,159
+agentex/_version.py,sha256=CXKkCgnAp9MvMYadHVymbnJuQa97_ituLGeUp9fx-FI,159
 agentex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/_utils/__init__.py,sha256=PNZ_QJuzZEgyYXqkO1HVhGkj5IU9bglVUcw7H-Knjzw,2062
 agentex/_utils/_logs.py,sha256=LUjFPc3fweSChBUmjhQD8uYmwQAmFMNDuVFKfjYBQfM,777
@@ -133,7 +133,7 @@ agentex/lib/core/services/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 agentex/lib/core/services/adk/acp/acp.py,sha256=Ei6r5CcM1RA7DqfTRpAWCmP7Buw97b3ajQuE4DsdSaQ,7559
 agentex/lib/core/services/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/providers/litellm.py,sha256=7RIVqQ5XaCP2cEL1pr4nl5jfG_0FDMvDTZVKu1iaN3M,9864
-agentex/lib/core/services/adk/providers/openai.py,sha256=Qply4Q7BJqRyukAuNPnFcLXf0eQWACaa5LamGc6xpTY,31559
+agentex/lib/core/services/adk/providers/openai.py,sha256=NLTE4MLSvKeiQ--puuBWaz97C7BbkFtn9n1uxYlkiyg,33203
 agentex/lib/core/services/adk/providers/sgp.py,sha256=9gm-sPNQ_OSTaBzL0onerKhokPk_ZHndaKNO-z16wyQ,3676
 agentex/lib/core/services/adk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/utils/templating.py,sha256=eaXSFq31Y9p5pRD6J6SL4QdTFtxy81dilbF2XXc2JYQ,1889
@@ -237,7 +237,7 @@ agentex/resources/tracker.py,sha256=YxSeiloMwIqrS9nY7SlHclauRA7142qrKw34xgwqYbA,
 agentex/resources/messages/__init__.py,sha256=_J1eusFtr_k6zrAntJSuqx6LWEUBSTrV1OZZh7MaDPE,1015
 agentex/resources/messages/batch.py,sha256=pegCmnjK_J0jek5ChX1pKpq5RmCucTYLbK69H6qGVS4,9629
 agentex/resources/messages/messages.py,sha256=fuFmmGNOjRJVzmYHcfP2trg0yii0n9MPPCpt7F8fDs4,17930
-agentex/types/__init__.py,sha256=qll_xWdoF455M3M1MKOomA3IxIjcRbikO8vrom0FF5A,3666
+agentex/types/__init__.py,sha256=_tsP_hk7b8V3b2wCIw1jjdIsedgmtlI5Jd0ijFtk6NA,3834
 agentex/types/acp_type.py,sha256=Fj-4SzmM6m95ck_ZXtNbcWggHiD9F49bxBLPbl1fxe4,208
 agentex/types/agent.py,sha256=t6CUsKSaK6Gs_X9g0YxSi5G3Hh21FrGVGnAD6MytVBk,981
 agentex/types/agent_list_params.py,sha256=81IWnRZ2rLfHH7GB6VkXShYjb44DO0guG1znJcV3tuI,316
@@ -260,7 +260,9 @@ agentex/types/message_list_response.py,sha256=YYDf-57zLS-E1eX3EZxz7c6XCuBcRBws01
 agentex/types/message_style.py,sha256=nuoXzoDyP3KAQsQeKHaiby1EMxeY-8TJjWr8eMMpQEE,219
 agentex/types/message_update_params.py,sha256=_KpnJ56FNeoIcwodQmAgsweqH1YAeIgYVT2jo9ahUhY,502
 agentex/types/reasoning_content.py,sha256=ieA_EmFlHHo5suNnY6_7Qq3u34h8ehZ0E-vhp32Am1o,915
+agentex/types/reasoning_content_delta.py,sha256=tU-ndHhj9x8Mmn3BWwSouqViYHWzVOlZGYL_QF17wC4,386
 agentex/types/reasoning_content_param.py,sha256=qpuwnaQK-ofl8OtvwxpSSiHCwL7nIY0knX10EgGl0W4,942
+agentex/types/reasoning_summary_delta.py,sha256=J2VY9VF1tLCNTIoN0qQmvZgGX35P9Ndmv1fRbcMl6R4,386
 agentex/types/span.py,sha256=OfNLddf5qjwv52ZrxLmgbXDDkBBovMeG7ovc4sRJReg,1045
 agentex/types/span_create_params.py,sha256=o-M8rzB_6yoLrn3gQJ9Ecc6ZrsKp7uVyILam_wsGrgc,1392
 agentex/types/span_list_params.py,sha256=b7gnqK17rE-UK7bUBSR9xsskI_jgmvGw9SuTsIIna9s,297
@@ -277,7 +279,7 @@ agentex/types/task_list_response.py,sha256=8Q-RIanLmUC9vOuDXoW5gjpZeE-HR6IrBugG7
 agentex/types/task_message.py,sha256=hoLAkiQJd1Fl7EjLof-vZq6zVnDw9SKmnV6V74Po58A,918
 agentex/types/task_message_content.py,sha256=3itFPfk9V3AHggZFpHKHkkzostFpyZ8O4vMnwZjXLis,646
 agentex/types/task_message_content_param.py,sha256=IZxdUJWkUynteRQa-y743fCyum54IR6nqr0uybfUTO4,675
-agentex/types/task_message_delta.py,sha256=nXNA0s38eBzue7PT_kU2Zp64CoVfxxMKAtLFxPPhEsg,547
+agentex/types/task_message_delta.py,sha256=3i1Qw2CcZYxXhofbCJnQDO0QrIIQsLX6P-UAynvc4fs,716
 agentex/types/task_message_update.py,sha256=gevnU4UW67qTzN_OGT-S3E5zbS9KQNehJZF3XhiirtU,2247
 agentex/types/text_content.py,sha256=i6ksdQXIyWosXAm-O_26m_eZA-4x26dgoNdzWnhue9k,1347
 agentex/types/text_content_param.py,sha256=5xtCBUJj6xKGlmpNuWVnBG3BBFQS3HyrKjq6KfpfVz0,1439
@@ -298,8 +300,8 @@ agentex/types/messages/batch_update_params.py,sha256=Ug5CThbD49a8j4qucg04OdmVrp_
 agentex/types/messages/batch_update_response.py,sha256=TbSBe6SuPzjXXWSj-nRjT1JHGBooTshHQQDa1AixQA8,278
 agentex/types/shared/__init__.py,sha256=IKs-Qn5Yja0kFh1G1kDqYZo43qrOu1hSoxlPdN-85dI,149
 agentex/types/shared/delete_response.py,sha256=8qH3zvQXaOHYQSHyXi7UQxdR4miTzR7V9K4zXVsiUyk,215
-agentex_sdk-0.3.0.dist-info/METADATA,sha256=rV595YQu--tZa9w-tUG7bF85GTmnZp_470AAd4mUNjM,15037
-agentex_sdk-0.3.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.3.0.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.3.0.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.3.0.dist-info/RECORD,,
+agentex_sdk-0.4.0.dist-info/METADATA,sha256=5on23JG2ehjAa2mOlvhXNoJgQXoh0MmikrAeaLusAhg,15067
+agentex_sdk-0.4.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.4.0.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.4.0.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.4.0.dist-info/RECORD,,