hsg_rag_eea / tools /test_langsmith_setup.py
Pygmales's picture
Sync from GitHub dbb86cf4b41e84b3c8d61170a6dca9742c686757
9f2df60 verified
Raw
History Blame
989 Bytes
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())