Compare commits

..
Author SHA1 Message Date
hinthornwandGitHub f03f8ea36b Update OpenAPI spec from LangGraph API v0.2.135 2025-08-19 09:05:40 +00:00
6 changed files with 430 additions and 152 deletions
File diff suppressed because it is too large Load Diff
+9 -9
View File
@@ -733,9 +733,9 @@ class RemoteGraph(PregelProtocol):
interrupt_after=interrupt_after,
stream_subgraphs=subgraphs or stream is not None,
if_not_exists="create",
headers=(
_merge_tracing_headers(headers) if self.distributed_tracing else headers
),
headers=_merge_tracing_headers(headers)
if self.distributed_tracing
else headers,
params=params,
**kwargs,
):
@@ -842,9 +842,9 @@ class RemoteGraph(PregelProtocol):
interrupt_after=interrupt_after,
stream_subgraphs=subgraphs or stream is not None,
if_not_exists="create",
headers=(
_merge_tracing_headers(headers) if self.distributed_tracing else headers
),
headers=_merge_tracing_headers(headers)
if self.distributed_tracing
else headers,
params=params,
**kwargs,
):
@@ -996,11 +996,11 @@ class RemoteGraph(PregelProtocol):
def _merge_tracing_headers(headers: dict[str, str] | None) -> dict[str, str] | None:
if rt := ls.get_current_run_tree():
tracing_headers = rt.to_headers()
baggage = tracing_headers.pop("baggage")
if headers:
if "baggage" in headers:
tracing_headers["baggage"] = (
f"{headers['baggage']},{tracing_headers['baggage']}"
)
baggage = headers["baggage"] + "," + baggage
tracing_headers["baggage"] = baggage
headers.update(tracing_headers)
else:
headers = tracing_headers
+1 -1
View File
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
[project]
name = "langgraph"
version = "0.6.7"
version = "0.6.5"
description = "Building stateful, multi-actor applications with LLMs"
authors = []
requires-python = ">=3.9"
+7 -12
View File
@@ -1183,10 +1183,7 @@ async def test_remote_graph_stream_messages_tuple(
@pytest.mark.anyio
@pytest.mark.parametrize("distributed_tracing", [False, True])
@pytest.mark.parametrize("stream", [False, True])
@pytest.mark.parametrize("headers", [None, {"foo": "bar"}])
async def test_include_headers(
distributed_tracing: bool, stream: bool, headers: dict[str, str] | None
):
async def test_include_headers(distributed_tracing: bool, stream: bool):
mock_async_client = MagicMock()
async_iter = MagicMock()
return_value = [
@@ -1216,7 +1213,7 @@ async def test_include_headers(
async for _ in remote_pregel.astream(
{"input": {"messages": [{"type": "human", "content": "hello"}]}},
config,
headers=headers,
headers={"foo": "bar"},
):
pass
@@ -1224,14 +1221,12 @@ async def test_include_headers(
await remote_pregel.ainvoke(
{"input": {"messages": [{"type": "human", "content": "hello"}]}},
config,
headers=headers,
headers={"foo": "bar"},
)
expected = headers.copy() if headers else None
expected = {"foo": "bar"}
if distributed_tracing:
if expected is None:
expected = {}
expected["langsmith-trace"] = AnyStr()
expected["baggage"] = AnyStr("langsmith-metadata=")
expected["baggage"] = AnyStr()
assert astream_mock.call_args.kwargs["headers"] == expected
stream_mock.assert_not_called()
@@ -1242,7 +1237,7 @@ async def test_include_headers(
for _ in remote_pregel.stream(
{"input": {"messages": [{"type": "human", "content": "hello"}]}},
config,
headers=headers,
headers={"foo": "bar"},
):
pass
@@ -1250,6 +1245,6 @@ async def test_include_headers(
remote_pregel.invoke(
{"input": {"messages": [{"type": "human", "content": "hello"}]}},
config,
headers=headers,
headers={"foo": "bar"},
)
assert stream_mock.call_args.kwargs["headers"] == expected
+1 -1
View File
@@ -1269,7 +1269,7 @@ wheels = [
[[package]]
name = "langgraph"
version = "0.6.7"
version = "0.6.5"
source = { editable = "." }
dependencies = [
{ name = "langchain-core" },
+1 -1
View File
@@ -316,7 +316,7 @@ wheels = [
[[package]]
name = "langgraph"
version = "0.6.6"
version = "0.6.5"
source = { editable = "../langgraph" }
dependencies = [
{ name = "langchain-core" },