Skip to content

Commit e16afaf

Browse files
committed
Drop OPENAI_KEY
For some reason, we used both `OPENAI_KEY` and `OPENAI_API_KEY` env variables, expecting them to have the same values. Drop the `OPENAI_KEY` one, since we document our use of `OPENAI_API_KEY`.
1 parent e5e90fa commit e16afaf

File tree

4 files changed

+17
-29
lines changed

4 files changed

+17
-29
lines changed

.github/workflows/ci.yml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,6 @@ jobs:
3939
cache: true
4040
- name: Run tests and generate artifacts
4141
env:
42-
OPENAI_KEY: ${{ secrets.OPENAI_KEY }}
4342
OPENAI_API_KEY: ${{ secrets.OPENAI_KEY }}
4443
AZURE_OPENAI_DEPLOYMENT: ${{ secrets.AZURE_DEPLOYMENT }}
4544
AZURE_OPENAI_ENDPOINT: ${{ secrets.AZURE_ENDPOINT }}

tests/texampleTests.m

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ function setUpAndTearDowns(testCase)
3030
import matlab.unittest.fixtures.CurrentFolderFixture
3131
testCase.applyFixture(CurrentFolderFixture("../examples/mlx-scripts"));
3232

33-
openAIEnvVar = "OPENAI_KEY";
33+
openAIEnvVar = "OPENAI_API_KEY";
3434
secretKey = getenv(openAIEnvVar);
3535
% Create an empty .env file because it is expected by our .mlx
3636
% example files

tests/textractOpenAIEmbeddings.m

Lines changed: 8 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,6 @@
33

44
% Copyright 2023-2024 The MathWorks, Inc.
55

6-
methods (TestClassSetup)
7-
function saveEnvVar(testCase)
8-
% Ensures key is not in environment variable for tests
9-
openAIEnvVar = "OPENAI_API_KEY";
10-
if isenv(openAIEnvVar)
11-
key = getenv(openAIEnvVar);
12-
unsetenv(openAIEnvVar);
13-
testCase.addTeardown(@(x) setenv(openAIEnvVar, x), key);
14-
end
15-
end
16-
end
17-
186
properties(TestParameter)
197
InvalidInput = iGetInvalidInput();
208
ValidDimensionsModelCombinations = iGetValidDimensionsModelCombinations();
@@ -29,6 +17,13 @@ function embedsDifferentStringTypes(testCase)
2917
end
3018

3119
function keyNotFound(testCase)
20+
% Ensures key is not in environment variable for tests
21+
openAIEnvVar = "OPENAI_API_KEY";
22+
if isenv(openAIEnvVar)
23+
key = getenv(openAIEnvVar);
24+
reset = onCleanup(@() setenv(openAIEnvVar, key));
25+
unsetenv(openAIEnvVar);
26+
end
3227
testCase.verifyError(@()extractOpenAIEmbeddings("bla"), "llms:keyMustBeSpecified");
3328
end
3429

@@ -40,8 +35,7 @@ function validCombinationOfModelAndDimension(testCase, ValidDimensionsModelCombi
4035
end
4136

4237
function embedStringWithSuccessfulOpenAICall(testCase)
43-
testCase.verifyWarningFree(@()extractOpenAIEmbeddings("bla", ...
44-
APIKey=getenv("OPENAI_KEY")));
38+
testCase.verifyWarningFree(@()extractOpenAIEmbeddings("bla"));
4539
end
4640

4741
function invalidCombinationOfModelAndDimension(testCase)

tests/topenAIImages.m

Lines changed: 8 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -3,18 +3,6 @@
33

44
% Copyright 2024 The MathWorks, Inc.
55

6-
methods (TestClassSetup)
7-
function saveEnvVar(testCase)
8-
% Ensures key is not in environment variable for tests
9-
openAIEnvVar = "OPENAI_API_KEY";
10-
if isenv(openAIEnvVar)
11-
key = getenv(openAIEnvVar);
12-
testCase.addTeardown(@() setenv(openAIEnvVar, key));
13-
unsetenv(openAIEnvVar);
14-
end
15-
end
16-
end
17-
186
properties(TestParameter)
197
InvalidConstructorInput = iGetInvalidConstructorInput;
208
InvalidGenerateInput = iGetInvalidGenerateInput;
@@ -32,6 +20,13 @@ function generateAcceptsSingleStringAsInput(testCase)
3220
end
3321

3422
function keyNotFound(testCase)
23+
% Ensures key is not in environment variable for tests
24+
openAIEnvVar = "OPENAI_API_KEY";
25+
if isenv(openAIEnvVar)
26+
key = getenv(openAIEnvVar);
27+
reset = onCleanup(@() setenv(openAIEnvVar, key));
28+
unsetenv(openAIEnvVar);
29+
end
3530
testCase.verifyError(@()openAIImages, "llms:keyMustBeSpecified");
3631
end
3732

@@ -131,7 +126,7 @@ function invalidInputsVariation(testCase, InvalidVariationInput)
131126
end
132127

133128
function testThatImageIsReturned(testCase)
134-
mdl = openAIImages(APIKey=getenv("OPENAI_KEY"));
129+
mdl = openAIImages;
135130

136131
[images, response] = generate(mdl, ...
137132
"Create a 3D avatar of a whimsical sushi on the beach. " + ...

0 commit comments

Comments
 (0)