Skip to content

Commit e511498

Browse files
authored
Merge pull request #1 from ssdeanx/develop
feat: add response modalities to various agents
2 parents 704ee9f + 91dd29c commit e511498

5 files changed

Lines changed: 20 additions & 22 deletions

File tree

src/mastra/agents/acpAgent.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,10 +119,10 @@ export const acpAgent = new Agent({
119119
providerOptions: {
120120
google: {
121121
thinkingConfig: {
122-
123122
includeThoughts: true,
124123
thinkingBudget: -1,
125124
},
125+
responseModalities: ['TEXT', 'IMAGE'],
126126
},
127127
},
128128
};

src/mastra/agents/businessLegalAgents.ts

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -227,7 +227,8 @@ You are a Senior Contract Analyst. Analyze legal documents for risks, obligation
227227
thinkingLevel: 'high',
228228
includeThoughts: true,
229229
thinkingBudget: -1,
230-
}
230+
},
231+
responseModalities: ['TEXT', 'IMAGE'],
231232
}
232233
}
233234
}
@@ -319,7 +320,8 @@ You are a Compliance Officer. Monitor regulatory compliance and identify risks a
319320
thinkingLevel: 'high',
320321
includeThoughts: true,
321322
thinkingBudget: -1,
322-
}
323+
},
324+
responseModalities: ['TEXT', 'IMAGE'],
323325
}
324326
}
325327
}
@@ -416,7 +418,8 @@ You are a Chief Strategy Officer with legal expertise. Align business strategy w
416418
thinkingLevel: 'high',
417419
includeThoughts: true,
418420
thinkingBudget: -1,
419-
}
421+
},
422+
responseModalities: ['TEXT', 'IMAGE'],
420423
}
421424
}
422425
}
@@ -431,10 +434,6 @@ You are a Chief Strategy Officer with legal expertise. Align business strategy w
431434
googleFinanceTool,
432435
googleScholarTool,
433436
webScraperTool,
434-
google_search: googleTools.googleSearch({
435-
mode: "MODE_DYNAMIC",
436-
dynamicThreshold: 0.7,
437-
}),
438437
// Integration tools for coordinating other agents would be added here
439438
},
440439
memory: pgMemory,
@@ -477,12 +476,4 @@ You are a Chief Strategy Officer with legal expertise. Align business strategy w
477476
],
478477
})
479478

480-
// Type definitions for provider metadata (similar to researchAgent)
481-
type ProviderMetadataMap = { google?: GoogleGenerativeAIProviderMetadata } & Record<string, unknown>;
482-
483-
const providerMetadata: ProviderMetadataMap | undefined =
484-
((googleAI as unknown) as { providerMetadata?: ProviderMetadataMap })?.providerMetadata ??
485-
((google as unknown) as { providerMetadata?: ProviderMetadataMap })?.providerMetadata;
486479

487-
const metadata = providerMetadata?.google;
488-
export const groundingMetadata = metadata?.groundingMetadata;

src/mastra/agents/calendarAgent.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ Current user: ${userId ?? 'anonymous'}`,
3737
includeThoughts: true,
3838
thinkingBudget: -1,
3939
},
40+
responseModalities: ['TEXT', 'IMAGE'],
4041
},
4142
},
4243
};

src/mastra/agents/csv_to_excalidraw.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -163,7 +163,8 @@ Structure:
163163
thinkingLevel: 'medium',
164164
includeThoughts: true,
165165
thinkingBudget: -1,
166-
}
166+
},
167+
responseModalities: ['TEXT', 'IMAGE'],
167168
}
168169
}
169170
}

src/mastra/agents/dane.ts

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,8 @@ export const daneCommitMessage = new Agent({
2626
thinkingLevel: 'medium',
2727
includeThoughts: true,
2828
thinkingBudget: -1,
29-
}
29+
},
30+
responseModalities: ['TEXT', 'IMAGE'],
3031
}
3132
}
3233
}
@@ -54,7 +55,8 @@ export const daneIssueLabeler = new Agent({
5455
thinkingLevel: 'medium',
5556
includeThoughts: true,
5657
thinkingBudget: -1,
57-
}
58+
},
59+
responseModalities: ['TEXT', 'IMAGE'],
5860
}
5961
}
6062
}
@@ -88,7 +90,8 @@ export const daneLinkChecker = new Agent({
8890
thinkingLevel: 'low',
8991
includeThoughts: true,
9092
thinkingBudget: -1,
91-
}
93+
},
94+
responseModalities: ['TEXT', 'IMAGE'],
9295
}
9396
}
9497
}
@@ -123,7 +126,8 @@ export const daneChangeLog = new Agent({
123126
thinkingLevel: 'medium',
124127
includeThoughts: true,
125128
thinkingBudget: -1,
126-
}
129+
},
130+
responseModalities: ['TEXT', 'IMAGE'],
127131
}
128132
}
129133
}
@@ -183,7 +187,8 @@ export const dane = new Agent({
183187
thinkingLevel: 'high',
184188
includeThoughts: true,
185189
thinkingBudget: -1,
186-
}
190+
},
191+
responseModalities: ['TEXT', 'IMAGE'],
187192
}
188193
}
189194
}

0 commit comments

Comments
 (0)