first commit

backend/app/prompts/extract_transaction.py

@@ -0,0 +1,41 @@
+"""Prompt for extracting transactions from billing screenshots."""
+
+EXTRACT_TRANSACTION_SYSTEM = """你是一个专业的金融交易数据提取助手，专门用于从手机APP账单或交易记录截图中提取结构化信息。"""
+
+EXTRACT_TRANSACTION_USER = """请分析这张手机APP账单/交易记录截图，提取所有可见的交易记录。
+
+要求：
+1. 只返回一个JSON数组，不要包含其他说明文字。
+2. 数组的每个元素是一条交易，包含以下字段（若截图中无该信息则填 null）：
+   - app_source: string，APP来源，如 "微信支付"、"支付宝"、"XX银行"、"XX钱包"
+   - transaction_type: string，交易类型，如 "转出"、"转入"、"消费"、"收款"、"提现"、"充值"
+   - amount: number，金额（数字，不含货币符号）
+   - currency: string，币种，如 "CNY"、"USDT"，默认 "CNY"
+   - counterparty_name: string | null，对方名称/姓名
+   - counterparty_account: string | null，对方账号、卡号尾号、钱包地址等
+   - order_number: string | null，订单号/交易号
+   - transaction_time: string | null，交易时间，请用 ISO 8601 格式，如 "2024-01-15T14:30:00"
+   - remark: string | null，备注/摘要
+   - confidence: string，识别置信度，取 "high"、"medium"、"low" 之一
+
+3. 注意区分转入和转出方向；金额统一为正数，方向由 transaction_type 体现。
+4. 若截图中没有交易记录或无法识别，返回空数组 []。
+
+直接输出JSON数组，不要用 markdown 代码块包裹。"""
+
+
+def get_extract_messages(image_b64: str) -> list[dict]:
+    """Build messages for vision API: system + user with image."""
+    return [
+        {"role": "system", "content": EXTRACT_TRANSACTION_SYSTEM},
+        {
+            "role": "user",
+            "content": [
+                {"type": "text", "text": EXTRACT_TRANSACTION_USER},
+                {
+                    "type": "image_url",
+                    "image_url": {"url": f"data:image/jpeg;base64,{image_b64}"},
+                },
+            ],
+        },
+    ]