丁云鹏 1 viikko sitten
vanhempi
commit
e555d727b8
1 muutettua tiedostoa jossa 3 lisäystä ja 11 poistoa
  1. 3 11
      3_handle.py

+ 3 - 11
3_handle.py

@@ -8,7 +8,7 @@ from typing import Dict, Any, List, Optional
 # 导入自定义模块
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-from utils.fei_shu import FeiShu
+from utils.mysql_db import MysqlHelper
 from gemini import GeminiProcessor
 from utils.file import File
 
@@ -17,19 +17,11 @@ class Handler:
     def __init__(self, table_id: Optional[str] = None):
         
         # 初始化飞书客户端
-        self.feishu = FeiShu(file_token='VEBsbCfaWa3gF3slQILc6Rybnde')
+        self.mysql = MysqlHelper()
         self.processor = GeminiProcessor()
-        
-        
-        # 获取表格ID:优先使用传入的参数,其次使用环境变量
-        self.table_id = table_id
-        # 字段名称配置
-        self.input_field = '识别结果'
-        self.output_field = '初步理解'
-
         self.system_prompt = File.read_file('prompt/handle.md')
 
-        print(self.system_prompt)
+        # print(self.system_prompt)
 
     def extract_content_from_record(self, record) -> Dict[str, Any]:
         """从飞书记录中提取内容"""