update return type of WolframAlphaAPIWrapper.run() (microsoft#523)

* update return type of WolframAlphaAPIWrapper.run * replace tuple by typing.Tuple --------- Co-authored-by: Chi Wang <[email protected]> Co-authored-by: Yiran Wu <[email protected]>
whiskyboy · Nov 9, 2023 · 495cf7c · 495cf7c
1 parent 8bf3a5e
commit 495cf7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autogen/agentchat/contrib/math_user_proxy_agent.py b/autogen/agentchat/contrib/math_user_proxy_agent.py
@@ -1,7 +1,7 @@
 import re
 import os
 from pydantic import BaseModel, Extra, root_validator
-from typing import Any, Callable, Dict, List, Optional, Union
+from typing import Any, Callable, Dict, List, Optional, Union, Tuple
 from time import sleep
 
 from autogen.agentchat import Agent, UserProxyAgent
@@ -402,7 +402,7 @@ def validate_environment(cls, values: Dict) -> Dict:
 
         return values
 
-    def run(self, query: str) -> str:
+    def run(self, query: str) -> Tuple[str, bool]:
         """Run query through WolframAlpha and parse result."""
         from urllib.error import HTTPError