APRG commited on
Commit
ee6d9a4
·
verified ·
1 Parent(s): 7ae7535

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -71,7 +71,7 @@ def agent_node(state: OrderState) -> OrderState:
71
 
72
  def interactive_tools_node(state: OrderState) -> OrderState:
73
  """Handles interactive tool calls."""
74
- logging.info("order node")
75
  tool_msg = state.get("messages", [])[-1]
76
  order = state.get("order", [])
77
  outbound_msgs = []
@@ -81,7 +81,7 @@ def interactive_tools_node(state: OrderState) -> OrderState:
81
  tool_args = tool_call["args"]
82
 
83
  if tool_name == "wikipedia_search":
84
- page = wikipedia.page(title)
85
  response = page.content[:300]
86
 
87
  else:
@@ -113,7 +113,7 @@ def maybe_route_to_tools(state: OrderState) -> str:
113
  print("from agent GOTO tools node")
114
  return "tools"
115
  else:
116
- logging.info("from chatbot GOTO order node")
117
  return "interactive_tools"
118
 
119
  print("tool call failed, quitting")
 
71
 
72
  def interactive_tools_node(state: OrderState) -> OrderState:
73
  """Handles interactive tool calls."""
74
+ logging.info("interactive tools node")
75
  tool_msg = state.get("messages", [])[-1]
76
  order = state.get("order", [])
77
  outbound_msgs = []
 
81
  tool_args = tool_call["args"]
82
 
83
  if tool_name == "wikipedia_search":
84
+ page = wikipedia.page(tool_args.get("title"))
85
  response = page.content[:300]
86
 
87
  else:
 
113
  print("from agent GOTO tools node")
114
  return "tools"
115
  else:
116
+ logging.info("from chatbot GOTO interactive tools node")
117
  return "interactive_tools"
118
 
119
  print("tool call failed, quitting")