Skip to content

Commit 9f072ca

Browse files
committed
Update GenerativeOpenAI to support new isAzure: true flag
1 parent 9184e62 commit 9f072ca

File tree

3 files changed

+6
-2
lines changed

3 files changed

+6
-2
lines changed

src/collections/config/types/generative.ts

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,8 +75,9 @@ export type GenerativeOpenAIConfig = GenerativeOpenAIConfigBase & {
7575
};
7676

7777
export type GenerativeAzureOpenAIConfig = GenerativeOpenAIConfigBase & {
78-
resourceName: string;
79-
deploymentId: string;
78+
resourceName?: string;
79+
deploymentId?: string;
80+
isAzure?: true;
8081
};
8182

8283
export type GenerativePaLMConfig = {

src/collections/configure/generative.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,7 @@ export default {
8989
return {
9090
name: 'generative-openai',
9191
config: {
92+
isAzure: true,
9293
deploymentId: config.deploymentId,
9394
resourceName: config.resourceName,
9495
baseURL: config.baseURL,

src/collections/configure/unit.test.ts

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1261,6 +1261,7 @@ describe('Unit testing of the generative factory class', () => {
12611261
expect(config).toEqual<ModuleConfig<'generative-openai', GenerativeAzureOpenAIConfig>>({
12621262
name: 'generative-openai',
12631263
config: {
1264+
isAzure: true,
12641265
resourceName: 'resource-name',
12651266
deploymentId: 'deployment-id',
12661267
},
@@ -1281,6 +1282,7 @@ describe('Unit testing of the generative factory class', () => {
12811282
expect(config).toEqual<ModuleConfig<'generative-openai', GenerativeAzureOpenAIConfig>>({
12821283
name: 'generative-openai',
12831284
config: {
1285+
isAzure: true,
12841286
resourceName: 'resource-name',
12851287
deploymentId: 'deployment-id',
12861288
baseURL: 'base-url',

0 commit comments

Comments
 (0)