Skip to content

Commit bf2205b

Browse files
committed
fix: tool-calls in streaming mode
1 parent e2bb30c commit bf2205b

34 files changed

+1673
-341
lines changed

dist/cjs/models/chat.cjs

Lines changed: 353 additions & 80 deletions
Large diffs are not rendered by default.

dist/cjs/models/chat.d.cts

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,14 @@ export type ToolChoiceInput = LanguageModelV1ToolChoice;
8484
*/
8585
export declare class HerokuChatLanguageModel implements LanguageModelV1 {
8686
private readonly model;
87-
private readonly apiKey;
88-
private readonly baseUrl;
8987
readonly specificationVersion: "v1";
9088
readonly provider: "heroku";
9189
readonly modelId: string;
9290
readonly defaultObjectGenerationMode: "json";
91+
private readonly apiKey;
92+
private readonly baseUrl;
93+
private streamingToolCalls;
94+
private textBuffer;
9395
/**
9496
* Creates a new HerokuChatLanguageModel instance.
9597
*
@@ -305,13 +307,24 @@ export declare class HerokuChatLanguageModel implements LanguageModelV1 {
305307
rawPrompt: unknown;
306308
rawSettings: Record<string, unknown>;
307309
};
310+
rawResponse: {
311+
headers: {
312+
[k: string]: string;
313+
} | undefined;
314+
};
315+
warnings: never[];
308316
}>;
309317
private mapPromptToMessages;
310318
private convertMessageToHerokuFormat;
311319
private mapToolsToHerokuFormat;
312320
private mapToolChoiceToHerokuFormat;
313321
private mapResponseToOutput;
314322
private mapChunkToStreamPart;
323+
/**
324+
* Helper method to check if a string is valid JSON using provider-utils
325+
* @private
326+
*/
327+
private isValidJson;
315328
private extractToolCalls;
316329
}
317330
//# sourceMappingURL=chat.d.ts.map

dist/cjs/models/chat.d.ts.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

dist/cjs/models/chat.js.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

dist/esm/models/chat.d.ts

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,14 @@ export type ToolChoiceInput = LanguageModelV1ToolChoice;
8484
*/
8585
export declare class HerokuChatLanguageModel implements LanguageModelV1 {
8686
private readonly model;
87-
private readonly apiKey;
88-
private readonly baseUrl;
8987
readonly specificationVersion: "v1";
9088
readonly provider: "heroku";
9189
readonly modelId: string;
9290
readonly defaultObjectGenerationMode: "json";
91+
private readonly apiKey;
92+
private readonly baseUrl;
93+
private streamingToolCalls;
94+
private textBuffer;
9395
/**
9496
* Creates a new HerokuChatLanguageModel instance.
9597
*
@@ -305,13 +307,24 @@ export declare class HerokuChatLanguageModel implements LanguageModelV1 {
305307
rawPrompt: unknown;
306308
rawSettings: Record<string, unknown>;
307309
};
310+
rawResponse: {
311+
headers: {
312+
[k: string]: string;
313+
} | undefined;
314+
};
315+
warnings: never[];
308316
}>;
309317
private mapPromptToMessages;
310318
private convertMessageToHerokuFormat;
311319
private mapToolsToHerokuFormat;
312320
private mapToolChoiceToHerokuFormat;
313321
private mapResponseToOutput;
314322
private mapChunkToStreamPart;
323+
/**
324+
* Helper method to check if a string is valid JSON using provider-utils
325+
* @private
326+
*/
327+
private isValidJson;
315328
private extractToolCalls;
316329
}
317330
//# sourceMappingURL=chat.d.ts.map

dist/esm/models/chat.d.ts.map

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)