agentex-sdk 0.1.0__py3-none-any.whl

agentex/types/agent_rpc_result.py

@@ -0,0 +1,90 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Union, Optional
+from typing_extensions import Literal, TypeAlias
+
+from .task import Task
+from .event import Event
+from .._models import BaseModel
+from .task_message import TaskMessage
+from .task_message_delta import TaskMessageDelta
+from .task_message_content import TaskMessageContent
+
+__all__ = [
+    "AgentRpcResult",
+    "StreamTaskMessageStart",
+    "StreamTaskMessageDelta",
+    "StreamTaskMessageFull",
+    "StreamTaskMessageDone",
+]
+
+
+class StreamTaskMessageStart(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["start"]] = None
+
+
+class StreamTaskMessageDelta(BaseModel):
+    delta: Optional[TaskMessageDelta] = None
+    """Delta for text updates"""
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["delta"]] = None
+
+
+class StreamTaskMessageFull(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["full"]] = None
+
+
+class StreamTaskMessageDone(BaseModel):
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["done"]] = None
+
+
+AgentRpcResult: TypeAlias = Union[
+    List[TaskMessage],
+    StreamTaskMessageStart,
+    StreamTaskMessageDelta,
+    StreamTaskMessageFull,
+    StreamTaskMessageDone,
+    Task,
+    Event,
+    None,
+]

agentex/types/agent_task_tracker.py

@@ -0,0 +1,34 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["AgentTaskTracker"]
+
+
+class AgentTaskTracker(BaseModel):
+    id: str
+    """The UUID of the agent task tracker"""
+
+    agent_id: str
+    """The UUID of the agent"""
+
+    created_at: datetime
+    """When the agent task tracker was created"""
+
+    task_id: str
+    """The UUID of the task"""
+
+    last_processed_event_id: Optional[str] = None
+    """The last processed event ID"""
+
+    status: Optional[str] = None
+    """Processing status"""
+
+    status_reason: Optional[str] = None
+    """Optional status reason"""
+
+    updated_at: Optional[datetime] = None
+    """When the agent task tracker was last updated"""

agentex/types/data_content.py

@@ -0,0 +1,30 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+from .message_style import MessageStyle
+from .message_author import MessageAuthor
+
+__all__ = ["DataContent"]
+
+
+class DataContent(BaseModel):
+    author: MessageAuthor
+    """
+    The role of the messages author, in this case `system`, `user`, `assistant`, or
+    `tool`.
+    """
+
+    data: Dict[str, object]
+    """The contents of the data message."""
+
+    style: MessageStyle = "static"
+    """The style of the message.
+
+    This is used by the client to determine how to display the message.
+    """
+
+    type: Literal["data"] = "data"
+    """The type of the message, in this case `data`."""

agentex/types/data_content_param.py

@@ -0,0 +1,31 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict
+from typing_extensions import Literal, Required, TypedDict
+
+from .message_style import MessageStyle
+from .message_author import MessageAuthor
+
+__all__ = ["DataContentParam"]
+
+
+class DataContentParam(TypedDict, total=False):
+    author: Required[MessageAuthor]
+    """
+    The role of the messages author, in this case `system`, `user`, `assistant`, or
+    `tool`.
+    """
+
+    data: Required[Dict[str, object]]
+    """The contents of the data message."""
+
+    style: MessageStyle
+    """The style of the message.
+
+    This is used by the client to determine how to display the message.
+    """
+
+    type: Literal["data"]
+    """The type of the message, in this case `data`."""

agentex/types/data_delta.py

@@ -0,0 +1,14 @@
+# 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__ = ["DataDelta"]
+
+
+class DataDelta(BaseModel):
+    data_delta: Optional[str] = None
+
+    type: Optional[Literal["data"]] = None

agentex/types/event.py

@@ -0,0 +1,29 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Optional
+from datetime import datetime
+
+from .._models import BaseModel
+from .task_message_content import TaskMessageContent
+
+__all__ = ["Event"]
+
+
+class Event(BaseModel):
+    id: str
+    """The UUID of the event"""
+
+    agent_id: str
+    """The UUID of the agent that the event belongs to"""
+
+    sequence_id: int
+    """The sequence ID of the event"""
+
+    task_id: str
+    """The UUID of the task that the event belongs to"""
+
+    content: Optional[TaskMessageContent] = None
+    """The content of the event"""
+
+    created_at: Optional[datetime] = None
+    """The timestamp of the event"""

agentex/types/event_list_params.py

@@ -0,0 +1,22 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Required, TypedDict
+
+__all__ = ["EventListParams"]
+
+
+class EventListParams(TypedDict, total=False):
+    agent_id: Required[str]
+    """The agent ID to filter events by"""
+
+    task_id: Required[str]
+    """The task ID to filter events by"""
+
+    last_processed_event_id: Optional[str]
+    """Optional event ID to get events after this ID"""
+
+    limit: Optional[int]
+    """Optional limit on number of results"""

agentex/types/event_list_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .event import Event
+
+__all__ = ["EventListResponse"]
+
+EventListResponse: TypeAlias = List[Event]

agentex/types/message_author.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing_extensions import Literal, TypeAlias
+
+__all__ = ["MessageAuthor"]
+
+MessageAuthor: TypeAlias = Literal["user", "agent"]

agentex/types/message_create_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+from .task_message_content_param import TaskMessageContentParam
+
+__all__ = ["MessageCreateParams"]
+
+
+class MessageCreateParams(TypedDict, total=False):
+    content: Required[TaskMessageContentParam]
+
+    task_id: Required[str]
+
+    streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]]

agentex/types/message_list_params.py

@@ -0,0 +1,14 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Required, TypedDict
+
+__all__ = ["MessageListParams"]
+
+
+class MessageListParams(TypedDict, total=False):
+    task_id: Required[str]
+
+    limit: Optional[int]

agentex/types/message_list_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .task_message import TaskMessage
+
+__all__ = ["MessageListResponse"]
+
+MessageListResponse: TypeAlias = List[TaskMessage]

agentex/types/message_style.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing_extensions import Literal, TypeAlias
+
+__all__ = ["MessageStyle"]
+
+MessageStyle: TypeAlias = Literal["static", "active"]

agentex/types/message_update_params.py

@@ -0,0 +1,18 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import Literal, Required, TypedDict
+
+from .task_message_content_param import TaskMessageContentParam
+
+__all__ = ["MessageUpdateParams"]
+
+
+class MessageUpdateParams(TypedDict, total=False):
+    content: Required[TaskMessageContentParam]
+
+    task_id: Required[str]
+
+    streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]]

agentex/types/messages/__init__.py

@@ -0,0 +1,8 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from .batch_create_params import BatchCreateParams as BatchCreateParams
+from .batch_update_params import BatchUpdateParams as BatchUpdateParams
+from .batch_create_response import BatchCreateResponse as BatchCreateResponse
+from .batch_update_response import BatchUpdateResponse as BatchUpdateResponse

agentex/types/messages/batch_create_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Iterable
+from typing_extensions import Required, TypedDict
+
+from ..task_message_content_param import TaskMessageContentParam
+
+__all__ = ["BatchCreateParams"]
+
+
+class BatchCreateParams(TypedDict, total=False):
+    contents: Required[Iterable[TaskMessageContentParam]]
+
+    task_id: Required[str]

agentex/types/messages/batch_create_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from ..task_message import TaskMessage
+
+__all__ = ["BatchCreateResponse"]
+
+BatchCreateResponse: TypeAlias = List[TaskMessage]

agentex/types/messages/batch_update_params.py

@@ -0,0 +1,16 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict
+from typing_extensions import Required, TypedDict
+
+from ..task_message_content_param import TaskMessageContentParam
+
+__all__ = ["BatchUpdateParams"]
+
+
+class BatchUpdateParams(TypedDict, total=False):
+    task_id: Required[str]
+
+    updates: Required[Dict[str, TaskMessageContentParam]]

agentex/types/messages/batch_update_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from ..task_message import TaskMessage
+
+__all__ = ["BatchUpdateResponse"]
+
+BatchUpdateResponse: TypeAlias = List[TaskMessage]

agentex/types/shared/__init__.py

@@ -0,0 +1,3 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from .task_message_update import TaskMessageUpdate as TaskMessageUpdate

agentex/types/shared/task_message_update.py

@@ -0,0 +1,83 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Union, Optional
+from typing_extensions import Literal, Annotated, TypeAlias
+
+from ..._utils import PropertyInfo
+from ..._models import BaseModel
+from ..task_message import TaskMessage
+from ..task_message_delta import TaskMessageDelta
+from ..task_message_content import TaskMessageContent
+
+__all__ = [
+    "TaskMessageUpdate",
+    "StreamTaskMessageStart",
+    "StreamTaskMessageDelta",
+    "StreamTaskMessageFull",
+    "StreamTaskMessageDone",
+]
+
+
+class StreamTaskMessageStart(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["start"]] = None
+
+
+class StreamTaskMessageDelta(BaseModel):
+    delta: Optional[TaskMessageDelta] = None
+    """Delta for text updates"""
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["delta"]] = None
+
+
+class StreamTaskMessageFull(BaseModel):
+    content: TaskMessageContent
+
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["full"]] = None
+
+
+class StreamTaskMessageDone(BaseModel):
+    index: Optional[int] = None
+
+    parent_task_message: Optional[TaskMessage] = None
+    """Represents a message in the agent system.
+
+    This entity is used to store messages in MongoDB, with each message associated
+    with a specific task.
+    """
+
+    type: Optional[Literal["done"]] = None
+
+
+TaskMessageUpdate: TypeAlias = Annotated[
+    Union[StreamTaskMessageStart, StreamTaskMessageDelta, StreamTaskMessageFull, StreamTaskMessageDone],
+    PropertyInfo(discriminator="type"),
+]

agentex/types/span.py

@@ -0,0 +1,36 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, List, Union, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["Span"]
+
+
+class Span(BaseModel):
+    id: str
+
+    name: str
+    """Name that describes what operation this span represents"""
+
+    start_time: datetime
+    """The time the span started"""
+
+    trace_id: str
+    """Unique identifier for the trace this span belongs to"""
+
+    data: Union[Dict[str, object], List[Dict[str, object]], None] = None
+    """Any additional metadata or context for the span"""
+
+    end_time: Optional[datetime] = None
+    """The time the span ended"""
+
+    input: Union[Dict[str, object], List[Dict[str, object]], None] = None
+    """Input parameters or data for the operation"""
+
+    output: Union[Dict[str, object], List[Dict[str, object]], None] = None
+    """Output data resulting from the operation"""
+
+    parent_id: Optional[str] = None
+    """ID of the parent span if this is a child span in a trace"""

agentex/types/span_create_params.py

@@ -0,0 +1,40 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union, Iterable, Optional
+from datetime import datetime
+from typing_extensions import Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["SpanCreateParams"]
+
+
+class SpanCreateParams(TypedDict, total=False):
+    name: Required[str]
+    """Name that describes what operation this span represents"""
+
+    start_time: Required[Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]]
+    """The time the span started"""
+
+    trace_id: Required[str]
+    """Unique identifier for the trace this span belongs to"""
+
+    id: Optional[str]
+    """Unique identifier for the span. If not provided, an ID will be generated."""
+
+    data: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Any additional metadata or context for the span"""
+
+    end_time: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
+    """The time the span ended"""
+
+    input: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Input parameters or data for the operation"""
+
+    output: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Output data resulting from the operation"""
+
+    parent_id: Optional[str]
+    """ID of the parent span if this is a child span in a trace"""

agentex/types/span_list_params.py

@@ -0,0 +1,12 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Optional
+from typing_extensions import TypedDict
+
+__all__ = ["SpanListParams"]
+
+
+class SpanListParams(TypedDict, total=False):
+    trace_id: Optional[str]

agentex/types/span_list_response.py

@@ -0,0 +1,10 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List
+from typing_extensions import TypeAlias
+
+from .span import Span
+
+__all__ = ["SpanListResponse"]
+
+SpanListResponse: TypeAlias = List[Span]

agentex/types/span_update_params.py

@@ -0,0 +1,37 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union, Iterable, Optional
+from datetime import datetime
+from typing_extensions import Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["SpanUpdateParams"]
+
+
+class SpanUpdateParams(TypedDict, total=False):
+    data: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Any additional metadata or context for the span"""
+
+    end_time: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
+    """The time the span ended"""
+
+    input: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Input parameters or data for the operation"""
+
+    name: Optional[str]
+    """Name that describes what operation this span represents"""
+
+    output: Union[Dict[str, object], Iterable[Dict[str, object]], None]
+    """Output data resulting from the operation"""
+
+    parent_id: Optional[str]
+    """ID of the parent span if this is a child span in a trace"""
+
+    start_time: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
+    """The time the span started"""
+
+    trace_id: Optional[str]
+    """Unique identifier for the trace this span belongs to"""

agentex/types/state.py

@@ -0,0 +1,25 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import Dict, Optional
+from datetime import datetime
+
+from .._models import BaseModel
+
+__all__ = ["State"]
+
+
+class State(BaseModel):
+    id: str
+    """The task state's unique id"""
+
+    agent_id: str
+
+    created_at: datetime
+    """The timestamp when the state was created"""
+
+    state: Dict[str, object]
+
+    task_id: str
+
+    updated_at: Optional[datetime] = None
+    """The timestamp when the state was last updated"""