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.
|