beca2c52c3
- Add NoteChart component using Recharts (bar, line, area, pie, radar, funnel, gauge)
- Add generate_chart and insert_chart_in_note AI tools
- Add chart code block support in MarkdownContent (```chart ... ```)
- Support JSON and simple data formats (label: value)
- Add i18n translations for chart features
Chart syntax examples:
- JSON: ```chart {"type":"bar","data":[{"label":"A","value":10}]} ```
- Simple: ```chart\nbar\nSales Data\nJan: 120\nFeb: 150\n```
Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
84 lines
2.3 KiB
TypeScript
84 lines
2.3 KiB
TypeScript
/**
|
|
* Tool Registry
|
|
* Central registry for all agent tools.
|
|
* Tools self-register on import via side-effect in index.ts.
|
|
*/
|
|
|
|
import { tool } from 'ai'
|
|
import { z } from 'zod'
|
|
|
|
export interface ToolContext {
|
|
userId: string
|
|
agentId?: string
|
|
actionId?: string
|
|
conversationId?: string
|
|
notebookId?: string
|
|
webSearch?: boolean
|
|
config: Record<string, string>
|
|
}
|
|
|
|
export interface RegisteredTool {
|
|
name: string
|
|
description: string
|
|
buildTool: (ctx: ToolContext) => any // Returns an AI SDK tool() synchronously
|
|
isInternal: boolean // true = no API key needed
|
|
}
|
|
|
|
class ToolRegistry {
|
|
private tools: Map<string, RegisteredTool> = new Map()
|
|
|
|
register(tool: RegisteredTool): void {
|
|
this.tools.set(tool.name, tool)
|
|
}
|
|
|
|
get(name: string): RegisteredTool | undefined {
|
|
return this.tools.get(name)
|
|
}
|
|
|
|
buildToolsForAgent(toolNames: string[], ctx: ToolContext): Record<string, any> {
|
|
const built: Record<string, any> = {}
|
|
for (const name of toolNames) {
|
|
const registered = this.tools.get(name)
|
|
if (registered) {
|
|
built[name] = registered.buildTool(ctx)
|
|
}
|
|
}
|
|
return built
|
|
}
|
|
|
|
/**
|
|
* Build tools for the chat endpoint.
|
|
* Includes internal tools (note_search, note_read) and web tools when configured.
|
|
* When webOnly is true, only web tools are included (no note access).
|
|
*/
|
|
buildToolsForChat(ctx: ToolContext & { webOnly?: boolean }): Record<string, any> {
|
|
const toolNames: string[] = ctx.webOnly ? [] : ['note_search', 'note_read', 'note_find_and_update', 'document_search', 'task_extract', 'generate_chart', 'insert_chart_in_note']
|
|
|
|
// Add web tools only when user toggled web search AND config is present
|
|
if (ctx.webSearch) {
|
|
const hasWebSearch = ctx.config.WEB_SEARCH_PROVIDER || ctx.config.BRAVE_SEARCH_API_KEY || ctx.config.SEARXNG_URL
|
|
if (hasWebSearch) {
|
|
toolNames.push('web_search')
|
|
}
|
|
|
|
const hasWebScrape = ctx.config.JINA_API_KEY
|
|
if (hasWebScrape) {
|
|
toolNames.push('web_scrape')
|
|
}
|
|
}
|
|
|
|
return this.buildToolsForAgent(toolNames, ctx)
|
|
}
|
|
|
|
getAvailableTools(): Array<{ name: string; description: string; isInternal: boolean }> {
|
|
return Array.from(this.tools.values()).map(t => ({
|
|
name: t.name,
|
|
description: t.description,
|
|
isInternal: t.isInternal,
|
|
}))
|
|
}
|
|
}
|
|
|
|
// Singleton
|
|
export const toolRegistry = new ToolRegistry()
|