import {
  ensureMainAgent as ensureMainAgentContext,
  IAgentLifecycleService,
  MAIN_AGENT_ID,
  type IAgentScopeHandle,
  type ISessionScopeHandle,
} from '@moonshot-ai/agent-core-v2';

export { MAIN_AGENT_ID };

export async function ensureMainAgent(session: ISessionScopeHandle): Promise<IAgentScopeHandle> {
  const context = await ensureMainAgentContext(session);
  const handle = session.accessor.get(IAgentLifecycleService).handleOf(context.agentId);
  if (handle === undefined) {
    throw new Error('Main agent was not found');
  }
  return handle;
}