feat: sync latest ARR implementation

This commit is contained in:
Wyndham ARR
2026-07-31 15:11:42 +08:00
parent d6f8a747fa
commit bf7939dd1a
185 changed files with 17527 additions and 2260 deletions

View File

@@ -142,6 +142,7 @@ class OpenAgentAPIClient:
*,
idempotency_key: Optional[str] = None,
metadata: Optional[Dict[str, Any]] = None,
include_trace: bool = False,
) -> Iterator[OpenAgentEvent]:
payload = self._message_payload(message, idempotency_key=idempotency_key, metadata=metadata)
path = f"/api/open/agent-sessions/{self._path_segment(session_id)}/messages/stream"
@@ -149,6 +150,7 @@ class OpenAgentAPIClient:
with self._client.stream(
"POST",
self._url(path),
params={"include_trace": "true"} if include_trace else None,
headers=self._headers(include_csrf=True, accept="text/event-stream"),
json=payload,
timeout=self.timeout,
@@ -332,4 +334,3 @@ class OpenAgentAPIClient:
response_text,
request_id=request_id,
)

View File

@@ -67,6 +67,7 @@ class OpenAgentService:
message_id: Optional[str] = None,
external_subject_id: Optional[str] = None,
metadata: Optional[Dict[str, Any]] = None,
include_trace: bool = False,
) -> Iterator[OpenAgentEvent]:
conversation_id = self._required("conversation_id", conversation_id)
message = self._message(message)
@@ -81,6 +82,7 @@ class OpenAgentService:
message,
idempotency_key=idempotency_key,
metadata=self._metadata(metadata),
include_trace=include_trace,
)
def send_message(