mirror of
https://github.com/langchain-ai/langgraph.git
synced 2026-09-30 05:25:05 +02:00
Apply patch [skip ci]
This commit is contained in:
@@ -0,0 +1,45 @@
|
||||
#!/usr/bin/env python
|
||||
"""Debug why InjectedState deprecation warning is not being emitted."""
|
||||
|
||||
from typing import Annotated
|
||||
from langgraph.prebuilt import ToolNode, InjectedState, InjectedStore
|
||||
from langgraph.prebuilt.tool_node import _get_state_args, _get_store_arg, _get_reserved_keyword_args
|
||||
from langgraph.store.base import BaseStore
|
||||
from langchain_core.tools import create_tool
|
||||
|
||||
|
||||
def tool_with_injected_state(x: int, state: Annotated[dict, InjectedState]) -> str:
|
||||
"""Tool using deprecated InjectedState annotation."""
|
||||
return f"state: {state.get('foo', 'none')}"
|
||||
|
||||
def tool_with_injected_store(x: int, store: Annotated[BaseStore, InjectedStore()]) -> str:
|
||||
"""Tool using deprecated InjectedStore annotation."""
|
||||
return "has store"
|
||||
|
||||
# Convert to tools
|
||||
tool1 = create_tool(tool_with_injected_state)
|
||||
tool2 = create_tool(tool_with_injected_store)
|
||||
|
||||
print("Debugging annotation detection:")
|
||||
print(f"\nTool 1 (InjectedState):")
|
||||
print(f" _get_state_args: {_get_state_args(tool1)}")
|
||||
print(f" _get_reserved_keyword_args: {_get_reserved_keyword_args(tool1)}")
|
||||
|
||||
print(f"\nTool 2 (InjectedStore):")
|
||||
print(f" _get_store_arg: {_get_store_arg(tool2)}")
|
||||
print(f" _get_reserved_keyword_args: {_get_reserved_keyword_args(tool2)}")
|
||||
|
||||
# Check the condition for warnings
|
||||
state_args = _get_state_args(tool1)
|
||||
reserved_args = _get_reserved_keyword_args(tool1)
|
||||
print(f"\nTool 1 warning condition:")
|
||||
print(f" state_args: {state_args}")
|
||||
print(f" reserved_args.get('state'): {reserved_args.get('state')}")
|
||||
print(f" Should warn: {bool(state_args and not reserved_args.get('state'))}")
|
||||
|
||||
store_arg = _get_store_arg(tool2)
|
||||
reserved_args2 = _get_reserved_keyword_args(tool2)
|
||||
print(f"\nTool 2 warning condition:")
|
||||
print(f" store_arg: {store_arg}")
|
||||
print(f" reserved_args.get('runtime'): {reserved_args2.get('runtime')}")
|
||||
print(f" Should warn: {bool(store_arg and not reserved_args2.get('runtime'))}")
|
||||
Reference in New Issue
Block a user