#!/usr/bin/env python3
# -*- coding: utf-8 -*-
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"
        out, err = d.sh(
            "docker exec docker-api-1 sh -c 'grep -n -A2 \"def moderation_trace\" %s | head -20'" % path,
            t=120,
        )
        print("=== moderation_trace context ===")
        print(out or err)
        # Show lines around add_generation
        out2, err2 = d.sh(
            "docker exec docker-api-1 sh -c 'grep -n \"add_generation\\|moderation_trace\" %s | tail -20'" % path,
            t=120,
        )
        print("=== add_generation + moderation_trace lines ===")
        print(out2 or err2)
        return 0
    finally:
        d.close()

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