feat: update test

星橙 · 星橙 · commit 36b4d3b08603 · 2025-03-13T20:14:31.000+08:00
diff --git a/examples/weather/main.py b/examples/weather/main.py
@@ -1,8 +1,20 @@
 import logging
 import asyncio
+import argparse
+import os
+
 from weather_manager import WeatherSearch
 
 
+def define_options():
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        '--api_key', dest='api_key', required=True,
+        help='OpenAI API KEY'
+    )
+    options = parser.parse_args()
+    return options
+
 def setup_logger():
     logger = logging.getLogger()
     handler = logging.StreamHandler()
@@ -13,6 +25,8 @@ def setup_logger():
     return logger
 
 async def main():
+    options = define_options()
+    os.environ['OPENAI_API_KEY'] = options.api_key
     query = input("请输入：")
     logger = setup_logger()
     weather_search = WeatherSearch(logger=logger)
