orchid-ai 2.1.0 → 2.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "orchid-ai",
-  "version": "2.1.0",
+  "version": "2.1.1",
   "description": "Shared Orchid AI chat UI and visualization components",
   "main": "src/index.js",
   "types": "src/index.d.ts",

package/src/index.d.ts

@@ -173,27 +173,32 @@ export function orchidAiProcessTraceEntryKind(entry: {
   value: string;
 }): 'text' | 'tool' | 'compile' | 'mind';
 
+export type OrchidAiProcessTraceItem =
+  | { type: 'status' | 'text'; value: string }
+  | { type: 'query'; tool: string; input: Record<string, unknown> };
+
 export function createOrchidAiProcessTraceCollector(): {
+  onQuery(tool: string, input: Record<string, unknown>): void;
   onStatus(text: string, opts?: { isClearStream?: boolean }): void;
   onDelta(text: string): void;
   getLiveMain(): string;
   getLiveInterim(): string;
-  getItems(): Array<{ type: 'status' | 'text'; value: string }>;
+  getItems(): Array<OrchidAiProcessTraceItem>;
   reset(): void;
   buildPersistedTrace():
-    | { items: Array<{ type: 'status' | 'text'; value: string }>; defaultCollapsed: boolean }
+    | { items: Array<OrchidAiProcessTraceItem>; defaultCollapsed: boolean }
     | undefined;
 };
 
 export function snapshotOrchidAiProcessTraceItems(
   collector: ReturnType<typeof createOrchidAiProcessTraceCollector>
-): { items: Array<{ type: 'status' | 'text'; value: string }>; defaultCollapsed: boolean };
+): { items: Array<OrchidAiProcessTraceItem>; defaultCollapsed: boolean };
 
 export function augmentLiveProcessTraceSnapshot(
-  trace: { items: Array<{ type: 'status' | 'text'; value: string }>; defaultCollapsed: boolean },
+  trace: { items: Array<OrchidAiProcessTraceItem>; defaultCollapsed: boolean },
   streamingMain: string,
   liveInterim?: string
-): { items: Array<{ type: 'status' | 'text'; value: string }>; defaultCollapsed: boolean };
+): { items: Array<OrchidAiProcessTraceItem>; defaultCollapsed: boolean };
 
 export function orchidAiProcessTraceHasDisplayableContent(
   trace: ChatMessage['processTrace'] | undefined,