|
@@ -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]:
|
|
|
"""从飞书记录中提取内容"""
|