66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
|
# MIT License
|
|||
|
|
# Copyright (c) 2024
|
|||
|
|
"""LLM 抽象与 Dummy 实现。"""
|
|||
|
|
|
|||
|
|
from abc import ABC, abstractmethod
|
|||
|
|
from typing import Any, Dict, List
|
|||
|
|
|
|||
|
|
import yaml
|
|||
|
|
|
|||
|
|
from .schema import DSLSpec, EventRecord
|
|||
|
|
|
|||
|
|
PROMPT_TEMPLATE = """你是一名自动化工程师，请将以下事件序列归纳为可参数化的自动化 DSL。
|
|||
|
|
事件序列使用 JSON 描述，每个事件包含 kind、control（AutomationId/Name/ClassName/ControlType/BoundingRect）等。
|
|||
|
|
输出 YAML，字段包括：params、steps、assertions、retry_policy、waits，支持 steps 内的 if/else、for_each。
|
|||
|
|
输出示例：
|
|||
|
|
params:
|
|||
|
|
text: "示例参数"
|
|||
|
|
steps:
|
|||
|
|
- action: click
|
|||
|
|
target: {{AutomationId: "15", ControlType: "Edit"}}
|
|||
|
|
- action: type
|
|||
|
|
target: {{AutomationId: "15"}}
|
|||
|
|
text: "{{text}}"
|
|||
|
|
assertions:
|
|||
|
|
- "输入框非空"
|
|||
|
|
retry_policy: {{max_attempts: 2, interval: 1.0}}
|
|||
|
|
waits: {{appear: 5.0, disappear: 5.0}}
|
|||
|
|
现在请基于输入事件生成 YAML："""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class LLMClient(ABC):
|
|||
|
|
"""LLM 抽象接口。"""
|
|||
|
|
|
|||
|
|
@abstractmethod
|
|||
|
|
def generate(self, events: List[EventRecord]) -> DSLSpec:
|
|||
|
|
"""将事件序列转为 DSL 规格。"""
|
|||
|
|
|
|||
|
|
|
|||
|
|
class DummyLLM(LLMClient):
|
|||
|
|
"""离线 dummy，实现一个简单的规则映射。"""
|
|||
|
|
|
|||
|
|
def generate(self, events: List[EventRecord]) -> DSLSpec:
|
|||
|
|
steps: List[Dict[str, Any]] = []
|
|||
|
|
for ev in events:
|
|||
|
|
ctrl = ev.control.dict(by_alias=True) if ev.control else {}
|
|||
|
|
if ev.kind == "mouse_click":
|
|||
|
|
steps.append({"action": "click", "target": ctrl})
|
|||
|
|
elif ev.kind == "key_down" and ev.data.get("name"):
|
|||
|
|
# 仅在按键时记录输入
|
|||
|
|
steps.append({"action": "type", "target": ctrl, "text": ev.data.get("name")})
|
|||
|
|
if not steps:
|
|||
|
|
steps.append({"action": "assert_exists", "target": {"Name": "dummy"}})
|
|||
|
|
spec = DSLSpec(
|
|||
|
|
params={},
|
|||
|
|
steps=steps,
|
|||
|
|
assertions=["dummy generated"],
|
|||
|
|
)
|
|||
|
|
return spec
|
|||
|
|
|
|||
|
|
|
|||
|
|
def render_prompt(events: List[EventRecord]) -> str:
|
|||
|
|
"""把事件序列渲染到 prompt。"""
|
|||
|
|
event_dicts = [ev.dict(by_alias=True) for ev in events]
|
|||
|
|
return f"{PROMPT_TEMPLATE}\n\n{yaml.safe_dump(event_dicts, allow_unicode=True)}"
|
|||
|
|
|