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

chore(testing): make LLAMA_STACK_TEST_RECORDING_DIR optional

Browse source 
# What does this PR do?


## Test Plan
This commit is contained in:
Eric Huang 2025-08-01 12:07:47 -07:00
parent 140ee7d337
commit 51f94df2e7
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/testing/inference_recorder.py
View file

@ -79,9 +79,11 @@ def setup_inference_recording():
if mode == InferenceMode.LIVE: if mode == InferenceMode.LIVE:
return None return None
if "LLAMA_STACK_TEST_RECORDING_DIR" not in os.environ: if "LLAMA_STACK_TEST_RECORDING_DIR" in os.environ:
raise ValueError("LLAMA_STACK_TEST_RECORDING_DIR must be set for recording or replaying") storage_dir = os.environ["LLAMA_STACK_TEST_RECORDING_DIR"]
storage_dir = os.environ["LLAMA_STACK_TEST_RECORDING_DIR"] else:
logger.warning("LLAMA_STACK_TEST_RECORDING_DIR not set, using default directory: tests/integration/recordings")
storage_dir = "tests/integration/recordings"
return inference_recording(mode=mode, storage_dir=storage_dir) return inference_recording(mode=mode, storage_dir=storage_dir)