diff --git a/packages/extension/src/agent/useAgent.ts b/packages/extension/src/agent/useAgent.ts index a4be760..42375fe 100644 --- a/packages/extension/src/agent/useAgent.ts +++ b/packages/extension/src/agent/useAgent.ts @@ -4,6 +4,7 @@ import type { AgentActivity, AgentStatus, + ExecutionResult, HistoricalEvent, SupportedLanguage, } from '@page-agent/core' @@ -32,7 +33,7 @@ export interface UseAgentResult { activity: AgentActivity | null currentTask: string config: ExtConfig | null - execute: (task: string) => Promise + execute: (task: string) => Promise stop: () => void configure: (config: ExtConfig) => Promise } @@ -110,7 +111,7 @@ export function useAgent(): UseAgentResult { setCurrentTask(task) setHistory([]) - await agent.execute(task) + return agent.execute(task) }, []) const stop = useCallback(() => {