knowledge/prehandler.py

import config
import logging
import os
from feishu_client import FeishuClient
from coze_client import CozeClient

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

if __name__ == "__main__":
    # --- 环境变量/配置信息加载 ---
    # 推荐使用环境变量加载敏感信息，而不是硬编码。
    # 例如：export FEISHU_APP_ID="your_id"
    # 或者从配置文件 (如 config.ini, .env 文件) 中加载

    # 飞书配置
    FEISHU_APP_ID = config.FEISHU_APP_ID
    FEISHU_APP_SECRET = config.FEISHU_APP_SECRET
    FEISHU_BASE_ID = config.FEISHU_BASE_ID
    FEISHU_TABLE_ID = config.FEISHU_TABLE_ID

    FEISHU_INPUT_FIELD = config.FEISHU_INPUT_FIELD # 你的飞书表格中用于输入的列名
    FEISHU_OUTPUT_FIELD = config.FEISHU_OUTPUT_FIELD # 你的飞书表格中用于输出的列名

    # Coze 配置
    COZE_API_KEY = config.COZE_API_KEY
    COZE_BOT_ID = config.COZE_BOT_ID # 例如: "7343685511394590740"

    # Coze 提示模板，请确保包含 {input_data} 占位符
    # 这是一个示例，你可以根据你的机器人功能设计更复杂的提示
    COZE_PROMPT_TEMPLATE = os.getenv("COZE_PROMPT_TEMPLATE", "请作为一位专业的编辑，总结以下文章的核心内容，要求言简意赅，200字以内: {input_data}")

    # --- 执行流程 ---
    if "YOUR_" in FEISHU_APP_ID or "YOUR_" in COZE_API_KEY:
        logging.error("⛔️ 请检查 main.py 或环境变量，确保所有 'YOUR_' 占位符都已替换为您的实际配置信息！⛔️")
        logging.error("流程未执行。")
    else:
        process_feishu_data_with_coze_flow(
            feishu_app_id=FEISHU_APP_ID,
            feishu_app_secret=FEISHU_APP_SECRET,
            feishu_base_id=FEISHU_BASE_ID,
            feishu_table_id=FEISHU_TABLE_ID,
            feishu_input_field_name=FEISHU_INPUT_FIELD,
            feishu_output_field_name=FEISHU_OUTPUT_FIELD,
            coze_api_key=COZE_API_KEY,
            coze_bot_id=COZE_BOT_ID,
            coze_prompt_template=COZE_PROMPT_TEMPLATE,
            max_records_to_process=10, # 每次运行最多处理10条记录
            overwrite_existing_output=True # 总是覆盖输出字段
        )