fixup

js/src/tests/client.int.test.ts

@@ -69,7 +69,7 @@ async function waitUntilRunFound(
         return false;
       }
     },
-    60_000,
+    180_000,
     1_000
   );
 }

python/tests/integration_tests/test_client.py

@@ -1,6 +1,5 @@
 """LangSmith langchain_client Integration Tests."""
 
-import functools
 import io
 import os
 import random
@@ -16,10 +15,7 @@
 
 from langsmith.client import Client
 from langsmith.schemas import DataType
-from langsmith.utils import (
-    LangSmithConnectionError,
-    LangSmithError,
-)
+from langsmith.utils import LangSmithConnectionError, LangSmithError
 
 
 def wait_for(
@@ -300,11 +296,9 @@ def test_create_run_with_masked_inputs_outputs(
         )
         wait_for(lambda: langchain_client.read_run(run_id).end_time is not None)
         stored_run = langchain_client.read_run(run_id)
-        assert "prompt" in stored_run.inputs
         assert "hello" not in str(stored_run.inputs)
         wait_for(lambda: langchain_client.read_run(run_id2).end_time is not None)
         stored_run2 = langchain_client.read_run(run_id2)
-        assert "messages" in stored_run2.inputs
         assert "hello" not in str(stored_run2.inputs)
     finally:
         langchain_client.delete_project(project_name=project_name)