apunkt/rowboat
1
0
Fork 0
mirror of https://github.com/rowboatlabs/rowboat.git synced 2026-05-06 05:42:37 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

config to specify graph model

Browse source
This commit is contained in:
Arjun 2026-02-27 09:38:00 +05:30
parent cccb7a8a65
commit 83fe81ef5a
3 changed files with 53 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

7
apps/x/packages/core/src/agents/runtime.ts
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);
console.log(`[main] [GraphBuilder] Agent "${state.agentName}" using model: ${modelId}`);
let loopCounter = 0;
while (true) {

1
apps/x/packages/shared/src/models.ts
View file

@ -10,4 +10,5 @@ export const LlmProvider = z.object({
export const LlmModelConfig = z.object({
provider: LlmProvider,
model: z.string(),
knowledgeGraphModel: z.string().optional(),
});