Skip to content

enable sensitive log mode#415

Merged
song-william merged 8 commits intomainfrom
wsong/sens-log-mode
Jan 4, 2024
Merged

enable sensitive log mode#415
song-william merged 8 commits intomainfrom
wsong/sens-log-mode

Conversation

@song-william
Copy link
Copy Markdown
Contributor

@song-william song-william commented Dec 31, 2023

Pull Request Summary

Enable sensitive log mode for use cases that want avoid logging user-specific data (e.g prompts and responses)

Test Plan and Usage Guide

Deploy in training and confirmed in datadog that logs no longer show up in vllm or logged by our gateway.



# When enabling this test, other tests fail with "RunTumeError got Future <Future pending> attached to a different loop"
# https://github.com/encode/starlette/issues/1315#issuecomment-980784457
Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

To reproduce, run

GIT_TAG=$(git rev-parse HEAD) WORKSPACE=..  pytest -v -s tests/unit/api/test_tasks.py::test_create_streaming_task_success tests/unit/api/test_llms.py::test_completion_stream_success

When running these tests individually, we don't have an issue. Only when running both does this error appear. I put id(asyncio.get_running_loop()) throughout the create_completion_stream_task method and everything lined up and didn't error, so there must be something up with the TestClient library itself. Time-boxing this for now.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I tried the small fixes in Kludex/starlette#1315 (comment), none of them seemed to work. I think we will either need to downgrade fastapi as folks suggested.

@song-william song-william enabled auto-merge (squash) January 3, 2024 23:47
@song-william song-william merged commit 756682f into main Jan 4, 2024
@song-william song-william deleted the wsong/sens-log-mode branch January 4, 2024 00:06
@yunfeng-scale yunfeng-scale mentioned this pull request Mar 6, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants