Merge pull request #102 from TraderAlice/dev

feat: session awareness tools, dev workbench, telegram onboarding

Author: luokerenx4

src/connectors/mcp-ask/mcp-ask-plugin.ts

@@ -61,7 +61,7 @@ export class McpAskPlugin implements Plugin {
           const session = await plugin.getSession(sessionId)
 
           const result = await ctx.agentCenter.askWithSession(message, session, {
-            historyPreamble: 'The following is the conversation from an external MCP client. Use it as context if the caller references earlier messages.',
+            historyPreamble: `You are operating via the MCP Ask connector (session: mcp-ask__${sessionId}). The following is the conversation from an external MCP client.`,
           })
 
           return {

src/connectors/telegram/telegram-plugin.ts

@@ -10,7 +10,7 @@ import { askAgentSdk } from '../../ai-providers/agent-sdk/query.js'
 import type { AgentSdkConfig } from '../../ai-providers/agent-sdk/query.js'
 import { SessionStore } from '../../core/session'
 import { forceCompact } from '../../core/compaction'
-import { readAIBackend, writeAIBackend, type AIBackend } from '../../core/config'
+import { readAIBackend, writeAIBackend, readConnectorsConfig, type AIBackend } from '../../core/config'
 import type { ConnectorCenter } from '../../core/connector-center.js'
 import { TelegramConnector, splitMessage, MAX_MESSAGE_LENGTH } from './telegram-connector.js'
 
@@ -34,6 +34,7 @@ export class TelegramPlugin implements Plugin {
 
   /** Throttle: last time we sent an auth-guidance reply per chatId. */
   private authReplyThrottle = new Map<number, number>()
+  private webPort = 3002
 
   constructor(
     config: Omit<TelegramConfig, 'pollingTimeout'> & { pollingTimeout?: number },
@@ -45,6 +46,7 @@ export class TelegramPlugin implements Plugin {
 
   async start(engineCtx: EngineContext) {
     this.connectorCenter = engineCtx.connectorCenter
+    this.webPort = engineCtx.config.connectors.web.port
 
     // Inject agent config into Claude Code config (used by /compact command)
     this.agentSdkConfig = {
@@ -63,19 +65,21 @@ export class TelegramPlugin implements Plugin {
       console.error('telegram bot error:', err)
     })
 
-    // ── Middleware: auth guard (always active) ──
+    // ── Middleware: auth guard (hot-reloads chatIds from connectors.json) ──
     bot.use(async (ctx, next) => {
       const chatId = ctx.chat?.id
       if (!chatId) return
-      if (this.config.allowedChatIds.includes(chatId)) return next()
+      const { telegram } = await readConnectorsConfig()
+      if (telegram.chatIds.includes(chatId)) return next()
 
       // Unauthorized — log chat ID for operator, throttle reply (60s)
       const now = Date.now()
       const last = this.authReplyThrottle.get(chatId) ?? 0
       if (now - last > 60_000) {
         this.authReplyThrottle.set(chatId, now)
-        console.log(`telegram: unauthorized chat ${chatId}, set TELEGRAM_CHAT_ID=${chatId} to allow`)
-        await ctx.reply('This chat is not authorized. Add this chat ID to TELEGRAM_CHAT_ID in your environment config.').catch(() => {})
+        const link = `http://localhost:${this.webPort}/connectors?addChatId=${chatId}`
+        console.log(`telegram: unauthorized chat ${chatId}, authorize via ${link}`)
+        await ctx.reply(`To authorize this chat, open:\n${link}`).catch(() => {})
       }
     })
 
@@ -239,7 +243,7 @@ export class TelegramPlugin implements Plugin {
         // Route through AgentCenter → GenerateRouter → active provider
         const session = await this.getSession(message.from.id)
         const result = await engineCtx.agentCenter.askWithSession(prompt, session, {
-          historyPreamble: 'The following is the recent conversation from this Telegram chat. Use it as context if the user references earlier messages.',
+          historyPreamble: `You are operating via Telegram (session: telegram/${message.from.id}). The following is the recent conversation.`,
         })
         stopTyping()
         await this.sendReplyWithPlaceholder(message.chatId, result.text, result.media, placeholder?.message_id)

src/connectors/telegram/types.ts

@@ -4,7 +4,7 @@ export type { Update, Message, Chat, User }
 
 export interface TelegramConfig {
   token: string
-  /** Chat IDs allowed to interact. Empty = allow all. */
+  /** Chat IDs allowed to interact. Empty = reject all. */
   allowedChatIds: number[]
   /** Polling timeout in seconds (Telegram long-poll parameter). Default: 30 */
   pollingTimeout: number

src/connectors/web/routes/chat.ts

@@ -35,7 +35,7 @@ export function createChatRoutes({ ctx, sessions, sseByChannel }: ChatDeps) {
 
     // Build AskOptions from channel config (if not default)
     const opts: AskOptions = {
-      historyPreamble: 'The following is the recent conversation from the Web UI. Use it as context if the user references earlier messages.',
+      historyPreamble: `You are operating via the Web UI (session: web/${channelId}). The following is the recent conversation.`,
     }
     if (channelId !== 'default') {
       const channels = await readWebSubchannels()

src/connectors/web/routes/tools.ts

@@ -1,11 +1,14 @@
 import { Hono } from 'hono'
+import { z } from 'zod'
 import type { ToolCenter } from '../../../core/tool-center.js'
 import { readToolsConfig, writeConfigSection } from '../../../core/config.js'
+import { extractMcpShape, wrapToolExecute } from '../../../core/mcp-export.js'
 
-/** Tools routes: GET / (inventory + disabled), PUT / (update disabled list) */
+/** Tools routes: inventory, detail, execute, enable/disable */
 export function createToolsRoutes(toolCenter: ToolCenter) {
   const app = new Hono()
 
+  /** GET / — inventory + disabled list */
   app.get('/', async (c) => {
     try {
       const inventory = toolCenter.getInventory()
@@ -16,6 +19,7 @@ export function createToolsRoutes(toolCenter: ToolCenter) {
     }
   })
 
+  /** PUT / — update disabled list */
   app.put('/', async (c) => {
     try {
       const body = await c.req.json()
@@ -29,5 +33,47 @@ export function createToolsRoutes(toolCenter: ToolCenter) {
     }
   })
 
+  /** GET /:name — full tool detail with JSON Schema */
+  app.get('/:name', (c) => {
+    const name = c.req.param('name')
+    const tool = toolCenter.get(name)
+    if (!tool) return c.json({ error: `Tool not found: ${name}` }, 404)
+
+    let inputSchema: unknown = {}
+    try {
+      inputSchema = z.toJSONSchema(tool.inputSchema as z.ZodType)
+    } catch { /* fallback to empty */ }
+
+    return c.json({
+      name,
+      group: toolCenter.getGroup(name),
+      description: tool.description ?? '',
+      inputSchema,
+    })
+  })
+
+  /** POST /:name/execute — execute a tool with given input */
+  app.post('/:name/execute', async (c) => {
+    const name = c.req.param('name')
+    const tool = toolCenter.get(name)
+    if (!tool) return c.json({ error: `Tool not found: ${name}` }, 404)
+
+    const rawInput = await c.req.json().catch(() => ({}))
+
+    // Validate + coerce through MCP shape (handles string→number etc.)
+    const shape = extractMcpShape(tool)
+    const schema = z.object(shape)
+    let validated: Record<string, unknown>
+    try {
+      validated = await schema.parseAsync(rawInput)
+    } catch (err) {
+      return c.json({ error: 'Validation failed', details: String(err) }, 400)
+    }
+
+    const execute = wrapToolExecute(tool)
+    const result = await execute(validated)
+    return c.json(result)
+  })
+
   return app
 }

src/core/config.ts

@@ -433,6 +433,16 @@ export async function readToolsConfig() {
   }
 }
 
+/** Read connectors config from disk (called per-request for hot-reload). */
+export async function readConnectorsConfig() {
+  try {
+    const raw = JSON.parse(await readFile(resolve(CONFIG_DIR, 'connectors.json'), 'utf-8'))
+    return connectorsSchema.parse(raw)
+  } catch {
+    return connectorsSchema.parse({})
+  }
+}
+
 // ==================== AI Backend Helpers ====================
 
 export type AIBackend = 'claude-code' | 'vercel-ai-sdk' | 'agent-sdk'

src/core/connector-center.ts

@@ -47,6 +47,17 @@ export class ConnectorCenter {
   private lastInteraction: LastInteraction | null = null
 
   constructor(eventLog?: EventLog) {
+    // Restore last interaction from event log buffer (survives restart)
+    if (eventLog) {
+      const recent = eventLog.recent({ type: 'message.received' })
+      if (recent.length > 0) {
+        const last = recent[recent.length - 1]
+        const { channel, to } = last.payload as { channel: string; to: string }
+        this.lastInteraction = { channel, to, ts: last.ts }
+      }
+    }
+
+    // Subscribe to future interactions
     eventLog?.subscribeType('message.received', (entry) => {
       const { channel, to } = entry.payload as { channel: string; to: string }
       this.touch(channel, to)

src/core/tool-center.ts

@@ -55,6 +55,16 @@ export class ToolCenter {
     }))
   }
 
+  /** Look up a single tool by name (for detail / execute endpoints). */
+  get(name: string): Tool | null {
+    return this.tools[name]?.tool ?? null
+  }
+
+  /** Look up a tool's group by name. */
+  getGroup(name: string): string | null {
+    return this.tools[name]?.group ?? null
+  }
+
   /** Tool name list (for logging / debugging). */
   list(): string[] {
     return Object.keys(this.tools)

src/main.ts

@@ -26,6 +26,7 @@ import { OpenBBCommodityClient } from './domain/market-data/client/openbb-api/co
 import { OpenBBServerPlugin } from './server/opentypebb.js'
 import { createMarketSearchTools } from './tool/market.js'
 import { createAnalysisTools } from './tool/analysis.js'
+import { createSessionTools } from './tool/session.js'
 import { SessionStore } from './core/session.js'
 import { ConnectorCenter } from './core/connector-center.js'
 import { ToolCenter } from './core/tool-center.js'
@@ -227,6 +228,9 @@ async function main() {
 
   const connectorCenter = new ConnectorCenter(eventLog)
 
+  // Session awareness tools (registered here because they need connectorCenter)
+  toolCenter.register(createSessionTools(connectorCenter), 'session')
+
   // ==================== Cron Lifecycle ====================
 
   await cronEngine.start()

src/task/cron/listener.ts

@@ -63,7 +63,7 @@ export function createCronListener(opts: CronListenerOpts): CronListener {
     try {
       // Ask the AI engine with the cron payload
       const result = await agentCenter.askWithSession(payload.payload, session, {
-        historyPreamble: 'The following is the recent cron session conversation. This is an automated cron job execution.',
+        historyPreamble: `You are operating in the cron job context (session: cron/default, job: ${payload.jobName}). This is an automated cron job execution.`,
       })
 
       // Send notification through the last-interacted connector