phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-14 22:12:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix unit test and add more recordings

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-04 09:01:40 -07:00
parent 7008a70a6e
commit 314ca13835
10 changed files with 5407 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
tests/unit/distribution/test_inference_recordings.py
View file

@ -131,10 +131,6 @@ class TestInferenceRecording:
temp_storage_dir = temp_storage_dir / "test_response_storage"
storage = ResponseStorage(temp_storage_dir)
# Test directory creation
assert storage.test_dir.exists()
assert storage.responses_dir.exists()
# Test storing and retrieving a recording
request_hash = "test_hash_123"
request_data = {
@ -174,7 +170,8 @@ class TestInferenceRecording:
# Verify recording was stored
storage = ResponseStorage(temp_storage_dir)
assert storage.responses_dir.exists()
dir = storage._get_test_dir()
assert dir.exists()
async def test_replay_mode(self, temp_storage_dir, real_openai_chat_response):
"""Test that replay mode returns stored responses without making real calls."""