feat: sync latest ARR implementation
This commit is contained in:
@@ -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,
|
||||
)
|
||||
|
||||
|
||||
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user