From ffc882c2d12af9d6b573115cf26f549ff03beabb Mon Sep 17 00:00:00 2001 From: braf Date: Tue, 5 Mar 2024 15:34:22 +0000 Subject: [PATCH] General cleanup + todos --- .../perf_analyzer/genai-pa/genai_pa/llm_inputs/llm_inputs.py | 1 + src/c++/perf_analyzer/genai-pa/tests/test_llm_inputs.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/src/c++/perf_analyzer/genai-pa/genai_pa/llm_inputs/llm_inputs.py b/src/c++/perf_analyzer/genai-pa/genai_pa/llm_inputs/llm_inputs.py index 5cb7f1471..d56deafd2 100644 --- a/src/c++/perf_analyzer/genai-pa/genai_pa/llm_inputs/llm_inputs.py +++ b/src/c++/perf_analyzer/genai-pa/genai_pa/llm_inputs/llm_inputs.py @@ -26,6 +26,7 @@ class InputType(Enum): URL = auto() FILE = auto() + SYNTHETIC = auto() class InputFormat(Enum): diff --git a/src/c++/perf_analyzer/genai-pa/tests/test_llm_inputs.py b/src/c++/perf_analyzer/genai-pa/tests/test_llm_inputs.py index 5e2c7278d..34aafb3c4 100644 --- a/src/c++/perf_analyzer/genai-pa/tests/test_llm_inputs.py +++ b/src/c++/perf_analyzer/genai-pa/tests/test_llm_inputs.py @@ -37,6 +37,8 @@ def default_configured_url(self): yield default_configured_url + # TODO: Add tests that verify json schemas + def test_input_type_url_no_model_name(self): """ Test for exception when input type is URL and no model name