Add Delete Session Method (#5193)

tests/integration_tests/client/test_client.py

@@ -0,0 +1,52 @@
+"""LangChain+ langchain_client Integration Tests."""
+from uuid import uuid4
+
+import pytest
+from tenacity import RetryError
+
+from langchain.callbacks.manager import tracing_v2_enabled
+from langchain.client import LangChainPlusClient
+from langchain.tools.base import tool
+
+
+@pytest.fixture
+def langchain_client(monkeypatch: pytest.MonkeyPatch) -> LangChainPlusClient:
+    monkeypatch.setenv("LANGCHAIN_ENDPOINT", "http://localhost:1984")
+    return LangChainPlusClient()
+
+
+def test_sessions(
+    langchain_client: LangChainPlusClient, monkeypatch: pytest.MonkeyPatch
+) -> None:
+    """Test sessions."""
+    session_names = set([session.name for session in langchain_client.list_sessions()])
+    new_session = f"Session {uuid4()}"
+    assert new_session not in session_names
+
+    @tool
+    def example_tool() -> str:
+        """Call me, maybe."""
+        return "test_tool"
+
+    monkeypatch.setenv("LANGCHAIN_ENDPOINT", "http://localhost:1984")
+    with tracing_v2_enabled(session_name=new_session):
+        example_tool({})
+    session = langchain_client.read_session(session_name=new_session)
+    assert session.name == new_session
+    session_names = set([sess.name for sess in langchain_client.list_sessions()])
+    assert new_session in session_names
+    runs = list(langchain_client.list_runs(session_name=new_session))
+    session_id_runs = list(langchain_client.list_runs(session_id=session.id))
+    assert len(runs) == len(session_id_runs) == 1
+    assert runs[0].id == session_id_runs[0].id
+    langchain_client.delete_session(session_name=new_session)
+
+    with pytest.raises(RetryError):
+        langchain_client.read_session(session_name=new_session)
+    assert new_session not in set(
+        [sess.name for sess in langchain_client.list_sessions()]
+    )
+    with pytest.raises(RetryError):
+        langchain_client.delete_session(session_name=new_session)
+    with pytest.raises(RetryError):
+        langchain_client.read_run(run_id=str(runs[0].id))