(null);
useEffect(() => {
setIsExpanded(data.isExpanded);
}, [data.isExpanded]);
+ useEffect(() => {
+ const runningTask = data.agent?.tasks?.find(
+ (task) =>
+ task.status === "running" && task.toolkits && task.toolkits.length > 0
+ );
+
+ if (runningTask && runningTask.id !== lastAutoExpandedTaskIdRef.current) {
+ if (!isExpanded) {
+ setIsExpanded(true);
+ data.onExpandChange(id, true);
+ setSelectedTask(runningTask);
+ }
+ lastAutoExpandedTaskIdRef.current = runningTask.id;
+ }
+ }, [data.agent?.tasks, id, data.onExpandChange, isExpanded]);
+
// manually control node size
useEffect(() => {
if (data.isEditMode) {
@@ -402,7 +419,7 @@ export function Node({ id, data }: NodeProps) {
{/* {JSON.stringify(data.agent)} */}
{agentToolkits[
diff --git a/src/lib/llm.ts b/src/lib/llm.ts
index fac1cb2a..7d673993 100644
--- a/src/lib/llm.ts
+++ b/src/lib/llm.ts
@@ -73,6 +73,15 @@ export const INIT_PROVODERS: Provider[] = [
is_valid: false,
model_type: ""
},
+ {
+ id: 'ModelArk',
+ name: 'ModelArk',
+ apiKey: '',
+ apiHost: 'https://ark.ap-southeast.bytepluses.com/api/v3',
+ description: "ModelArk model configuration.",
+ is_valid: false,
+ model_type: ""
+ },
{
id: 'aws-bedrock',
name: 'AWS Bedrock',
diff --git a/src/store/chatStore.ts b/src/store/chatStore.ts
index 7f242771..342f5143 100644
--- a/src/store/chatStore.ts
+++ b/src/store/chatStore.ts
@@ -157,6 +157,8 @@ const resolveProcessTaskIdForToolkitEvent = (
// Throttle streaming decompose text updates to prevent excessive re-renders
const streamingDecomposeTextBuffer: Record = {};
const streamingDecomposeTextTimers: Record> = {};
+// TTFT (Time to First Token) tracking for task decomposition
+const ttftTracking: Record = {};
const chatStore = (initial?: Partial) => createStore()(
(set, get) => ({
@@ -747,6 +749,11 @@ const chatStore = (initial?: Partial) => createStore()(
//Enable it for the rest of current SSE session
skipFirstConfirm = false;
+
+ // Record confirmed time for TTFT tracking
+ const ttftTaskId = getCurrentTaskId();
+ ttftTracking[ttftTaskId] = { confirmedAt: performance.now(), firstTokenLogged: false };
+ console.log(`[TTFT] Task ${ttftTaskId} confirmed at ${new Date().toISOString()}, starting TTFT measurement`);
return
}
@@ -785,6 +792,13 @@ const chatStore = (initial?: Partial) => createStore()(
const text = content;
const currentId = getCurrentTaskId();
+ // Log TTFT (Time to First Token) on first decompose_text event
+ if (ttftTracking[currentId] && !ttftTracking[currentId].firstTokenLogged) {
+ ttftTracking[currentId].firstTokenLogged = true;
+ const ttft = performance.now() - ttftTracking[currentId].confirmedAt;
+ console.log(`%c[TTFT] 🚀 Time to First Token: ${ttft.toFixed(2)}ms - First streaming token received for task ${currentId}`, 'color: #4CAF50; font-weight: bold');
+ }
+
// Get current buffer or task state
const currentContent = streamingDecomposeTextBuffer[currentId] ||
getCurrentChatStore().task?.streamingDecomposeText || "";
@@ -818,6 +832,8 @@ const chatStore = (initial?: Partial) => createStore()(
if (agentMessages.step === "to_sub_tasks") {
// Clear streaming decompose text when task splitting is done
clearStreamingDecomposeText();
+ // Clean up TTFT tracking
+ delete ttftTracking[currentTaskId];
// Check if this is a multi-turn scenario after task completion
const isMultiTurnAfterCompletion = task?.status === 'finished';
diff --git a/test/mocks/environmentMocks.ts b/test/mocks/environmentMocks.ts
index 92229db0..51782df5 100644
--- a/test/mocks/environmentMocks.ts
+++ b/test/mocks/environmentMocks.ts
@@ -429,11 +429,19 @@ export function createProcessUtilsMock() {
getVenvsBaseDir: vi.fn(),
cleanupOldVenvs: vi.fn(),
isBinaryExists: vi.fn(),
+ getUvEnv: vi.fn(),
mockState: {} as MockEnvironmentState,
setup: (mockState: MockEnvironmentState) => {
utilsMock.mockState = mockState
+ utilsMock.getUvEnv.mockReturnValue({
+ UV_PYTHON_INSTALL_DIR: `${mockState.system.homedir}/.eigent/cache/uv_python`,
+ UV_TOOL_DIR: `${mockState.system.homedir}/.eigent/cache/uv_tool`,
+ UV_PROJECT_ENVIRONMENT: `${mockState.system.homedir}/.eigent/venvs/backend-mock`,
+ UV_HTTP_TIMEOUT: '300',
+ })
+
utilsMock.getResourcePath.mockReturnValue(
`${mockState.app.appPath}/resources`
)