Browse Source

fix prompt ci

Alex Cheema 8 months ago
parent
commit
7f72604853
1 changed files with 4 additions and 0 deletions
  1. 4 0
      .circleci/config.yml

+ 4 - 0
.circleci/config.yml

@@ -188,6 +188,7 @@ jobs:
       - run_chatgpt_api_test:
       - run_chatgpt_api_test:
           inference_engine: mlx
           inference_engine: mlx
           model_id: llama-3.2-1b
           model_id: llama-3.2-1b
+          prompt: "Keep responses concise. Who was the king of pop?"
           expected_output: "Michael Jackson"
           expected_output: "Michael Jackson"
 
 
   chatgpt_api_integration_test_dummy:
   chatgpt_api_integration_test_dummy:
@@ -211,6 +212,7 @@ jobs:
       - run_chatgpt_api_test:
       - run_chatgpt_api_test:
           inference_engine: dummy
           inference_engine: dummy
           model_id: dummy-model
           model_id: dummy-model
+          prompt: "Dummy prompt."
           expected_output: "dummy"
           expected_output: "dummy"
 
 
   test_macos_m1:
   test_macos_m1:
@@ -242,6 +244,8 @@ jobs:
   #     - run_chatgpt_api_test:
   #     - run_chatgpt_api_test:
   #         inference_engine: tinygrad
   #         inference_engine: tinygrad
   #         model_id: llama-3-8b
   #         model_id: llama-3-8b
+  #         prompt: "Keep responses concise. Who was the king of pop?"
+  #         expected_output: "Michael Jackson"
 
 
 workflows:
 workflows:
   version: 2
   version: 2