update websearch run

utopic-dev · May 3, 2024 · de8a778 · de8a778
1 parent 9e8aa34
commit de8a778
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/phi/assistant/assistant.py b/phi/assistant/assistant.py
@@ -874,9 +874,9 @@ def _run(
  llm_response_type = "markdown"
  functions = {}
  if self.llm:
- for _func in self.llm.functions:
+ for _f_name, _func in self.llm.functions.items():
  if isinstance(_func, Function):
- functions[_func.name] = _func.to_dict()
+ functions[_f_name] = _func.to_dict()
  event_data = {
  "run_type": "assistant",
  "user_message": message,
@@ -1070,9 +1070,9 @@ async def _arun(
  llm_response_type = "markdown"
  functions = {}
  if self.llm:
- for _func in self.llm.functions:
+ for _f_name, _func in self.llm.functions.items():
  if isinstance(_func, Function):
- functions[_func.name] = _func.to_dict()
+ functions[_f_name] = _func.to_dict()
  event_data = {
  "run_type": "assistant",
  "user_message": message,