Skip to content

Commit

Permalink
fix test
Browse files Browse the repository at this point in the history
  • Loading branch information
isahers1 committed Dec 17, 2024
1 parent 34c0dd8 commit e802e31
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions python/tests/integration_tests/test_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -941,7 +941,9 @@ def test_multipart_ingest_create_with_attachments(
langchain_client: Client, caplog: pytest.LogCaptureFixture
) -> None:
_session = "__test_multipart_ingest_create_with_attachments"

langchain_client = Client(
api_key="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"
)
trace_a_id = uuid4()
current_time = datetime.datetime.now(datetime.timezone.utc).strftime(
"%Y%m%dT%H%M%S%fZ"
Expand Down Expand Up @@ -971,7 +973,7 @@ def test_multipart_ingest_create_with_attachments(
langchain_client.multipart_ingest(create=runs_to_create, update=[])

assert not caplog.records
time.sleep(3)
time.sleep(5) # Need this so the run persists
created_run = langchain_client.read_run(run_id=str(trace_a_id))
assert created_run.attachments
assert sorted(created_run.attachments.keys()) == sorted(["foo", "bar"])
Expand Down

0 comments on commit e802e31

Please sign in to comment.