lemma-sdk 0.2.44 → 0.2.46

package/dist/react/useAssistantController.js

@@ -1,6 +1,7 @@
 import { useCallback, useEffect, useMemo, useRef, useState } from "react";
 import { useAssistantRuntime } from "./useAssistantRuntime.js";
 import { useAssistantSession } from "./useAssistantSession.js";
+const CONVERSATIONS_PAGE_SIZE = 30;
 const EMPTY_SCOPE_KEY = JSON.stringify({
     podId: null,
     assistantName: null,
@@ -10,43 +11,6 @@ const EMPTY_SCOPE_KEY = JSON.stringify({
 function isRecord(value) {
     return !!value && typeof value === "object" && !Array.isArray(value);
 }
-function stringifyContent(value) {
-    if (typeof value === "string")
-        return value;
-    if (Array.isArray(value)) {
-        const text = value
-            .map((entry) => extractTextFromStructuredContentEntry(entry))
-            .filter((entry) => entry.length > 0)
-            .join("\n\n")
-            .trim();
-        if (text.length > 0)
-            return text;
-        return "";
-    }
-    if (!isRecord(value))
-        return "";
-    const direct = value.content;
-    if (typeof direct === "string")
-        return direct;
-    if (Array.isArray(direct)) {
-        const text = direct
-            .map((entry) => extractTextFromStructuredContentEntry(entry))
-            .filter((entry) => entry.length > 0)
-            .join("\n\n")
-            .trim();
-        if (text.length > 0)
-            return text;
-    }
-    const text = value.text;
-    if (typeof text === "string")
-        return text;
-    try {
-        return JSON.stringify(value, null, 2);
-    }
-    catch {
-        return "";
-    }
-}
 function parseMaybeJsonObject(value) {
     if (isRecord(value))
         return value;
@@ -71,37 +35,6 @@ function parseMaybeJsonValue(value) {
         return value;
     }
 }
-function extractTextFromStructuredContentEntry(entry) {
-    if (typeof entry === "string")
-        return entry.trim();
-    if (!isRecord(entry))
-        return "";
-    if (typeof entry.text === "string")
-        return entry.text.trim();
-    if (typeof entry.content === "string")
-        return entry.content.trim();
-    if (typeof entry.value === "string")
-        return entry.value.trim();
-    if (Array.isArray(entry.content)) {
-        const nested = entry.content
-            .map((child) => extractTextFromStructuredContentEntry(child))
-            .filter((text) => text.length > 0)
-            .join("\n")
-            .trim();
-        if (nested.length > 0)
-            return nested;
-    }
-    if (Array.isArray(entry.summary)) {
-        const summary = entry.summary
-            .map((child) => extractTextFromStructuredContentEntry(child))
-            .filter((text) => text.length > 0)
-            .join("\n")
-            .trim();
-        if (summary.length > 0)
-            return summary;
-    }
-    return "";
-}
 function parseTimestampMs(value) {
     if (typeof value === "number" && Number.isFinite(value)) {
         return value;
@@ -137,27 +70,19 @@ function parseThinkingDurationFromRecord(record) {
         ?? parseDurationMs(record.thought_duration_ms)
         ?? parseDurationMs(record.thoughtDurationMs);
 }
-function extractThinkingPartFromContent(content) {
-    if (!isRecord(content))
-        return null;
-    const rawType = typeof content.type === "string" ? content.type.toLowerCase() : "";
-    if (rawType !== "thinking" && rawType !== "reasoning") {
+function extractThinkingPart(msg) {
+    if (msg.kind !== "thinking")
         return null;
-    }
-    const text = extractTextFromStructuredContentEntry(content.content ?? content.text ?? content.value ?? content.summary);
+    const text = typeof msg.text === "string" ? msg.text.trim() : "";
     if (!text)
         return null;
-    const stateValue = [content.state, content.status, content.phase]
-        .find((value) => typeof value === "string");
-    const normalizedState = stateValue?.toLowerCase() || "";
-    const isStreaming = normalizedState.includes("stream")
-        || normalizedState.includes("progress")
-        || normalizedState.includes("running")
-        || normalizedState.includes("thinking");
+    const metadata = getMessageMetadata(msg);
     return {
         text,
-        state: isStreaming ? "streaming" : "done",
-        durationMs: parseThinkingDurationFromRecord(content),
+        state: "done",
+        durationMs: metadata
+            ? parseThinkingDurationFromRecord(metadata)
+            : undefined,
     };
 }
 function normalizeToolResult(value) {
@@ -172,143 +97,33 @@ function normalizeToolResult(value) {
 function getMessageMetadata(msg) {
     return (msg.message_metadata || msg.metadata || undefined);
 }
-function getNativeToolPayload(content) {
-    if (!isRecord(content))
-        return null;
-    const toolCallId = typeof content.tool_call_id === "string" ? content.tool_call_id : null;
-    if (!toolCallId)
-        return null;
-    const toolName = typeof content.tool_name === "string" ? content.tool_name : undefined;
-    if ("tool_output" in content) {
+function getNativeToolPayload(msg) {
+    const toolName = typeof msg.tool_name === "string" ? msg.tool_name : undefined;
+    if (msg.kind === "tool_call") {
         return {
-            kind: "result",
-            toolCallId,
+            kind: "call",
+            toolCallId: (typeof msg.tool_call_id === "string" && msg.tool_call_id) || `${msg.id}-tool-call`,
             toolName,
-            result: normalizeToolResult(content.tool_output),
+            args: parseMaybeJsonObject(parseMaybeJsonValue(msg.tool_args)),
         };
     }
-    if (typeof content.tool_name === "string") {
+    if (msg.kind === "tool_return") {
         return {
-            kind: "call",
-            toolCallId,
+            kind: "result",
+            toolCallId: (typeof msg.tool_call_id === "string" && msg.tool_call_id) || `${msg.id}-tool-result`,
             toolName,
-            args: parseMaybeJsonObject(parseMaybeJsonValue(content.tool_input)),
+            result: normalizeToolResult(msg.tool_result),
         };
     }
     return null;
 }
-function hasNativeToolPayloadContent(content) {
-    return getNativeToolPayload(content) !== null;
-}
 function toolInvocationKey(tool) {
     return `${tool.toolCallId}:${tool.state}`;
 }
-function toolInvocationFromStructuredContentEntry(entry, fallbackId) {
-    const type = typeof entry.type === "string" ? entry.type.toLowerCase() : "";
-    const functionObj = isRecord(entry.function) ? entry.function : {};
-    const hasToolShape = type.includes("tool")
-        || type.includes("function_call")
-        || type.includes("function_result")
-        || typeof entry.tool_call_id === "string"
-        || typeof entry.call_id === "string"
-        || typeof entry.tool_name === "string"
-        || typeof functionObj.name === "string"
-        || "tool_output" in entry
-        || ("result" in entry && typeof entry.call_id === "string");
-    if (!hasToolShape)
-        return null;
-    const rawResult = entry.tool_output ?? entry.output ?? entry.result;
-    const isResultLike = type.includes("result")
-        || type.includes("output")
-        || type.includes("return")
-        || typeof rawResult !== "undefined";
-    const toolCallId = ((typeof entry.tool_call_id === "string" && entry.tool_call_id)
-        || (typeof entry.toolCallId === "string" && entry.toolCallId)
-        || (typeof entry.call_id === "string" && entry.call_id)
-        || (typeof entry.id === "string" && entry.id)
-        || fallbackId);
-    const toolName = ((typeof entry.tool_name === "string" && entry.tool_name)
-        || (typeof entry.toolName === "string" && entry.toolName)
-        || (typeof functionObj.name === "string" && functionObj.name)
-        || (typeof entry.name === "string" && entry.name)
-        || "tool");
-    const argsRaw = functionObj.arguments
-        ?? entry.tool_input
-        ?? entry.input
-        ?? entry.args
-        ?? entry.arguments;
-    const state = isResultLike ? "result" : "call";
-    return {
-        toolCallId,
-        toolName,
-        args: parseMaybeJsonObject(parseMaybeJsonValue(argsRaw)),
-        state,
-        ...(isResultLike ? { result: normalizeToolResult(rawResult) } : {}),
-    };
-}
-function parseStructuredAssistantParts(content) {
-    if (!Array.isArray(content))
-        return null;
-    const parts = [];
-    const textChunks = [];
-    const representedToolKeys = new Set();
-    content.forEach((rawPart, index) => {
-        if (!isRecord(rawPart))
-            return;
-        const partType = typeof rawPart.type === "string" ? rawPart.type.toLowerCase() : "";
-        const partId = (typeof rawPart.id === "string" && rawPart.id) || `content-part-${index}`;
-        const toolInvocation = toolInvocationFromStructuredContentEntry(rawPart, `${partId}-tool`);
-        if (toolInvocation) {
-            representedToolKeys.add(toolInvocationKey(toolInvocation));
-            parts.push({
-                id: `${partId}-tool`,
-                type: "tool",
-                toolInvocation,
-            });
-            return;
-        }
-        const text = extractTextFromStructuredContentEntry(rawPart);
-        if (!text)
-            return;
-        if (partType.includes("reasoning") || partType.includes("thinking")) {
-            const stateValue = [rawPart.state, rawPart.status, rawPart.phase]
-                .find((value) => typeof value === "string");
-            const normalizedState = stateValue?.toLowerCase() || partType;
-            const isStreaming = normalizedState.includes("stream")
-                || normalizedState.includes("progress")
-                || normalizedState.includes("running")
-                || normalizedState.includes("thinking");
-            parts.push({
-                id: `${partId}-reasoning`,
-                type: "reasoning",
-                text,
-                state: isStreaming ? "streaming" : "done",
-                durationMs: parseThinkingDurationFromRecord(rawPart),
-                startedAtMs: parseTimestampMs(rawPart.started_at)
-                    ?? parseTimestampMs(rawPart.startedAt)
-                    ?? parseTimestampMs(rawPart.created_at)
-                    ?? parseTimestampMs(rawPart.createdAt)
-                    ?? undefined,
-            });
-            return;
-        }
-        textChunks.push(text);
-        parts.push({
-            id: `${partId}-text`,
-            type: "text",
-            text,
-        });
-    });
-    return {
-        parts,
-        textContent: textChunks.join("\n\n").trim(),
-        representedToolKeys,
-    };
-}
 function mapToolInvocations(msg) {
     const invocations = [];
     const metadata = getMessageMetadata(msg);
-    const nativeToolPayload = getNativeToolPayload(msg.content);
+    const nativeToolPayload = getNativeToolPayload(msg);
     if (metadata?.message_type === "tool_call") {
         invocations.push({
             toolCallId: metadata.tool_call_id || `${msg.id}-tool-call`,
@@ -362,20 +177,6 @@ function mapToolInvocations(msg) {
             result: nativeToolPayload.result || {},
         });
     }
-    const contentObj = isRecord(msg.content) ? msg.content : null;
-    if (contentObj && nativeToolPayload === null && "tool_output" in contentObj) {
-        invocations.push({
-            toolCallId: (typeof contentObj.tool_call_id === "string" && contentObj.tool_call_id)
-                || metadata?.tool_call_id
-                || `${msg.id}-tool-output`,
-            toolName: (typeof contentObj.tool_name === "string" && contentObj.tool_name)
-                || metadata?.tool_name
-                || "tool",
-            args: metadata?.args || {},
-            state: "result",
-            result: normalizeToolResult(contentObj.tool_output),
-        });
-    }
     const seen = new Set();
     return invocations.filter((invocation) => {
         const key = toolInvocationKey(invocation);
@@ -387,43 +188,34 @@ function mapToolInvocations(msg) {
 }
 function mapConversationMessage(msg, options) {
     const toolInvocations = mapToolInvocations(msg);
-    const structured = parseStructuredAssistantParts(msg.content);
-    const explicitThinkingPart = extractThinkingPartFromContent(msg.content);
     const createdAtMs = parseTimestampMs(msg.created_at) ?? undefined;
-    const parts = structured
-        ? structured.parts.map((part) => (part.type === "reasoning"
-            ? {
-                ...part,
-                startedAtMs: part.startedAtMs ?? createdAtMs,
-            }
-            : part))
-        : [];
-    const representedToolKeys = structured?.representedToolKeys || new Set();
-    let content = structured
-        ? structured.textContent
-        : (hasNativeToolPayloadContent(msg.content) ? "" : stringifyContent(msg.content));
-    if (explicitThinkingPart) {
-        content = "";
+    const parts = [];
+    let content = "";
+    // Flat shape: a message is exactly one kind. Thinking renders as a reasoning
+    // part; text/notification render as a text part; tool_call/tool_return render
+    // via toolInvocations below.
+    const thinkingPart = extractThinkingPart(msg);
+    if (thinkingPart) {
         parts.push({
             id: `${msg.id}-reasoning`,
             type: "reasoning",
-            text: explicitThinkingPart.text,
-            state: explicitThinkingPart.state,
-            durationMs: explicitThinkingPart.durationMs ?? options?.thinkingDurationMs,
+            text: thinkingPart.text,
+            state: thinkingPart.state,
+            durationMs: thinkingPart.durationMs ?? options?.thinkingDurationMs,
             startedAtMs: createdAtMs,
         });
     }
-    else if (!structured && content.trim()) {
-        parts.push({
-            id: `${msg.id}-text`,
-            type: "text",
-            text: content,
-        });
+    else if (msg.kind === "text" || msg.kind === "notification") {
+        content = typeof msg.text === "string" ? msg.text.trim() : "";
+        if (content) {
+            parts.push({
+                id: `${msg.id}-text`,
+                type: "text",
+                text: content,
+            });
+        }
     }
     toolInvocations.forEach((toolInvocation, index) => {
-        const key = toolInvocationKey(toolInvocation);
-        if (representedToolKeys.has(key))
-            return;
         parts.push({
             id: `${msg.id}-tool-${index}`,
             type: "tool",
@@ -442,8 +234,11 @@ function mapConversationMessage(msg, options) {
         agent_run_id: msg.agent_run_id,
         metadata: msg.metadata ?? null,
         message_metadata: msg.message_metadata ?? null,
+        kind: msg.kind,
         tool_call_id: msg.tool_call_id ?? null,
         tool_name: msg.tool_name ?? null,
+        tool_args: msg.tool_args ?? null,
+        tool_result: msg.tool_result ?? null,
     };
 }
 function mapConversationMessages(messages) {
@@ -451,7 +246,7 @@ function mapConversationMessages(messages) {
     const pendingToolCalls = new Map();
     const estimateThinkingDurationMs = (index) => {
         const message = messages[index];
-        if (!message || !extractThinkingPartFromContent(message.content))
+        if (!message || !extractThinkingPart(message))
             return undefined;
         const startedAtMs = parseTimestampMs(message.created_at);
         if (!startedAtMs)
@@ -477,7 +272,7 @@ function mapConversationMessages(messages) {
                 pendingToolCalls.set(invocation.toolCallId, invocation);
             }
         });
-        const nativePayload = getNativeToolPayload(rawMessage.content);
+        const nativePayload = getNativeToolPayload(rawMessage);
         const isToolRole = rawMessage.role === "tool";
         if (isToolRole && nativePayload?.kind === "result" && mappedMessage.toolInvocations && mappedMessage.toolInvocations.length > 0) {
             let mergedIntoPriorCall = false;
@@ -625,6 +420,8 @@ export function useAssistantController({ client, podId, agentName, assistantName
     const [conversationRuntime, setConversationRuntimeState] = useState(null);
     const [isStreaming, setIsStreaming] = useState(false);
     const [isLoadingConversations, setIsLoadingConversations] = useState(false);
+    const [isLoadingMoreConversations, setIsLoadingMoreConversations] = useState(false);
+    const [conversationsCursor, setConversationsCursor] = useState(null);
     const [isLoadingMessages, setIsLoadingMessages] = useState(false);
     const [isLoadingOlderMessages, setIsLoadingOlderMessages] = useState(false);
     const [isUploadingFiles, setIsUploadingFiles] = useState(false);
@@ -739,9 +536,10 @@ export function useAssistantController({ client, podId, agentName, assistantName
     const loadConversations = useCallback(async () => {
         setIsLoadingConversations(true);
         try {
-            const response = await sessionListConversations({ scope });
+            const response = await sessionListConversations({ scope, limit: CONVERSATIONS_PAGE_SIZE });
             const nextConversations = sortConversationsByUpdatedAt(response.items || []);
             setConversations(nextConversations);
+            setConversationsCursor(response.next_page_token ?? null);
             setActiveConversationId((current) => {
                 if (current && nextConversations.some((conversation) => conversation.id === current)) {
                     return current;
@@ -759,6 +557,36 @@ export function useAssistantController({ client, podId, agentName, assistantName
             setIsLoadingConversations(false);
         }
     }, [scope, sessionListConversations]);
+    const loadMoreConversations = useCallback(async () => {
+        if (!conversationsCursor || isLoadingConversations || isLoadingMoreConversations) {
+            return [];
+        }
+        setIsLoadingMoreConversations(true);
+        try {
+            const response = await sessionListConversations({
+                scope,
+                limit: CONVERSATIONS_PAGE_SIZE,
+                pageToken: conversationsCursor,
+            });
+            const moreConversations = response.items || [];
+            setConversations((prev) => {
+                const byId = new Map(prev.map((conversation) => [conversation.id, conversation]));
+                for (const conversation of moreConversations) {
+                    byId.set(conversation.id, conversation);
+                }
+                return sortConversationsByUpdatedAt(Array.from(byId.values()));
+            });
+            setConversationsCursor(response.next_page_token ?? null);
+            return moreConversations;
+        }
+        catch (err) {
+            setLocalError((prev) => prev || (err instanceof Error ? err.message : "Failed to load more conversations"));
+            return [];
+        }
+        finally {
+            setIsLoadingMoreConversations(false);
+        }
+    }, [conversationsCursor, isLoadingConversations, isLoadingMoreConversations, scope, sessionListConversations]);
     const loadAvailableModels = useCallback(async () => {
         try {
             const response = await client.conversations.listModels({
@@ -909,9 +737,11 @@ export function useAssistantController({ client, podId, agentName, assistantName
             setConversationModelState(null);
             setConversationRuntimeState(null);
             setConversations([]);
+            setConversationsCursor(null);
             setLocalError(null);
             setOlderMessagesCursor(null);
             setIsLoadingConversations(false);
+            setIsLoadingMoreConversations(false);
             setIsLoadingMessages(false);
             setIsLoadingOlderMessages(false);
             return;
@@ -925,6 +755,7 @@ export function useAssistantController({ client, podId, agentName, assistantName
         setConversationModelState(null);
         setConversationRuntimeState(null);
         setConversations([]);
+        setConversationsCursor(null);
         setLocalError(null);
         clearRuntimeMessages();
         setOlderMessagesCursor(null);
@@ -1075,6 +906,7 @@ export function useAssistantController({ client, podId, agentName, assistantName
         const createdConversation = await sessionCreateConversation({
             title: titleSeed.slice(0, 120),
             instructions: typeof options.instructions === "undefined" ? instructions : options.instructions,
+            metadata: options.metadata ?? undefined,
             model: conversationModel,
             agentRuntime: conversationRuntime,
             ...scope,
@@ -1136,7 +968,10 @@ export function useAssistantController({ client, podId, agentName, assistantName
         let conversationId = forceNewConversation ? null : activeConversationId;
         try {
             if (!conversationId) {
-                conversationId = await ensureConversation(trimmed, { instructions: options.instructions });
+                conversationId = await ensureConversation(trimmed, {
+                    instructions: options.instructions,
+                    metadata: options.conversationMetadata,
+                });
             }
             if (!conversationId) {
                 throw new Error("Conversation could not be initialized");
@@ -1282,6 +1117,8 @@ export function useAssistantController({ client, podId, agentName, assistantName
         isActiveConversationRunning,
         isLoading,
         isLoadingConversations,
+        isLoadingMoreConversations,
+        hasMoreConversations: !!conversationsCursor,
         isLoadingMessages,
         isLoadingOlderMessages,
         hasOlderMessages: !!olderMessagesCursor,
@@ -1299,6 +1136,7 @@ export function useAssistantController({ client, podId, agentName, assistantName
         removePendingFile,
         clearPendingFiles,
         loadOlderMessages,
+        loadMoreConversations,
         resolveUserApproval,
         clearMessages,
         stop,
@@ -1312,12 +1150,15 @@ export function useAssistantController({ client, podId, agentName, assistantName
         conversationModel,
         conversations,
         error,
+        conversationsCursor,
         isActiveConversationRunning,
         isLoading,
         isLoadingConversations,
+        isLoadingMoreConversations,
         isLoadingMessages,
         isLoadingOlderMessages,
         isUploadingFiles,
+        loadMoreConversations,
         loadOlderMessages,
         messages,
         olderMessagesCursor,

package/dist/react/useAssistantRuntime.js

@@ -1,18 +1,6 @@
 import { useCallback, useEffect, useRef, useState } from "react";
-function isRecord(value) {
-    return !!value && typeof value === "object" && !Array.isArray(value);
-}
-function messageText(content) {
-    if (typeof content === "string") {
-        return content.trim();
-    }
-    if (isRecord(content)) {
-        if (typeof content.content === "string")
-            return content.content.trim();
-        if (typeof content.text === "string")
-            return content.text.trim();
-    }
-    return "";
+function messageText(message) {
+    return typeof message.text === "string" ? message.text.trim() : "";
 }
 function messageTime(message) {
     const timestamp = new Date(message.created_at).getTime();
@@ -30,7 +18,7 @@ function upsertRuntimeMessage(previous, incoming) {
         return next;
     }
     if (incoming.role === "user") {
-        const incomingText = messageText(incoming.content);
+        const incomingText = messageText(incoming);
         if (incomingText) {
             const incomingTimestamp = messageTime(incoming);
             let optimisticIndex = -1;
@@ -38,7 +26,7 @@ function upsertRuntimeMessage(previous, incoming) {
             next.forEach((message, index) => {
                 if (message.role !== "user"
                     || !isOptimisticId(message.id)
-                    || messageText(message.content) !== incomingText) {
+                    || messageText(message) !== incomingText) {
                     return;
                 }
                 const distance = Math.abs(messageTime(message) - incomingTimestamp);
@@ -100,7 +88,8 @@ export function useAssistantRuntime({ conversationId = null, sessionConversation
         const optimistic = {
             id: buildOptimisticId(),
             role: "user",
-            content: { content: trimmed },
+            kind: "text",
+            text: trimmed,
             created_at: new Date().toISOString(),
             metadata: null,
             ...(optimisticConversationId ? { conversation_id: optimisticConversationId } : {}),

package/dist/react/useAssistantSession.d.ts

@@ -40,10 +40,13 @@ export interface UseAssistantSessionOptions {
 export interface CreateConversationInput {
     title?: string | null;
     instructions?: string | null;
+    metadata?: Record<string, unknown> | null;
     model?: ConversationModel | null;
     agentRuntime?: AgentRuntimeConfig | null;
     podId?: string | null;
     agentName?: string | null;
+    /** Parent conversation id for sub-agent (child) conversations. */
+    parentId?: string | null;
     /**
      * @deprecated Use agentName instead.
      */
@@ -81,9 +84,9 @@ export interface UseAssistantSessionResult {
     status?: string;
     messages: ConversationMessage[];
     latestAssistantMessage: ConversationMessage | null;
-    output: ConversationMessage["content"] | null;
+    output: ConversationMessage | null;
     outputText: string;
-    finalOutput: ConversationMessage["content"] | null;
+    finalOutput: ConversationMessage | null;
     finalOutputText: string;
     streamingText: string;
     streamingTool: AssistantStreamingTool | null;