dify中如何将LLM节点输出的内容(自然语言+MD表格+自然语言)中的MD表格提取出来并渲染成表格

import re

def main(text: str) -> dict:
    pattern = r'\|.*\|\n\|.*\|\n(\|.*\|\n)*'
    matches = re.findall(pattern, text)
    if matches:
        table = matches[0].strip()
        rows = [row.strip() for row in table.split('\n')]
        headers = [h.strip() for h in rows[0].split('|')[1:-1]]
        data = []
        for row in rows[2:]:
            cells = [c.strip() for c in row.split('|')[1:-1]]
            data.append(dict(zip(headers, cells)))
        return {'result': data}
    return {'result': []}

评论

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注