@@ -2,6 +2,7 @@ import { expect } from 'chai';
22import { describe , it } from 'mocha' ;
33import { Context } from 'mocha' ;
44import sinon from 'sinon' ;
5+ import * as path from 'path' ;
56import { parseMessage , getInstructionFiles , parseMessageAndSetOptions , getParentHash , handleTopic , handlerResponseText , sendMessageBase , stopDevChatBase } from '../../src/handler/sendMessageBase' ;
67import DevChat , { ChatResponse } from '../../src/toolwrapper/devchat' ;
78import CommandManager from '../../src/command/commandManager' ;
@@ -10,6 +11,10 @@ import { TopicManager } from '../../src/topic/topicManager';
1011import CustomCommands from '../../src/command/customCommand' ;
1112import { UiUtilWrapper } from '../../src/util/uiUtil' ;
1213
14+ import * as dotenv from 'dotenv' ;
15+
16+ const envPath = path . join ( __dirname , '../../' , '.env' ) ;
17+ dotenv . config ( { path : envPath } ) ;
1318
1419describe ( 'sendMessageBase' , ( ) => {
1520 let workspaceFoldersFirstPathStub : sinon . SinonStub ;
@@ -194,7 +199,7 @@ describe('sendMessageBase', () => {
194199
195200 workspaceFoldersFirstPathStub . returns ( './' ) ;
196201
197- getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( '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' ) ;
202+ getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( process . env . TEST_OPENAI_API_KEY ) ;
198203 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.model' ) . returns ( 'gpt-4' ) ;
199204 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.temperature' ) . returns ( 0 ) ;
200205 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.stream' ) . returns ( 'true' ) ;
@@ -221,7 +226,7 @@ describe('sendMessageBase', () => {
221226
222227 workspaceFoldersFirstPathStub . returns ( './' ) ;
223228
224- getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( 'DC.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJvcmdfaWQiOjY2MTI0NDU1ODE2LCJqdGkiOjcyMzc4ODIzMjI3Mjc4MzI2NTJ9.qGWJ_NyWjjj66oa5mbfi3Hjawe-Yp8syCDLkpyu4yS0' ) ;
229+ getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( process . env . TEST_DEVCHAT_KEY ) ;
225230 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.model' ) . returns ( 'gpt-4' ) ;
226231 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.temperature' ) . returns ( 0 ) ;
227232 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.stream' ) . returns ( 'true' ) ;
@@ -277,7 +282,7 @@ describe('sendMessageBase', () => {
277282
278283 workspaceFoldersFirstPathStub . returns ( './' ) ;
279284
280- getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( 'DC.eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJvcmdfaWQiOjY2MTI0NDU1ODE2LCJqdGkiOjcyMzc4ODIzMjI3Mjc4MzI2NTJ9.qGWJ_NyWjjj66oa5mbfi3Hjawe-Yp8syCDLkpyu4yS0' ) ;
285+ getConfigurationStub . withArgs ( 'DevChat' , 'API_KEY' ) . returns ( process . env . TEST_DEVCHAT_KEY ) ;
281286 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.model' ) . returns ( 'gpt-4' ) ;
282287 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.temperature' ) . returns ( 0 ) ;
283288 getConfigurationStub . withArgs ( 'DevChat' , 'OpenAI.stream' ) . returns ( 'true' ) ;
0 commit comments