Config graph model (#411)

* config to specify graph model
This commit is contained in:
arkml 2026-02-27 15:52:19 +05:30 committed by GitHub
parent cccb7a8a65
commit 9df1bb6765
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
4 changed files with 166 additions and 86 deletions

View file

@ -706,7 +706,12 @@ export async function* streamAgent({
// set up provider + model
const provider = createProvider(modelConfig.provider);
const model = provider.languageModel(modelConfig.model);
const knowledgeGraphAgents = ["note_creation", "email-draft", "meeting-prep"];
const modelId = (knowledgeGraphAgents.includes(state.agentName!) && modelConfig.knowledgeGraphModel)
? modelConfig.knowledgeGraphModel
: modelConfig.model;
const model = provider.languageModel(modelId);
logger.log(`using model: ${modelId}`);
let loopCounter = 0;
while (true) {