wrigleyDan commited on
Commit
cc122da
·
verified ·
1 Parent(s): 7a99b64

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -28,7 +28,7 @@ api_key = os.getenv("OPENAI_API_KEY")
28
  model = ChatOpenAI(model="gpt-4o-mini", api_key=api_key, temperature=0)
29
 
30
  @tool
31
- def search_wiki(query: str, max_results: int = 3) -> str:
32
  """
33
  Searches Wikipedia for the given query and returns a maximum of 'max_results'
34
  relevant article summaries, titles, and URLs.
 
28
  model = ChatOpenAI(model="gpt-4o-mini", api_key=api_key, temperature=0)
29
 
30
  @tool
31
+ def search_wiki(query: str, max_results: int = 2) -> str:
32
  """
33
  Searches Wikipedia for the given query and returns a maximum of 'max_results'
34
  relevant article summaries, titles, and URLs.