Update agent.py
Browse files
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 =
|
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.
|