feat: migrate call-omo-agent tool callers to SDK message finders

This commit is contained in:
YeonGyu-Kim
2026-02-15 14:57:15 +09:00
parent 2bf8b15f24
commit 553817c1a0
4 changed files with 66 additions and 20 deletions
@@ -1,21 +1,38 @@
import type { BackgroundManager } from "../../features/background-agent"
import { findFirstMessageWithAgent, findNearestMessageWithFields } from "../../features/hook-message-injector"
import type { PluginInput } from "@opencode-ai/plugin"
import {
findFirstMessageWithAgent,
findFirstMessageWithAgentFromSDK,
findNearestMessageWithFields,
findNearestMessageWithFieldsFromSDK,
} from "../../features/hook-message-injector"
import { getSessionAgent } from "../../features/claude-code-session-state"
import { log } from "../../shared"
import type { CallOmoAgentArgs } from "./types"
import type { ToolContextWithMetadata } from "./tool-context-with-metadata"
import { getMessageDir } from "./message-storage-directory"
import { getSessionTools } from "../../shared/session-tools-store"
import { isSqliteBackend } from "../../shared/opencode-storage-detection"
export async function executeBackgroundAgent(
args: CallOmoAgentArgs,
toolContext: ToolContextWithMetadata,
manager: BackgroundManager,
client: PluginInput["client"],
): Promise<string> {
try {
const messageDir = getMessageDir(toolContext.sessionID)
const prevMessage = messageDir ? findNearestMessageWithFields(messageDir) : null
const firstMessageAgent = messageDir ? findFirstMessageWithAgent(messageDir) : null
const [prevMessage, firstMessageAgent] = isSqliteBackend()
? await Promise.all([
findNearestMessageWithFieldsFromSDK(client, toolContext.sessionID),
findFirstMessageWithAgentFromSDK(client, toolContext.sessionID),
])
: [
messageDir ? findNearestMessageWithFields(messageDir) : null,
messageDir ? findFirstMessageWithAgent(messageDir) : null,
]
const sessionAgent = getSessionAgent(toolContext.sessionID)
const parentAgent =
toolContext.agent ?? sessionAgent ?? firstMessageAgent ?? prevMessage?.agent