#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Langfuse trace plugin 关键字 grep（排查工具调用是否生成 span/observation）。"""

from __future__ import annotations

import sys

import _bootstrap  # noqa: F401
import _config as C
from dify_api import Dify


def main() -> int:
    d = Dify(C.DIFY_EMAIL, C.DIFY_PASSWORD)
    try:
        d.login()
        path = (
            "/app/api/.venv/lib/python3.12/site-packages/dify_trace_langfuse/"
            "langfuse_trace.py"
        )
        pats = [
            "CreateSpanBody",
            "add_span",
            "dataset_retrieval",
            "tool",
            "observation",
        ]
        for pat in pats:
            grep = "grep -n %r %s | head -60" % (pat, path)
            cmd = "docker exec docker-api-1 sh -lc %r" % grep
            print("\n=== grep pat ===", pat)
            print("CMD:", cmd)
            out, err = d.sh(cmd)
            print(out or err or "(empty)")
        return 0
    finally:
        d.close()


if __name__ == "__main__":
    sys.exit(main())