Skip to content

Commit 09b8662

Browse files
committed
test NumCompletions
1 parent 949fe42 commit 09b8662

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

tests/tazureChat.m

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,13 @@ function doGenerate(testCase,StringInputs)
4141
testCase.verifyGreaterThan(strlength(response),0);
4242
end
4343

44+
function generateMultipleResponses(testCase)
45+
chat = azureChat(getenv("AZURE_OPENAI_ENDPOINT"), getenv("AZURE_OPENAI_DEPLOYMENT"));
46+
[~,~,response] = generate(chat,"What is a cat?",NumCompletions=3);
47+
testCase.verifySize(response.Body.Data.choices,[3,1]);
48+
end
49+
50+
4451
function doReturnErrors(testCase)
4552
testCase.assumeTrue(isenv("AZURE_OPENAI_API_KEY"),"end-to-end test requires environment variables AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, and AZURE_OPENAI_DEPLOYMENT.");
4653
chat = azureChat(getenv("AZURE_OPENAI_ENDPOINT"), getenv("AZURE_OPENAI_DEPLOYMENT"));

tests/topenAIChat.m

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,12 @@ function generateAcceptsSingleStringAsInput(testCase,StringInputs)
1818
testCase.verifyWarningFree(@()generate(chat,StringInputs));
1919
end
2020

21+
function generateMultipleResponses(testCase)
22+
chat = openAIChat;
23+
[~,~,response] = generate(chat,"What is a cat?",NumCompletions=3);
24+
testCase.verifySize(response.Body.Data.choices,[3,1]);
25+
end
26+
2127
function generateAcceptsMessagesAsInput(testCase)
2228
chat = openAIChat;
2329
messages = messageHistory;

0 commit comments

Comments
 (0)