Skip to content

Commit 91611a5

Browse files
authored
Add tests for different input formats (#2006)
Signed-off-by: ZePan110 <ze.pan@intel.com>
1 parent 52eef7e commit 91611a5

2 files changed

Lines changed: 14 additions & 0 deletions

File tree

CodeGen/tests/test_compose_on_gaudi.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,6 +143,13 @@ function validate_megaservice() {
143143
"codegen-gaudi-backend-server" \
144144
'{ "index_name": "test_redis", "agents_flag": "True", "messages": "def print_hello_world():", "max_tokens": 256}'
145145

146+
validate_services \
147+
"${ip_address}:7778/v1/codegen" \
148+
"class" \
149+
"mega-codegen" \
150+
"codegen-xeon-backend-server" \
151+
'{"model": "Qwen/Qwen2.5-Coder-7B-Instruct", "messages": [{"role": "user", "content": "Implement a basic Python class"}], "max_tokens":32}'
152+
146153
}
147154

148155
function validate_frontend() {

CodeGen/tests/test_compose_on_xeon.sh

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,13 @@ function validate_megaservice() {
146146
"codegen-xeon-backend-server" \
147147
'{ "index_name": "test_redis", "agents_flag": "True", "messages": "def print_hello_world():", "max_tokens": 256}'
148148

149+
validate_services \
150+
"${ip_address}:7778/v1/codegen" \
151+
"class" \
152+
"mega-codegen" \
153+
"codegen-xeon-backend-server" \
154+
'{"model": "Qwen/Qwen2.5-Coder-7B-Instruct", "messages": [{"role": "user", "content": "Implement a basic Python class"}], "max_tokens":32}'
155+
149156
}
150157

151158
function validate_frontend() {

0 commit comments

Comments
 (0)