添加env

.gitignore

@@ -1,3 +1,4 @@
 __pycache__/
 .codeartsdoer/
 .idea/
+.env

dify_api.py

@@ -1,14 +1,17 @@
 from logger_config import logger
 import requests
+import os
+import dotenv
+dotenv.load_dotenv()
 
 class ChatHistoryCollector:
     """聊天历史数据收集器"""
     def __init__(self, limit=100, timeout=5):
 
         # 聊天历史API的URL和请求头
-        self.base_url = "https://newaterobot.gtwater.cn/v1/conversations"
+        self.base_url = os.getenv('DIFY_URL', '')
         self.headers = {
-            'Authorization': 'Bearer app-qpPo2VXUAnPxKOIQm4NrytaL',
+            'Authorization': F'Bearer {os.getenv('DIFY_API_KEY', '')}',
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/146.0.0.0 Safari/537.36'
         }
         self.logger = logger

main.py

@@ -21,6 +21,7 @@ def run(user='admin'):
         counter = 0
         with db:
             # 先获取一次数据
+            time.sleep(1)
             has_more, conversations = collector.collect(user=user)
             for conversation in conversations:
                 if db.insert_record(id_=conversation['id'], record=conversation['name'], time=conversation['created_at']):

user_api.py

@@ -1,11 +1,13 @@
 import requests
 from logger_config import logger
-
+import os
+import dotenv
+dotenv.load_dotenv()
 class UserCollector:
 
     def __init__(self):
-        self.base_url = "http://120.55.44.4:8900/api/v2/user?"
-        token = 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJJRCI6NywiVXNlcm5hbWUiOiJhZG1pbiIsIkRlcCI6IjEzNSIsImV4cCI6MTc3NjExOTExNCwiaXNzIjoiZ2luLWJsb2cifQ.0HTtzHZjyd2mHo8VCy8icYROxmntRMuQhyoZsAYRL_M'
+        self.base_url = os.getenv('USER_URL', '')
+        token = os.getenv('JWT_TOKEN', '')
         self.headers = {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/146.0.0.0 Safari/537.36',
             'JWT-TOKEN': f'{token}'