Add test for ollamaChat.generate for the case when ollamaChat is cal…

…led with the system prompt defined
matlab-deep-learning · Jun 27, 2024 · e43cb20 · e43cb20
1 parent d7f194c
commit e43cb20
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/tollamaChat.m b/tests/tollamaChat.m
@@ -39,6 +39,13 @@ function doGenerate(testCase,StringInputs)
             testCase.verifyGreaterThan(strlength(response),0);
         end
 
+        function doGenerateUsingSystemPrompt(testCase)
+            chat = ollamaChat("mistral","You are a helpful assistant");
+            response = testCase.verifyWarningFree(@() generate(chat,"Hi"));
+            testCase.verifyClass(response,'string');
+            testCase.verifyGreaterThan(strlength(response),0);
+        end
+
         function extremeTopK(testCase)
             % setting top-k to k=1 leaves no random choice,
             % so we expect to get a fixed response.