Skip to content

Commit

Permalink
Changed optional to Partial for model config param
Browse files Browse the repository at this point in the history
  • Loading branch information
abhimasand committed Jul 21, 2022
1 parent c02d44f commit 657dc24
Show file tree
Hide file tree
Showing 4 changed files with 20 additions and 20 deletions.
2 changes: 1 addition & 1 deletion src/ChatEngine.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ export class ChatEngine extends PromptEngine {
description: string = "",
examples: Interaction[] = [],
flowResetText: string = "",
languageConfig: IChatConfig = DefaultChatConfig
languageConfig: Partial<IChatConfig> = DefaultChatConfig
) {
super(description, examples, flowResetText);
this.languageConfig = { ...DefaultChatConfig, ...languageConfig};
Expand Down
2 changes: 1 addition & 1 deletion src/CodeEngine.ts
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ export class CodeEngine extends PromptEngine {
description: string = "",
examples: Interaction[] = [],
flowResetText: string = "",
languageConfig: ICodePromptConfig = JavaScriptConfig
languageConfig: Partial<ICodePromptConfig> = JavaScriptConfig
) {
super(description, examples, flowResetText);
this.languageConfig = { ...JavaScriptConfig, ...languageConfig};
Expand Down
2 changes: 1 addition & 1 deletion src/PromptEngine.ts
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ export class PromptEngine implements IPromptEngine {
description: string = "",
examples: Interaction[] = [],
flowResetText: string = "",
promptConfig: IPromptConfig = DefaultPromptConfig
promptConfig: Partial<IPromptConfig> = DefaultPromptConfig
) {
this.description = description;
this.examples = examples;
Expand Down
34 changes: 17 additions & 17 deletions src/types.ts
Original file line number Diff line number Diff line change
Expand Up @@ -13,30 +13,30 @@ export interface Interaction {
}

export interface IModelConfig {
maxTokens?: number;
maxTokens: number;
}

export interface IPromptConfig {
modelConfig?: IModelConfig;
descriptionPrefix?: string;
descriptionPostfix?: string;
inputPrefix?: string;
inputPostfix?: string;
outputPrefix?: string;
outputPostfix?: string;
newlineOperator?: string;
modelConfig: IModelConfig;
descriptionPrefix: string;
descriptionPostfix: string;
inputPrefix: string;
inputPostfix: string;
outputPrefix: string;
outputPostfix: string;
newlineOperator: string;
}

export interface ICodePromptConfig {
modelConfig?: IModelConfig;
commentOperator?: string;
closeCommentOperator?: string;
newlineOperator?: string;
modelConfig: IModelConfig;
commentOperator: string;
closeCommentOperator: string;
newlineOperator: string;
}

export interface IChatConfig {
modelConfig?: IModelConfig;
userName?: string;
botName?: string;
newlineOperator?: string;
modelConfig: IModelConfig;
userName: string;
botName: string;
newlineOperator: string;
}

0 comments on commit 657dc24

Please sign in to comment.