|
|
| from typing import TypedDict, Annotated, Any |
| from langchain_core.messages import AnyMessage |
| from langgraph.graph.message import add_messages |
|
|
| tools_config = [ |
| { |
| "name": "book_table", |
| "description": "Book a table at a restaurant", |
| "parameters": [ |
| {"name": "restaurant", "type": "string", "required": True}, |
| {"name": "people", "type": "integer", "required": True}, |
| {"name": "date", "type": "string", "required": True}, |
| {"name": "time", "type": "string", "required": True} |
| ] |
| }, |
| { |
| "name": "check_menu", |
| "description": "Check if a dish is on the restaurant menu", |
| "parameters": [ |
| {"name": "restaurant", "type": "string", "required": True}, |
| {"name": "dish", "type": "string", "required": True} |
| ] |
| } |
| ] |
|
|
| def build_parameter_catalog(tools_config): |
| catalog = {} |
|
|
| for tool in tools_config: |
| for param in tool["parameters"]: |
| name = param["name"] |
|
|
| if name not in catalog: |
| catalog[name] = { |
| "type": param["type"] |
| } |
|
|
| return catalog |
|
|
| parameter_catalog = build_parameter_catalog(tools_config) |
|
|
| def merge_dicts(old: dict, new: dict) -> dict: |
| return {**old, **new} |
|
|
| class AgentState(TypedDict): |
| messages: Annotated[list[AnyMessage], add_messages] |
| global_params: Annotated[dict[str, Any], merge_dicts] |
| selected_tool: str | None |
| missing_params: list[str] |
| tool_args: dict[str, Any] |
| tool_result: dict[str, Any] | None |
