Nicholai/jan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add tool choice type

Browse Source
This commit is contained in:
Akarshan 2025-06-23 16:45:00 +05:30 committed by Louis
parent 6d5251d1c6
commit 6b86baaa2f
No known key found for this signature in database
GPG Key ID: 44FA9F4D33C37DE2
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
core/src/browser/extensions/engines/AIEngine.ts
View File

@ -38,10 +38,22 @@ export interface ToolCallOptions {
tools?: Tool[]; tools?: Tool[];
} }
// A specific tool choice to force the model to call
export interface ToolCallSpec {
type: 'function';
function: {
name: string;
};
}
// tool_choice may be one of several modes or a specific call
export type ToolChoice = 'none' | 'auto' | 'required' | ToolCallSpec;
export interface chatCompletionRequest { export interface chatCompletionRequest {
model: string; // Model ID, though for local it might be implicit via sessionInfo model: string; // Model ID, though for local it might be implicit via sessionInfo
messages: chatCompletionRequestMessage[]; messages: chatCompletionRequestMessage[];
tools?: Tool[]; tools?: Tool[];
tool_choice?: ToolChoice;
// Core sampling parameters // Core sampling parameters
temperature?: number | null temperature?: number | null
dynatemp_range?: number | null dynatemp_range?: number | null