@anvia/studio 1.0.1 → 1.0.4

package/README.md

@@ -1,6 +1,7 @@
 # @anvia/studio
 
-Studio UI and HTTP runtime for Anvia agents, pipelines, tools, MCPs, memory, status, and knowledge inspection.
+Studio UI and HTTP runtime for Anvia agents, pipelines, graphs, tools, MCPs, memory, status, and
+knowledge inspection.
 
 Use this package to serve local agents and pipelines over HTTP, inspect sessions, traces, tools, MCPs, Memory, Status, and Knowledge in the browser UI, and exercise tool approval workflows during development.
 
@@ -35,7 +36,7 @@ const agent = new Agent({
   instructions: "Answer support questions clearly.",
 });
 
-new Studio([agent]).start({
+await new Studio([agent]).serve({
   port: 4021,
 });
 ```
@@ -46,6 +47,25 @@ Then open:
 http://localhost:4021/ui/playground
 ```
 
+## Graceful shutdown
+
+`serve()` handles both `SIGINT` and `SIGTERM`. Studio stops accepting work, aborts active Agent and
+Pipeline runs, waits for their cancellation observers, and then runs `onShutdown`. Use that callback
+to close observability clients or other caller-owned resources:
+
+```ts
+await new Studio([agent]).serve({
+  port: 4021,
+  shutdownTimeoutMs: 30_000,
+  onShutdown: async () => {
+    await Promise.all([lens.close(), langfuse.close(), otelSdk.shutdown()]);
+  },
+});
+```
+
+`shutdown()` provides the same draining behavior for application-managed lifecycles. `close()`
+remains synchronous for compatibility: it aborts active work but does not wait for cleanup.
+
 ## Multi-Provider Models
 
 Studio can expose a shared model catalog and let each agent choose from registered providers:
@@ -123,6 +143,7 @@ Studio exposes:
 - Realtime observability stream for session logs, pipeline logs, and completed traces
 - Eval suite runner for registered `runEvalSuite` configurations
 - Pipeline graph, logs, run history, and replay-from-history controls
+- Knowledge-graph explorer with type filters, search, node details, and bounded neighborhood expansion
 - Rich agent runtime details, direct tool invocation, static tool, dynamic tool, and MCP inspectors
 - Memory explorer for users, conversations, messages, and transcript steps backed by the session store
 - Status dashboard for storage adapters, record counts, and enabled capabilities
@@ -132,6 +153,27 @@ Studio reads MCP provenance directly from `Agent.mcpServers`. Prefixes configure
 `McpClient` remain visible in both the MCP inspector and direct tool runner, while the remote tool
 name stays available on the typed MCP registration.
 
+### Graph explorer
+
+Register any graph that implements the provider-neutral `GraphExplorer` contract. Both
+`@anvia/neo4j` and `@anvia/memgraph` graph registrations can be passed directly:
+
+```ts
+const studio = new Studio([agent], {
+  graphs: [
+    {
+      id: "support",
+      name: "Support knowledge graph",
+      graph,
+    },
+  ],
+});
+```
+
+Open `/graphs` and select nodes to inspect their public properties or expand their one-hop
+neighborhood. Studio exposes only bounded overview and expansion requests; it does not accept raw
+Cypher. Adapter explorer responses also omit stored embeddings and reserved `__anvia_*` properties.
+
 ## Session Storage
 
 Studio uses an in-memory store by default. Sessions, traces, and pipeline run history are available while the process is running, but they do not create local files unless you pass an explicit SQLite store. If you omit the port, Studio uses `RUNNER_PORT` and then falls back to `4021`.
@@ -155,7 +197,7 @@ SQLite storage uses dedicated `anvia_studio_*` tables so it can share an applica
 - `Studio`
 - `createInMemoryStudioStore`
 - `createSqliteSessionStore`
-- Studio session, trace, approval, pipeline, memory, status, knowledge, tool, MCP, and runtime types
+- Studio session, trace, approval, pipeline, graph, memory, status, knowledge, tool, MCP, and runtime types
 
 ## Development
 

package/dist/index.d.ts

@@ -6,10 +6,11 @@ import { JsonObject, Message, JsonValue, CompletionModelCapabilities, ToolResult
 import { MemoryStore, MemoryAppendOptions, MemoryErrorOptions, MemoryScope } from '@anvia/core/memory';
 import { ModelList } from '@anvia/core/model-listing';
 import { Pipeline, PipelineGraph } from '@anvia/core/pipeline';
+import { GraphExplorer } from '@anvia/graph';
 
 declare function traceSummary(trace: StudioTrace): StudioTraceSummary;
 
-type StudioCapability = "agents" | "interactions" | "evals" | "memory" | "knowledge" | "mcps" | "observability" | "pipelines" | "sandboxes" | "sessions" | "status" | "tools" | "traces";
+type StudioCapability = "agents" | "interactions" | "evals" | "graphs" | "memory" | "knowledge" | "mcps" | "observability" | "pipelines" | "sandboxes" | "sessions" | "status" | "tools" | "traces";
 type AgentTraceInfo = {
     observer: string;
     traceId?: string | undefined;
@@ -147,6 +148,43 @@ type StudioPipelineConfig = {
 type StudioPipelineDetail = StudioPipelineConfig & {
     graph: PipelineGraph;
 };
+type StudioGraphRegistration = {
+    id: string;
+    graph: GraphExplorer;
+    name?: string;
+    description?: string;
+};
+type StudioGraphConfig = {
+    id: string;
+    name?: string;
+    description?: string;
+    nodeTypes: Array<{
+        name: string;
+        description: string;
+    }>;
+    relationshipTypes: Array<{
+        name: string;
+        description: string;
+        from: string;
+        to: string;
+    }>;
+};
+type StudioGraphExploreRequest = {
+    mode: "overview";
+    nodeTypes?: string[];
+    relationships?: string[];
+    maxNodes?: number;
+    maxRelationships?: number;
+} | {
+    mode: "expand";
+    nodeIds: string[];
+    nodeTypes?: string[];
+    relationships?: string[];
+    direction?: "outgoing" | "incoming" | "both";
+    maxDepth?: number;
+    maxNodes?: number;
+    maxRelationships?: number;
+};
 type StudioEvalSuite<Input = any, _Output = any, _Expected = any> = {
     name: string;
     cases: Array<Input>;
@@ -210,6 +248,7 @@ type StudioConfig = {
     agents: StudioAgentConfig[];
     models?: StudioModelsConfig;
     pipelines: StudioPipelineConfig[];
+    graphs: StudioGraphConfig[];
     evals: StudioEvalSuiteConfig[];
     chat: {
         quickPrompts: Record<string, string[]>;
@@ -581,7 +620,7 @@ type StudioSessionStore = StudioMemoryStore & {
     listSessionLogs?(options: StudioSessionLogListOptions): StudioSessionLogEntry[] | Promise<StudioSessionLogEntry[]>;
     deleteSession?(id: string): boolean | Promise<boolean>;
 };
-type StudioTraceStatus = "running" | "success" | "suspended" | "error";
+type StudioTraceStatus = "running" | "success" | "suspended" | "error" | "cancelled";
 type StudioTraceObservationKind = "agent" | "generation" | "tool";
 type StudioTraceObservation = {
     id: string;
@@ -842,12 +881,14 @@ type StudioOptions = {
     ui?: boolean | StudioUiOptions;
     models?: StudioModelConfig;
     sandboxes?: readonly StudioSandboxRegistration[];
+    graphs?: readonly StudioGraphRegistration[];
 };
 type StudioServeOptions = {
     port?: number;
     hostname?: string;
     log?: boolean;
     handleSignals?: boolean;
+    shutdownTimeoutMs?: number;
 };
 type StudioServeLifecycleOptions = Omit<StudioServeOptions, "handleSignals"> & {
     signal?: AbortSignal;
@@ -1011,7 +1052,7 @@ type AgentRunRequest = (AgentRunRequestBase & {
 });
 type AgentRunResponse = Exclude<AgentOutcome, AgentInteractionOutcome> | Omit<AgentInteractionOutcome, "continuation" | "messages">;
 type AgentRunStreamEvent = AgentStreamEvent | StudioSessionLogEvent | StudioPipelineLogEvent | StudioPipelineFinalEvent;
-type StudioErrorCode = "bad_request" | "conflict" | "forbidden" | "not_found" | "payload_too_large" | "unsupported_capability" | "internal_error";
+type StudioErrorCode = "bad_request" | "conflict" | "forbidden" | "not_found" | "payload_too_large" | "service_unavailable" | "unsupported_capability" | "internal_error";
 type StudioErrorResponse = {
     error: {
         code: StudioErrorCode;
@@ -1025,6 +1066,9 @@ type AnviaStudio = {
     fetch(request: Request): Response | Promise<Response>;
     config(): StudioConfig;
     close(): void;
+    shutdown(options?: {
+        timeoutMs?: number;
+    }): Promise<void>;
 };
 
 type StudioTraceObserverOptions = {
@@ -1042,7 +1086,8 @@ declare class Studio implements AnviaStudio {
     private studio;
     private server;
     private websocketServer;
-    private sigintHandler;
+    private signalHandlers;
+    private shutdownPromise;
     constructor(targets?: StudioTarget[], options?: StudioOptions);
     get app(): Hono;
     fetch(request: Request): Response | Promise<Response>;
@@ -1051,6 +1096,12 @@ declare class Studio implements AnviaStudio {
     start(serveOptions?: StudioServeOptions): this;
     serve(serveOptions?: StudioServeLifecycleOptions): Promise<void>;
     close(): void;
+    shutdown(options?: {
+        timeoutMs?: number;
+    }): Promise<void>;
+    private shutdownResources;
+    private removeSignalHandlers;
+    private closeNetwork;
     private logAddress;
 }
 
@@ -1061,4 +1112,4 @@ declare function createSqliteSessionStore(options?: SqliteSessionStoreOptions):
 
 declare function createInMemoryStudioStore(): StudioSessionStore & StudioTraceStore & StudioPipelineLogStore & StudioPipelineRunStore;
 
-export { type AgentRunRequest, type AgentRunResponse, type AgentRunStreamEvent, type AgentTraceInfo, type AgentTraceOptions, type AnviaStudio, type SqliteSessionStoreOptions, Studio, type StudioAgent, type StudioAgentConfig, type StudioAgentKnowledgeConfig, type StudioAgentMcpServerMetadata, type StudioAgentMcpToolMetadata, type StudioAgentMcpsSummary, type StudioAgentModelPolicy, type StudioAgentModelPolicyConfig, type StudioAgentModelsSummary, type StudioAgentRuntimeSummary, type StudioAgentToolApprovalMetadata, type StudioAgentToolMetadata, type StudioAgentToolSource, type StudioAgentToolsSummary, type StudioCapability, type StudioCapabilityConfig, type StudioConfig, type StudioErrorCode, type StudioErrorResponse, type StudioEvalCasePreview, type StudioEvalMetricSummary, type StudioEvalRunRequest, type StudioEvalRunResponse, type StudioEvalSuite, type StudioEvalSuiteConfig, type StudioKnowledgeEvidence, type StudioKnowledgeEvidenceDocument, type StudioKnowledgeItem, type StudioKnowledgeItemKind, type StudioKnowledgeItemsPage, type StudioKnowledgeSourceKind, type StudioKnowledgeSourceSummary, type StudioKnowledgeSummary, type StudioMemoryAppendOptions, type StudioMemoryConversationMessages, type StudioMemoryConversationSteps, type StudioMemoryConversationSummary, type StudioMemoryConversationsPage, type StudioMemoryErrorOptions, type StudioMemoryMessageRecord, type StudioMemoryScope, type StudioMemorySourceConversationMessages, type StudioMemorySourceConversationSteps, type StudioMemorySourceConversationSummary, type StudioMemorySourceConversationsPage, type StudioMemorySourceKind, type StudioMemorySourceSummary, type StudioMemorySourceUsersPage, type StudioMemorySourcesPage, type StudioMemoryStore, type StudioMemoryUserSummary, type StudioMemoryUsersPage, type StudioModelConfig, type StudioModelDefinition, type StudioModelModalities, type StudioModelModality, type StudioModelProvider, type StudioModelProviderConfig, type StudioModelRef, type StudioModelSummary, type StudioModelsConfig, type StudioObservabilityEvent, type StudioObservabilityEventType, type StudioOptions, type StudioPipeline, type StudioPipelineConfig, type StudioPipelineDetail, type StudioPipelineFinalEvent, type StudioPipelineLogAppendInput, type StudioPipelineLogCategory, type StudioPipelineLogEntry, type StudioPipelineLogEvent, type StudioPipelineLogLevel, type StudioPipelineLogListOptions, type StudioPipelineLogStore, type StudioPipelineReplayRequest, type StudioPipelineRunGetOptions, type StudioPipelineRunListOptions, type StudioPipelineRunRecord, type StudioPipelineRunRequest, type StudioPipelineRunResponse, type StudioPipelineRunSaveInput, type StudioPipelineRunStatus, type StudioPipelineRunStore, type StudioSandboxCapabilities, type StudioSandboxFileEntry, type StudioSandboxFileType, type StudioSandboxFilesResponse, type StudioSandboxInspector, type StudioSandboxInspectorFileEntry, type StudioSandboxInspectorPort, type StudioSandboxInspectorProcess, type StudioSandboxInspectorProcessLogs, type StudioSandboxPort, type StudioSandboxPortsResponse, type StudioSandboxProcess, type StudioSandboxProcessLogsResponse, type StudioSandboxProcessStatus, type StudioSandboxProcessesResponse, type StudioSandboxRegistration, type StudioSandboxSummary, type StudioSandboxViewAccess, type StudioSandboxViewAuthentication, type StudioSandboxViewAuthorizeArgs, type StudioSandboxViewConnection, type StudioSandboxViewControl, type StudioSandboxViewControlLease, type StudioSandboxViewControlSnapshot, type StudioSandboxViewRegistration, type StudioSandboxViewSource, type StudioSandboxViewSummary, type StudioSandboxesSummary, type StudioServeLifecycleOptions, type StudioServeOptions, type StudioSession, type StudioSessionCreateInput, type StudioSessionListOptions, type StudioSessionLogAppendInput, type StudioSessionLogCategory, type StudioSessionLogEntry, type StudioSessionLogEvent, type StudioSessionLogLevel, type StudioSessionLogListOptions, type StudioSessionRunStatus, type StudioSessionRunTranscriptInput, type StudioSessionStore, type StudioSessionSummary, type StudioSessionTraceListOptions, type StudioStaticKnowledgeDocument, type StudioStatusSummary, type StudioStores, type StudioTarget, type StudioToolApprovalStatus, type StudioToolApprovalTranscript, type StudioToolQuestionAnswer, type StudioToolQuestionChoice, type StudioToolQuestionPrompt, type StudioToolQuestionStatus, type StudioToolQuestionTranscript, type StudioToolRunRequest, type StudioToolRunResponse, type StudioTrace, type StudioTraceListOptions, type StudioTraceObservation, type StudioTraceObservationKind, StudioTraceObserver, type StudioTraceObserverOptions, type StudioTraceStatus, type StudioTraceStore, type StudioTraceSummary, type StudioTranscriptAttachment, type StudioTranscriptChatEntry, type StudioTranscriptChildAgentEvent, type StudioTranscriptEntry, type StudioTranscriptReasoningEntry, type StudioTranscriptToolEntry, type StudioUiOptions, createInMemoryStudioStore, createSqliteSessionStore, traceSummary };
+export { type AgentRunRequest, type AgentRunResponse, type AgentRunStreamEvent, type AgentTraceInfo, type AgentTraceOptions, type AnviaStudio, type SqliteSessionStoreOptions, Studio, type StudioAgent, type StudioAgentConfig, type StudioAgentKnowledgeConfig, type StudioAgentMcpServerMetadata, type StudioAgentMcpToolMetadata, type StudioAgentMcpsSummary, type StudioAgentModelPolicy, type StudioAgentModelPolicyConfig, type StudioAgentModelsSummary, type StudioAgentRuntimeSummary, type StudioAgentToolApprovalMetadata, type StudioAgentToolMetadata, type StudioAgentToolSource, type StudioAgentToolsSummary, type StudioCapability, type StudioCapabilityConfig, type StudioConfig, type StudioErrorCode, type StudioErrorResponse, type StudioEvalCasePreview, type StudioEvalMetricSummary, type StudioEvalRunRequest, type StudioEvalRunResponse, type StudioEvalSuite, type StudioEvalSuiteConfig, type StudioGraphConfig, type StudioGraphExploreRequest, type StudioGraphRegistration, type StudioKnowledgeEvidence, type StudioKnowledgeEvidenceDocument, type StudioKnowledgeItem, type StudioKnowledgeItemKind, type StudioKnowledgeItemsPage, type StudioKnowledgeSourceKind, type StudioKnowledgeSourceSummary, type StudioKnowledgeSummary, type StudioMemoryAppendOptions, type StudioMemoryConversationMessages, type StudioMemoryConversationSteps, type StudioMemoryConversationSummary, type StudioMemoryConversationsPage, type StudioMemoryErrorOptions, type StudioMemoryMessageRecord, type StudioMemoryScope, type StudioMemorySourceConversationMessages, type StudioMemorySourceConversationSteps, type StudioMemorySourceConversationSummary, type StudioMemorySourceConversationsPage, type StudioMemorySourceKind, type StudioMemorySourceSummary, type StudioMemorySourceUsersPage, type StudioMemorySourcesPage, type StudioMemoryStore, type StudioMemoryUserSummary, type StudioMemoryUsersPage, type StudioModelConfig, type StudioModelDefinition, type StudioModelModalities, type StudioModelModality, type StudioModelProvider, type StudioModelProviderConfig, type StudioModelRef, type StudioModelSummary, type StudioModelsConfig, type StudioObservabilityEvent, type StudioObservabilityEventType, type StudioOptions, type StudioPipeline, type StudioPipelineConfig, type StudioPipelineDetail, type StudioPipelineFinalEvent, type StudioPipelineLogAppendInput, type StudioPipelineLogCategory, type StudioPipelineLogEntry, type StudioPipelineLogEvent, type StudioPipelineLogLevel, type StudioPipelineLogListOptions, type StudioPipelineLogStore, type StudioPipelineReplayRequest, type StudioPipelineRunGetOptions, type StudioPipelineRunListOptions, type StudioPipelineRunRecord, type StudioPipelineRunRequest, type StudioPipelineRunResponse, type StudioPipelineRunSaveInput, type StudioPipelineRunStatus, type StudioPipelineRunStore, type StudioSandboxCapabilities, type StudioSandboxFileEntry, type StudioSandboxFileType, type StudioSandboxFilesResponse, type StudioSandboxInspector, type StudioSandboxInspectorFileEntry, type StudioSandboxInspectorPort, type StudioSandboxInspectorProcess, type StudioSandboxInspectorProcessLogs, type StudioSandboxPort, type StudioSandboxPortsResponse, type StudioSandboxProcess, type StudioSandboxProcessLogsResponse, type StudioSandboxProcessStatus, type StudioSandboxProcessesResponse, type StudioSandboxRegistration, type StudioSandboxSummary, type StudioSandboxViewAccess, type StudioSandboxViewAuthentication, type StudioSandboxViewAuthorizeArgs, type StudioSandboxViewConnection, type StudioSandboxViewControl, type StudioSandboxViewControlLease, type StudioSandboxViewControlSnapshot, type StudioSandboxViewRegistration, type StudioSandboxViewSource, type StudioSandboxViewSummary, type StudioSandboxesSummary, type StudioServeLifecycleOptions, type StudioServeOptions, type StudioSession, type StudioSessionCreateInput, type StudioSessionListOptions, type StudioSessionLogAppendInput, type StudioSessionLogCategory, type StudioSessionLogEntry, type StudioSessionLogEvent, type StudioSessionLogLevel, type StudioSessionLogListOptions, type StudioSessionRunStatus, type StudioSessionRunTranscriptInput, type StudioSessionStore, type StudioSessionSummary, type StudioSessionTraceListOptions, type StudioStaticKnowledgeDocument, type StudioStatusSummary, type StudioStores, type StudioTarget, type StudioToolApprovalStatus, type StudioToolApprovalTranscript, type StudioToolQuestionAnswer, type StudioToolQuestionChoice, type StudioToolQuestionPrompt, type StudioToolQuestionStatus, type StudioToolQuestionTranscript, type StudioToolRunRequest, type StudioToolRunResponse, type StudioTrace, type StudioTraceListOptions, type StudioTraceObservation, type StudioTraceObservationKind, StudioTraceObserver, type StudioTraceObserverOptions, type StudioTraceStatus, type StudioTraceStore, type StudioTraceSummary, type StudioTranscriptAttachment, type StudioTranscriptChatEntry, type StudioTranscriptChildAgentEvent, type StudioTranscriptEntry, type StudioTranscriptReasoningEntry, type StudioTranscriptToolEntry, type StudioUiOptions, createInMemoryStudioStore, createSqliteSessionStore, traceSummary };