WEBDIENSTE/langgraph
1
0
Fork 0
mirror of https://github.com/langchain-ai/langgraph.git synced 2026-08-31 12:19:58 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix ids in debug task events

Browse Source
This commit is contained in:
Nuno Campos
2024-09-04 14:31:20 -07:00
parent dc10b3aa8e
commit b8d8a879d8
4 changed files with 26 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
libs/langgraph/langgraph/pregel/debug.py
+2 -8
View File
@@ -1,10 +1,9 @@
import json
from collections import defaultdict
from dataclasses import asdict
from datetime import datetime, timezone
from pprint import pformat
from typing import Any, Iterator, Literal, Mapping, Optional, Sequence, TypedDict, Union
from uuid import UUID, uuid5
from uuid import UUID
from langchain_core.runnables.config import RunnableConfig
from langchain_core.utils.input import get_bolded_text, get_colored_text
@@ -82,17 +81,12 @@ def map_debug_tasks(
if task.config is not None and TAG_HIDDEN in task.config.get("tags", []):
continue
metadata = task.config["metadata"].copy()
metadata.pop("checkpoint_id", None)
yield {
"type": "task",
"timestamp": ts,
"step": step,
"payload": {
"id": str(
uuid5(TASK_NAMESPACE, json.dumps((task.name, step, metadata)))
),
"id": task.id,
"name": task.name,
"input": task.input,
"triggers": task.triggers,
libs/langgraph/poetry.lock
Generated
+3 -3
View File
@@ -1884,13 +1884,13 @@ url = "../checkpoint-sqlite"
[[package]]
name = "langsmith"
version = "0.1.111"
version = "0.1.112"
description = "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform."
optional = false
python-versions = "<4.0,>=3.8.1"
files = [
{file = "langsmith-0.1.111-py3-none-any.whl", hash = "sha256:e5c702764911193c9812fe55136ae01cd0b9ddf5dff0b068ce6fd60eeddbcb40"},
{file = "langsmith-0.1.111.tar.gz", hash = "sha256:bab24fd6125685f588d682693c4a3253e163804242829b1ff902e1a3e984a94c"},
{file = "langsmith-0.1.112-py3-none-any.whl", hash = "sha256:38ecc62df35e55e5aa9a05ecf1f1483d5a4db3275b9da4876ae27364714b3717"},
{file = "langsmith-0.1.112.tar.gz", hash = "sha256:08c47a9497261c332bc4b15e2930d8e751310e63652ecdc78382a5dd07cb42e2"},
]
[package.dependencies]
libs/langgraph/tests/test_pregel.py
+10 -10
View File
@@ -1058,7 +1058,7 @@ def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 0,
"payload": {
"id": "2687f72c-e3a8-5f6f-9afa-047cbf24e923",
"id": AnyStr(),
"name": "one",
"input": 2,
"triggers": ["input"],
@@ -1069,7 +1069,7 @@ def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 0,
"payload": {
"id": "18f52f6a-828d-58a1-a501-53cc0c7af33e",
"id": AnyStr(),
"name": "two",
"input": [12],
"triggers": ["inbox"],
@@ -1104,7 +1104,7 @@ def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "871d6e74-7bb3-565f-a4fe-cef4b8f19b62",
"id": AnyStr(),
"name": "two",
"input": [3],
"triggers": ["inbox"],
@@ -6333,7 +6333,7 @@ def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "592f3430-c17c-5d1c-831f-fecebb2c05bf",
"id": AnyStr(),
"name": "rewrite_query",
"input": {"query": "what is weather in sf", "docs": []},
"triggers": ["start:rewrite_query"],
@@ -6364,7 +6364,7 @@ def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "7db5e9d8-e132-5079-ab99-ced15e67d48b",
"id": AnyStr(),
"name": "retriever_one",
"input": {"query": "query: what is weather in sf", "docs": []},
"triggers": ["rewrite_query"],
@@ -6378,7 +6378,7 @@ def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "96965ed0-2c10-52a1-86eb-081ba6de73b2",
"id": AnyStr(),
"name": "retriever_two",
"input": {"query": "query: what is weather in sf", "docs": []},
"triggers": ["rewrite_query"],
@@ -6437,7 +6437,7 @@ def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 3,
"payload": {
"id": "8959fb57-d0f5-5725-9ac4-ec1c554fb0a0",
"id": AnyStr(),
"name": "qa",
"input": {
"query": "query: what is weather in sf",
@@ -6861,7 +6861,7 @@ def test_branch_then(
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "7b7b0713-e958-5d07-803c-c9910a7cc162",
"id": AnyStr(),
"name": "prepare",
"input": {"my_key": "value", "market": "DE"},
"triggers": ["start:prepare"],
@@ -6914,7 +6914,7 @@ def test_branch_then(
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "dd9f2fa5-ccfa-5d12-81ec-942563056a08",
"id": AnyStr(),
"name": "tool_two_slow",
"input": {"my_key": "value prepared", "market": "DE"},
"triggers": ["branch:prepare:condition:tool_two_slow"],
@@ -6967,7 +6967,7 @@ def test_branch_then(
"timestamp": AnyStr(),
"step": 3,
"payload": {
"id": "9b590c54-15ef-54b1-83a7-140d27b0bc52",
"id": AnyStr(),
"name": "finish",
"input": {"my_key": "value prepared slow", "market": "DE"},
"triggers": ["branch:prepare:condition::then"],
libs/langgraph/tests/test_pregel_async.py
+11 -11
View File
@@ -1306,7 +1306,7 @@ async def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 0,
"payload": {
"id": "2687f72c-e3a8-5f6f-9afa-047cbf24e923",
"id": AnyStr(),
"name": "one",
"input": 2,
"triggers": ["input"],
@@ -1317,7 +1317,7 @@ async def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 0,
"payload": {
"id": "18f52f6a-828d-58a1-a501-53cc0c7af33e",
"id": AnyStr(),
"name": "two",
"input": [12],
"triggers": ["inbox"],
@@ -1352,7 +1352,7 @@ async def test_invoke_two_processes_in_dict_out(mocker: MockerFixture) -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "871d6e74-7bb3-565f-a4fe-cef4b8f19b62",
"id": AnyStr(),
"name": "two",
"input": [3],
"triggers": ["inbox"],
@@ -5019,7 +5019,7 @@ async def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "592f3430-c17c-5d1c-831f-fecebb2c05bf",
"id": AnyStr(),
"name": "rewrite_query",
"input": {"query": "what is weather in sf", "docs": []},
"triggers": ["start:rewrite_query"],
@@ -5050,7 +5050,7 @@ async def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "7db5e9d8-e132-5079-ab99-ced15e67d48b",
"id": AnyStr(),
"name": "retriever_one",
"input": {"query": "query: what is weather in sf", "docs": []},
"triggers": ["rewrite_query"],
@@ -5064,7 +5064,7 @@ async def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "96965ed0-2c10-52a1-86eb-081ba6de73b2",
"id": AnyStr(),
"name": "retriever_two",
"input": {"query": "query: what is weather in sf", "docs": []},
"triggers": ["rewrite_query"],
@@ -5123,7 +5123,7 @@ async def test_in_one_fan_out_out_one_graph_state() -> None:
"timestamp": AnyStr(),
"step": 3,
"payload": {
"id": "8959fb57-d0f5-5725-9ac4-ec1c554fb0a0",
"id": AnyStr(),
"name": "qa",
"input": {
"query": "query: what is weather in sf",
@@ -5482,7 +5482,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "7b7b0713-e958-5d07-803c-c9910a7cc162",
"id": AnyStr(),
"name": "prepare",
"input": {"my_key": "value", "market": "DE"},
"triggers": ["start:prepare"],
@@ -5537,7 +5537,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
"timestamp": AnyStr(),
"step": 2,
"payload": {
"id": "dd9f2fa5-ccfa-5d12-81ec-942563056a08",
"id": AnyStr(),
"name": "tool_two_slow",
"input": {"my_key": "value prepared", "market": "DE"},
"triggers": ["branch:prepare:condition:tool_two_slow"],
@@ -5590,7 +5590,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
"timestamp": AnyStr(),
"step": 3,
"payload": {
"id": "9b590c54-15ef-54b1-83a7-140d27b0bc52",
"id": AnyStr(),
"name": "finish",
"input": {"my_key": "value prepared slow", "market": "DE"},
"triggers": ["branch:prepare:condition::then"],
@@ -5719,7 +5719,7 @@ async def test_branch_then(checkpointer_name: str) -> None:
"timestamp": AnyStr(),
"step": 1,
"payload": {
"id": "1a591be4-f85c-558f-8d00-1ccac0d1877f",
"id": AnyStr(),
"name": "prepare",
"input": {"my_key": "value", "market": "DE"},
"triggers": ["start:prepare"],