mirror of https://github.com/buster-so/buster.git
update agents
This commit is contained in:
parent
40a44cf1f9
commit
09c96ab4f9
|
@ -1,35 +1,35 @@
|
|||
import { Agent } from "@mastra/core";
|
||||
import { Agent } from '@mastra/core';
|
||||
import {
|
||||
executeSql,
|
||||
messageUserClarifyingQuestion,
|
||||
respondWithoutAssetCreation,
|
||||
sequentialThinking,
|
||||
submitThoughts,
|
||||
} from "../../tools";
|
||||
import { GPT5 } from "../../utils";
|
||||
import { Sonnet4 } from "../../utils/models/sonnet-4";
|
||||
executeSql,
|
||||
messageUserClarifyingQuestion,
|
||||
respondWithoutAssetCreation,
|
||||
sequentialThinking,
|
||||
submitThoughts,
|
||||
} from '../../tools';
|
||||
import { GPT5 } from '../../utils';
|
||||
import { Sonnet4 } from '../../utils/models/sonnet-4';
|
||||
|
||||
const DEFAULT_OPTIONS = {
|
||||
maxSteps: 25,
|
||||
temperature: 1,
|
||||
providerOptions: {
|
||||
anthropic: {
|
||||
disableParallelToolCalls: true,
|
||||
},
|
||||
},
|
||||
maxSteps: 25,
|
||||
temperature: 1,
|
||||
providerOptions: {
|
||||
anthropic: {
|
||||
disableParallelToolCalls: true,
|
||||
},
|
||||
},
|
||||
};
|
||||
|
||||
export const thinkAndPrepAgent = new Agent({
|
||||
name: "Think and Prep Agent",
|
||||
instructions: "", // We control the system messages in the step at stream instantiation
|
||||
model: GPT5,
|
||||
tools: {
|
||||
sequentialThinking,
|
||||
executeSql,
|
||||
respondWithoutAssetCreation,
|
||||
submitThoughts,
|
||||
messageUserClarifyingQuestion,
|
||||
},
|
||||
defaultGenerateOptions: DEFAULT_OPTIONS,
|
||||
defaultStreamOptions: DEFAULT_OPTIONS,
|
||||
name: 'Think and Prep Agent',
|
||||
instructions: '', // We control the system messages in the step at stream instantiation
|
||||
model: GPT5,
|
||||
tools: {
|
||||
sequentialThinking,
|
||||
executeSql,
|
||||
respondWithoutAssetCreation,
|
||||
submitThoughts,
|
||||
messageUserClarifyingQuestion,
|
||||
},
|
||||
defaultGenerateOptions: DEFAULT_OPTIONS,
|
||||
defaultStreamOptions: DEFAULT_OPTIONS,
|
||||
});
|
||||
|
|
|
@ -8,4 +8,4 @@ export const openaiModel = (modelId: string) => {
|
|||
|
||||
// Wrap the model with Braintrust tracing and return it
|
||||
return wrapAISDKModel(openai(modelId));
|
||||
};
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue