From 47b6d5892dd20f12c3198f192c44c40beb77a19b Mon Sep 17 00:00:00 2001
From: tylertitsworth <tyler.titsworth@intel.com>
Date: Tue, 9 Jul 2024 16:02:15 -0700
Subject: [PATCH] add and validate smoke tests

Signed-off-by: tylertitsworth <tyler.titsworth@intel.com>
---
 AudioQnA/tests/tests.yaml    | 4 ++--
 ChatQnA/tests/tests.yaml     | 3 ++-
 CodeGen/tests/tests.yaml     | 3 ++-
 CodeTrans/tests/tests.yaml   | 3 ++-
 DocSum/tests/tests.yaml      | 3 ++-
 SearchQnA/tests/tests.yaml   | 3 ++-
 Translation/tests/tests.yaml | 3 ++-
 VisualQnA/tests/tests.yaml   | 3 ++-
 8 files changed, 16 insertions(+), 9 deletions(-)

diff --git a/AudioQnA/tests/tests.yaml b/AudioQnA/tests/tests.yaml
index 7e11647eb..7b0749c0d 100644
--- a/AudioQnA/tests/tests.yaml
+++ b/AudioQnA/tests/tests.yaml
@@ -4,5 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/audioqna:${GITHUB_RUN_NUMBER:-0}
-  cmd: python -c "from comps import AudioQnAGateway, MicroService, ServiceOrchestrator, ServiceType"
-  entrypoint: ""
+  cmd: -c "from comps import AudioQnAGateway, MicroService, ServiceOrchestrator, ServiceType"
+  entrypoint: "python"
diff --git a/ChatQnA/tests/tests.yaml b/ChatQnA/tests/tests.yaml
index 59691d365..ffe53707c 100644
--- a/ChatQnA/tests/tests.yaml
+++ b/ChatQnA/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/chatqna:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import ChatQnAGateway, MicroService, ServiceOrchestrator, ServiceType"
+  entrypoint: "python"
diff --git a/CodeGen/tests/tests.yaml b/CodeGen/tests/tests.yaml
index 3924135a2..ef30751bb 100644
--- a/CodeGen/tests/tests.yaml
+++ b/CodeGen/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/codegen:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import CodeGenGateway, MicroService, ServiceOrchestrator, ServiceType"
+  entrypoint: "python"
diff --git a/CodeTrans/tests/tests.yaml b/CodeTrans/tests/tests.yaml
index bed087600..2322b3a6e 100644
--- a/CodeTrans/tests/tests.yaml
+++ b/CodeTrans/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/codetrans:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import CodeTransGateway, MicroService, ServiceOrchestrator"
+  entrypoint: "python"
diff --git a/DocSum/tests/tests.yaml b/DocSum/tests/tests.yaml
index 0ce5fb8f9..9ff4feac1 100644
--- a/DocSum/tests/tests.yaml
+++ b/DocSum/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/docsum:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import DocSumGateway, MicroService, ServiceOrchestrator, ServiceType"
+  entrypoint: "python"
diff --git a/SearchQnA/tests/tests.yaml b/SearchQnA/tests/tests.yaml
index 22fd89fc2..d64773513 100644
--- a/SearchQnA/tests/tests.yaml
+++ b/SearchQnA/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/searchqna:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import MicroService, SearchQnAGateway, ServiceOrchestrator, ServiceType"
+  entrypoint: "python"
diff --git a/Translation/tests/tests.yaml b/Translation/tests/tests.yaml
index 440aebc8d..51df659b7 100644
--- a/Translation/tests/tests.yaml
+++ b/Translation/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/translation:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: -c "from comps import MicroService, ServiceOrchestrator, ServiceType, TranslationGateway"
+  entrypoint: "python"
diff --git a/VisualQnA/tests/tests.yaml b/VisualQnA/tests/tests.yaml
index 7bb716406..f8ff3b1b6 100644
--- a/VisualQnA/tests/tests.yaml
+++ b/VisualQnA/tests/tests.yaml
@@ -4,4 +4,5 @@
 # https://github.com/intel/ai-containers/tree/main/test-runner
 import-comps:
   img: ${REGISTRY}/visualqna:${GITHUB_RUN_NUMBER:-0}
-  cmd: -h
+  cmd: llava_server.py -h
+  entrypoint: "python"