From 98007f59a884b1d6de35f251fefa81e9bce103fc Mon Sep 17 00:00:00 2001
From: William Fu-Hinthorn <13333726+hinthornw@users.noreply.github.com>
Date: Fri, 9 Feb 2024 00:06:16 -0800
Subject: [PATCH] fixup

---
 js/src/tests/client.int.test.ts               | 2 +-
 python/tests/integration_tests/test_client.py | 8 +-------
 2 files changed, 2 insertions(+), 8 deletions(-)

diff --git a/js/src/tests/client.int.test.ts b/js/src/tests/client.int.test.ts
index 704153ce5..909f31bf2 100644
--- a/js/src/tests/client.int.test.ts
+++ b/js/src/tests/client.int.test.ts
@@ -69,7 +69,7 @@ async function waitUntilRunFound(
         return false;
       }
     },
-    60_000,
+    180_000,
     1_000
   );
 }
diff --git a/python/tests/integration_tests/test_client.py b/python/tests/integration_tests/test_client.py
index 4c624d379..5f57caaec 100644
--- a/python/tests/integration_tests/test_client.py
+++ b/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)