File size: 989 Bytes
9f2df60
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import os
from langsmith import Client
from dotenv import load_dotenv


def main() -> int:
    load_dotenv()

    print("Tracing:", os.getenv("LANGSMITH_TRACING"))
    print("Project:", os.getenv("LANGSMITH_PROJECT"))
    print("OpenAI-key present:", os.getenv("OPENAI_API_KEY") is not None)

    try:
        client = Client()
        _ = list(client.list_projects())
        print("LangSmith connection: OK")
    except Exception as e:
        print("LangSmith connection: ERROR")
        print(e)
        return 1

    try:
        client.create_run(
            name="langsmith_tracing_smoketest",
            run_type="chain",
            inputs={"test": True},
            outputs={"ok": True},
            project_name=os.getenv("LANGSMITH_PROJECT"),
        )
        print("Tracing: OK (run created)")
    except Exception as e:
        print("Tracing: ERROR (run failed)")
        print(e)
        return 1

    return 0


if __name__ == "__main__":
    raise SystemExit(main())