Aurele000 commited on
Commit
e2d8259
·
1 Parent(s): a2c9a5a

modif format output

Browse files
Files changed (1) hide show
  1. agent.py +3 -3
agent.py CHANGED
@@ -71,7 +71,7 @@ Answer: a, b
71
  """
72
 
73
  @tool
74
- def wiki_search(query: str) -> str:
75
  """Search Wikipedia for a query and return maximum 2 results.
76
 
77
  Args:
@@ -86,7 +86,7 @@ def wiki_search(query: str) -> str:
86
 
87
 
88
  @tool
89
- def arvix_search(query: str) -> str:
90
  """Search Arxiv for a query and return maximum 3 result.
91
 
92
  Args:
@@ -110,7 +110,7 @@ prompt_web="""
110
  "- You can browse the web then give your results to your supervisor "
111
  """
112
  @tool
113
- def web_search_openai_tool(query: str) -> str:
114
  """Call web search and the output is a structured text answering the query)
115
 
116
  Args:
 
71
  """
72
 
73
  @tool
74
+ def wiki_search(query: str) -> dict[str, str]:
75
  """Search Wikipedia for a query and return maximum 2 results.
76
 
77
  Args:
 
86
 
87
 
88
  @tool
89
+ def arvix_search(query: str) -> dict[str, str]:
90
  """Search Arxiv for a query and return maximum 3 result.
91
 
92
  Args:
 
110
  "- You can browse the web then give your results to your supervisor "
111
  """
112
  @tool
113
+ def web_search_openai_tool(query: str) -> dict[str, str]:
114
  """Call web search and the output is a structured text answering the query)
115
 
116
  Args: