From ad890179a6adb4e314c9ace675d5fe5d335af632 Mon Sep 17 00:00:00 2001
From: aaron <>
Date: Wed, 11 Jun 2025 15:09:27 +0800
Subject: [PATCH] update

---
 cryptoai/tasks/token_selector.py | 7 +++++--
 test.py                          | 4 ++--
 2 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/cryptoai/tasks/token_selector.py b/cryptoai/tasks/token_selector.py
index 344c6b0..f035364 100644
--- a/cryptoai/tasks/token_selector.py
+++ b/cryptoai/tasks/token_selector.py
@@ -88,7 +88,7 @@ def analyze_symbol(symbol, interval='1h', limit=1000):
             '得分': 0
         }
     
-if __name__ == "__main__":
+def main():
     # 执行分析
     results = [analyze_symbol(sym) for sym in top_symbols]
     df_result = pd.DataFrame(results)
@@ -169,4 +169,7 @@ if __name__ == "__main__":
         border_style="blue"
     )
     console.print("\n")
-    console.print(info_panel)
\ No newline at end of file
+    console.print(info_panel)
+    
+if __name__ == "__main__":
+    main()
\ No newline at end of file
diff --git a/test.py b/test.py
index 1379706..72cca2c 100644
--- a/test.py
+++ b/test.py
@@ -1,5 +1,5 @@
 from cryptoai.agents.crypto_agent import CryptoAgent
+import cryptoai.tasks.token_selector as token_selector
 
 if __name__ == "__main__":
-    agent = CryptoAgent()
-    agent.start_agent()
\ No newline at end of file
+    token_selector.main()
\ No newline at end of file