a7d6d88f6f
CI / changes (push) Has been cancelled
CI / cd libs/checkpoint (push) Has been cancelled
CI / cd libs/checkpoint-conformance (push) Has been cancelled
CI / cd libs/checkpoint-postgres (push) Has been cancelled
CI / cd libs/checkpoint-sqlite (push) Has been cancelled
CI / cd libs/cli (push) Has been cancelled
CI / cd libs/prebuilt (push) Has been cancelled
CI / cd libs/sdk-py (push) Has been cancelled
CI / cd libs/langgraph (push) Has been cancelled
CI / Check SDK methods matching (push) Has been cancelled
CI / Check CLI schema hasn't changed #3.13 (push) Has been cancelled
CI / CLI integration test (push) Has been cancelled
CI / sdk-py integration test (push) Has been cancelled
CI / CI Success (push) Has been cancelled
baseline / benchmark (push) Has been cancelled
Deploy Redirects to GitHub Pages / deploy (push) Has been cancelled
59 lines
1.8 KiB
Python
59 lines
1.8 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
|
|
import httpx
|
|
import pytest
|
|
|
|
from langgraph_sdk.client import (
|
|
HttpClient,
|
|
SyncHttpClient,
|
|
SyncThreadsClient,
|
|
ThreadsClient,
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_async_threads_update_return_minimal():
|
|
async def handler(request: httpx.Request) -> httpx.Response:
|
|
assert request.method == "PATCH"
|
|
assert request.url.path == "/threads/thread_123"
|
|
assert request.headers["Prefer"] == "return=minimal"
|
|
assert json.loads(request.content) == {"metadata": {"foo": "bar"}}
|
|
return httpx.Response(204)
|
|
|
|
transport = httpx.MockTransport(handler)
|
|
async with httpx.AsyncClient(
|
|
transport=transport, base_url="https://example.com"
|
|
) as client:
|
|
http_client = HttpClient(client)
|
|
threads_client = ThreadsClient(http_client)
|
|
result = await threads_client.update(
|
|
"thread_123",
|
|
metadata={"foo": "bar"},
|
|
return_minimal=True,
|
|
)
|
|
|
|
assert result is None
|
|
|
|
|
|
def test_sync_threads_update_return_minimal():
|
|
def handler(request: httpx.Request) -> httpx.Response:
|
|
assert request.method == "PATCH"
|
|
assert request.url.path == "/threads/thread_123"
|
|
assert request.headers["Prefer"] == "return=minimal"
|
|
assert json.loads(request.content) == {"metadata": {"foo": "bar"}}
|
|
return httpx.Response(204)
|
|
|
|
transport = httpx.MockTransport(handler)
|
|
with httpx.Client(transport=transport, base_url="https://example.com") as client:
|
|
http_client = SyncHttpClient(client)
|
|
threads_client = SyncThreadsClient(http_client)
|
|
result = threads_client.update(
|
|
"thread_123",
|
|
metadata={"foo": "bar"},
|
|
return_minimal=True,
|
|
)
|
|
|
|
assert result is None
|