Skip to content

Commit 1befdfa

Browse files
committed
Use the default prompt in anthropic provider
1 parent 7aa62fa commit 1befdfa

File tree

1 file changed

+13
-1
lines changed

1 file changed

+13
-1
lines changed

src/llm-models/anthropic-completer.ts

+13-1
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ import { BaseChatModel } from '@langchain/core/language_models/chat_models';
77
import { AIMessage, SystemMessage } from '@langchain/core/messages';
88

99
import { BaseCompleter, IBaseCompleter } from './base-completer';
10+
import { COMPLETION_SYSTEM_PROMPT } from '../provider';
1011

1112
export class AnthropicCompleter implements IBaseCompleter {
1213
constructor(options: BaseCompleter.IOptions) {
@@ -17,6 +18,16 @@ export class AnthropicCompleter implements IBaseCompleter {
1718
return this._anthropicProvider;
1819
}
1920

21+
/**
22+
* Getter and setter for the initial prompt.
23+
*/
24+
get prompt(): string {
25+
return this._prompt;
26+
}
27+
set prompt(value: string) {
28+
this._prompt = value;
29+
}
30+
2031
async fetch(
2132
request: CompletionHandler.IRequest,
2233
context: IInlineCompletionContext
@@ -29,7 +40,7 @@ export class AnthropicCompleter implements IBaseCompleter {
2940

3041
const messages = [
3142
new SystemMessage(
32-
'You are a code-completion AI completing the following code from a Jupyter Notebook cell.'
43+
this._prompt
3344
),
3445
new AIMessage(trimmedPrompt)
3546
];
@@ -62,4 +73,5 @@ export class AnthropicCompleter implements IBaseCompleter {
6273
}
6374

6475
private _anthropicProvider: ChatAnthropic;
76+
private _prompt: string = COMPLETION_SYSTEM_PROMPT;
6577
}

0 commit comments

Comments
 (0)