@arbidocs/react 0.1.0

package/README.md

@@ -0,0 +1,162 @@
+# @arbidocs/react
+
+React hooks and provider for the ARBI SDK. Built on `@arbidocs/core` and `@tanstack/react-query`.
+
+## Install
+
+```bash
+npm install @arbidocs/react
+```
+
+## Quick Start
+
+Wrap your app with `ArbiProvider` and a React Query `QueryClientProvider`:
+
+```tsx
+import { QueryClient, QueryClientProvider } from '@tanstack/react-query'
+import { ArbiProvider } from '@arbidocs/react'
+
+const queryClient = new QueryClient()
+
+function App() {
+  return (
+    <QueryClientProvider client={queryClient}>
+      <ArbiProvider url="https://arbi.mycompany.com">
+        <MyApp />
+      </ArbiProvider>
+    </QueryClientProvider>
+  )
+}
+```
+
+Then use the hooks in any component:
+
+```tsx
+import { useArbi, useWorkspaces, useDocuments } from '@arbidocs/react'
+
+function WorkspaceView() {
+  const arbi = useArbi()
+  const { data: workspaces, isLoading } = useWorkspaces()
+
+  if (isLoading) return <div>Loading...</div>
+
+  return (
+    <ul>
+      {workspaces?.map((ws) => (
+        <li key={ws.external_id}>{ws.name}</li>
+      ))}
+    </ul>
+  )
+}
+```
+
+## Hooks
+
+### `useArbi()`
+
+Returns the `Arbi` instance from context. Use this for imperative operations like login, workspace selection, and direct API calls.
+
+```tsx
+const arbi = useArbi()
+
+await arbi.login('user@example.com', 'password')
+await arbi.selectWorkspace('wrk-123')
+```
+
+### `useWorkspaces(options?)`
+
+Fetches the user's workspaces. Automatically enabled when logged in.
+
+```tsx
+const { data, isLoading, error } = useWorkspaces()
+const { data } = useWorkspaces({ enabled: false }) // manual control
+```
+
+### `useDocuments(workspaceId, options?)`
+
+Fetches documents for a workspace. Automatically disabled when `workspaceId` is undefined.
+
+```tsx
+const { data: docs } = useDocuments('wrk-123')
+const { data: docs } = useDocuments(undefined) // won't fetch
+```
+
+### `useConversations(workspaceId, options?)`
+
+Fetches conversations for a workspace.
+
+```tsx
+const { data: conversations } = useConversations('wrk-123')
+```
+
+### `useTags(workspaceId, options?)`
+
+Fetches tags for a workspace.
+
+```tsx
+const { data: tags } = useTags('wrk-123')
+```
+
+### `useHealth(options?)`
+
+Checks backend health.
+
+```tsx
+const { data } = useHealth({ enabled: true })
+// { status: 'ok' }
+```
+
+### `useAssistantQuery()`
+
+Returns a React Query mutation for streaming AI queries.
+
+```tsx
+const mutation = useAssistantQuery()
+
+const handleAsk = () => {
+  mutation.mutate({
+    question: 'What does section 3 say?',
+    docIds: ['doc-123'],
+    onToken: (token) => setResponse((prev) => prev + token),
+    onComplete: () => console.log('Done'),
+  })
+}
+```
+
+## Query Keys
+
+All hooks use the `arbiQueryKeys` factory for cache management:
+
+```tsx
+import { arbiQueryKeys } from '@arbidocs/react'
+import { useQueryClient } from '@tanstack/react-query'
+
+const queryClient = useQueryClient()
+
+// Invalidate all documents
+queryClient.invalidateQueries({ queryKey: arbiQueryKeys.documents('wrk-123') })
+
+// Available keys:
+arbiQueryKeys.workspaces()            // ['arbi', 'workspaces']
+arbiQueryKeys.documents(workspaceId)  // ['arbi', 'documents', workspaceId]
+arbiQueryKeys.conversations(workspaceId)
+arbiQueryKeys.tags(workspaceId)
+arbiQueryKeys.health()
+```
+
+## Re-exports
+
+The package re-exports key types from `@arbidocs/core/browser` for convenience:
+
+```tsx
+import { Arbi, ArbiError } from '@arbidocs/react'
+import type { ArbiOptions, QueryOptions, SSEStreamResult, SSEStreamCallbacks } from '@arbidocs/react'
+```
+
+## Examples
+
+See the **[React Chat App](../../examples/react-chat/)** for a complete working example with login, workspace selection, document picker, and streaming chat.
+
+## Peer Dependencies
+
+- `react` >= 18

package/dist/index.cjs

@@ -0,0 +1,98 @@
+'use strict';
+
+var react = require('react');
+var browser = require('@arbidocs/core/browser');
+var jsxRuntime = require('react/jsx-runtime');
+var reactQuery = require('@tanstack/react-query');
+
+// src/ArbiProvider.tsx
+var ArbiContext = react.createContext(null);
+function ArbiProvider({ children, ...options }) {
+  const arbi = react.useMemo(
+    () => new browser.Arbi(options),
+    // Stable on url + deploymentDomain — re-creating the instance on every render would be wrong
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [options.url, options.deploymentDomain, options.credentials]
+  );
+  return /* @__PURE__ */ jsxRuntime.jsx(ArbiContext.Provider, { value: arbi, children });
+}
+function useArbi() {
+  const arbi = react.useContext(ArbiContext);
+  if (!arbi) {
+    throw new Error("useArbi() must be used within an <ArbiProvider>");
+  }
+  return arbi;
+}
+var arbiQueryKeys = {
+  workspaces: () => ["arbi", "workspaces"],
+  documents: (workspaceId) => ["arbi", "documents", workspaceId],
+  conversations: (workspaceId) => ["arbi", "conversations", workspaceId],
+  tags: (workspaceId) => ["arbi", "tags", workspaceId],
+  health: () => ["arbi", "health"]
+};
+function useWorkspaces(options) {
+  const arbi = useArbi();
+  return reactQuery.useQuery({
+    queryKey: arbiQueryKeys.workspaces(),
+    queryFn: () => arbi.workspaces.list(),
+    enabled: options?.enabled ?? arbi.isLoggedIn
+  });
+}
+function useDocuments(workspaceId, options) {
+  const arbi = useArbi();
+  return reactQuery.useQuery({
+    queryKey: arbiQueryKeys.documents(workspaceId ?? ""),
+    queryFn: () => arbi.documents.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useConversations(workspaceId, options) {
+  const arbi = useArbi();
+  return reactQuery.useQuery({
+    queryKey: arbiQueryKeys.conversations(workspaceId ?? ""),
+    queryFn: () => arbi.conversations.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useTags(workspaceId, options) {
+  const arbi = useArbi();
+  return reactQuery.useQuery({
+    queryKey: arbiQueryKeys.tags(workspaceId ?? ""),
+    queryFn: () => arbi.tags.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useAssistantQuery() {
+  const arbi = useArbi();
+  return reactQuery.useMutation({
+    mutationFn: ({ question, ...options }) => arbi.assistant.query(question, options)
+  });
+}
+function useHealth(options) {
+  const arbi = useArbi();
+  return reactQuery.useQuery({
+    queryKey: arbiQueryKeys.health(),
+    queryFn: () => arbi.health.check(),
+    enabled: options?.enabled
+  });
+}
+
+Object.defineProperty(exports, "Arbi", {
+  enumerable: true,
+  get: function () { return browser.Arbi; }
+});
+Object.defineProperty(exports, "ArbiError", {
+  enumerable: true,
+  get: function () { return browser.ArbiError; }
+});
+exports.ArbiProvider = ArbiProvider;
+exports.arbiQueryKeys = arbiQueryKeys;
+exports.useArbi = useArbi;
+exports.useAssistantQuery = useAssistantQuery;
+exports.useConversations = useConversations;
+exports.useDocuments = useDocuments;
+exports.useHealth = useHealth;
+exports.useTags = useTags;
+exports.useWorkspaces = useWorkspaces;
+//# sourceMappingURL=index.cjs.map
+//# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ArbiProvider.tsx","../src/useArbi.ts","../src/hooks.ts"],"names":["createContext","useMemo","Arbi","useContext","useQuery","useMutation"],"mappings":";;;;;;;;AAYO,IAAM,WAAA,GAAcA,oBAA2B,IAAI,CAAA;AAMnD,SAAS,YAAA,CAAa,EAAE,QAAA,EAAU,GAAG,SAAQ,EAAsB;AACxE,EAAA,MAAM,IAAA,GAAOC,aAAA;AAAA,IACX,MAAM,IAAIC,YAAA,CAAK,OAAO,CAAA;AAAA;AAAA;AAAA,IAGtB,CAAC,OAAA,CAAQ,GAAA,EAAK,OAAA,CAAQ,gBAAA,EAAkB,QAAQ,WAAW;AAAA,GAC7D;AAEA,EAAA,sCAAQ,WAAA,CAAY,QAAA,EAAZ,EAAqB,KAAA,EAAO,MAAO,QAAA,EAAS,CAAA;AACtD;ACnBO,SAAS,OAAA,GAAgB;AAC9B,EAAA,MAAM,IAAA,GAAOC,iBAAW,WAAW,CAAA;AACnC,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAM,IAAI,MAAM,iDAAiD,CAAA;AAAA,EACnE;AACA,EAAA,OAAO,IAAA;AACT;ACJO,IAAM,aAAA,GAAgB;AAAA,EAC3B,UAAA,EAAY,MAAM,CAAC,MAAA,EAAQ,YAAY,CAAA;AAAA,EACvC,WAAW,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,aAAa,WAAW,CAAA;AAAA,EACrE,eAAe,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,iBAAiB,WAAW,CAAA;AAAA,EAC7E,MAAM,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,QAAQ,WAAW,CAAA;AAAA,EAC3D,MAAA,EAAQ,MAAM,CAAC,MAAA,EAAQ,QAAQ;AACjC;AAKO,SAAS,cAAc,OAAA,EAAiC;AAC7D,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAOC,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,cAAc,UAAA,EAAW;AAAA,IACnC,OAAA,EAAS,MAAM,IAAA,CAAK,UAAA,CAAW,IAAA,EAAK;AAAA,IACpC,OAAA,EAAS,OAAA,EAAS,OAAA,IAAW,IAAA,CAAK;AAAA,GACnC,CAAA;AACH;AAGO,SAAS,YAAA,CAAa,aAAiC,OAAA,EAAiC;AAC7F,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAOA,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,SAAA,CAAU,WAAA,IAAe,EAAE,CAAA;AAAA,IACnD,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,KAAK,WAAW,CAAA;AAAA,IAC9C,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,gBAAA,CAAiB,aAAiC,OAAA,EAAiC;AACjG,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAOA,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,aAAA,CAAc,WAAA,IAAe,EAAE,CAAA;AAAA,IACvD,OAAA,EAAS,MAAM,IAAA,CAAK,aAAA,CAAc,KAAK,WAAW,CAAA;AAAA,IAClD,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,OAAA,CAAQ,aAAiC,OAAA,EAAiC;AACxF,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAOA,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,IAAA,CAAK,WAAA,IAAe,EAAE,CAAA;AAAA,IAC9C,OAAA,EAAS,MAAM,IAAA,CAAK,IAAA,CAAK,KAAK,WAAW,CAAA;AAAA,IACzC,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,iBAAA,GAAoB;AAClC,EAAA,MAAM,OAAO,OAAA,EAAQ;AAErB,EAAA,OAAOC,sBAAA,CAcL;AAAA,IACA,UAAA,EAAY,CAAC,EAAE,QAAA,EAAU,GAAG,OAAA,EAAQ,KAAM,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,QAAA,EAAU,OAAO;AAAA,GACjF,CAAA;AACH;AAGO,SAAS,UAAU,OAAA,EAAiC;AACzD,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAOD,mBAAA,CAAS;AAAA,IACd,QAAA,EAAU,cAAc,MAAA,EAAO;AAAA,IAC/B,OAAA,EAAS,MAAM,IAAA,CAAK,MAAA,CAAO,KAAA,EAAM;AAAA,IACjC,SAAS,OAAA,EAAS;AAAA,GACnB,CAAA;AACH","file":"index.cjs","sourcesContent":["/**\n * ArbiProvider — React context that holds an Arbi instance.\n *\n * Usage:\n *   <ArbiProvider url=\"https://arbi.mycompany.com\">\n *     <App />\n *   </ArbiProvider>\n */\n\nimport { createContext, useMemo, type ReactNode } from 'react'\nimport { Arbi, type ArbiOptions } from '@arbidocs/core/browser'\n\nexport const ArbiContext = createContext<Arbi | null>(null)\n\nexport interface ArbiProviderProps extends ArbiOptions {\n  children: ReactNode\n}\n\nexport function ArbiProvider({ children, ...options }: ArbiProviderProps) {\n  const arbi = useMemo(\n    () => new Arbi(options),\n    // Stable on url + deploymentDomain — re-creating the instance on every render would be wrong\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [options.url, options.deploymentDomain, options.credentials]\n  )\n\n  return <ArbiContext.Provider value={arbi}>{children}</ArbiContext.Provider>\n}\n","/**\n * Hook to access the Arbi instance from the nearest ArbiProvider.\n */\n\nimport { useContext } from 'react'\nimport type { Arbi } from '@arbidocs/core/browser'\nimport { ArbiContext } from './ArbiProvider'\n\nexport function useArbi(): Arbi {\n  const arbi = useContext(ArbiContext)\n  if (!arbi) {\n    throw new Error('useArbi() must be used within an <ArbiProvider>')\n  }\n  return arbi\n}\n","/**\n * React Query hooks that use the Arbi instance from context.\n */\n\nimport { useQuery, useMutation } from '@tanstack/react-query'\nimport type { SSEStreamResult } from '@arbidocs/core/browser'\nimport { useArbi } from './useArbi'\n\n// ── Query keys ──────────────────────────────────────────────────────────────\n\nexport const arbiQueryKeys = {\n  workspaces: () => ['arbi', 'workspaces'] as const,\n  documents: (workspaceId: string) => ['arbi', 'documents', workspaceId] as const,\n  conversations: (workspaceId: string) => ['arbi', 'conversations', workspaceId] as const,\n  tags: (workspaceId: string) => ['arbi', 'tags', workspaceId] as const,\n  health: () => ['arbi', 'health'] as const,\n}\n\n// ── Hooks ───────────────────────────────────────────────────────────────────\n\n/** List the logged-in user's workspaces. */\nexport function useWorkspaces(options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.workspaces(),\n    queryFn: () => arbi.workspaces.list(),\n    enabled: options?.enabled ?? arbi.isLoggedIn,\n  })\n}\n\n/** List documents in a workspace. */\nexport function useDocuments(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.documents(workspaceId ?? ''),\n    queryFn: () => arbi.documents.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** List conversations in a workspace. */\nexport function useConversations(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.conversations(workspaceId ?? ''),\n    queryFn: () => arbi.conversations.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** List tags in a workspace. */\nexport function useTags(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.tags(workspaceId ?? ''),\n    queryFn: () => arbi.tags.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** Send a streaming query to the RAG assistant. Returns a mutation. */\nexport function useAssistantQuery() {\n  const arbi = useArbi()\n\n  return useMutation<\n    SSEStreamResult,\n    Error,\n    {\n      question: string\n      docIds: string[]\n      parentMessageExtId?: string | null\n      configExtId?: string\n      onToken?: (content: string) => void\n      onStreamStart?: (data: { assistant_message_ext_id?: string }) => void\n      onAgentStep?: (data: { focus?: string; status?: string }) => void\n      onError?: (message: string) => void\n      onComplete?: () => void\n    }\n  >({\n    mutationFn: ({ question, ...options }) => arbi.assistant.query(question, options),\n  })\n}\n\n/** Check backend health. */\nexport function useHealth(options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.health(),\n    queryFn: () => arbi.health.check(),\n    enabled: options?.enabled,\n  })\n}\n"]}

package/dist/index.d.cts

@@ -0,0 +1,132 @@
+import * as react_jsx_runtime from 'react/jsx-runtime';
+import { ReactNode } from 'react';
+import { ArbiOptions, Arbi, SSEStreamResult } from '@arbidocs/core/browser';
+export { Arbi, ArbiError, ArbiOptions, QueryOptions, SSEStreamCallbacks, SSEStreamResult } from '@arbidocs/core/browser';
+import * as _tanstack_react_query from '@tanstack/react-query';
+import * as _arbidocs_sdk from '@arbidocs/sdk';
+
+interface ArbiProviderProps extends ArbiOptions {
+    children: ReactNode;
+}
+declare function ArbiProvider({ children, ...options }: ArbiProviderProps): react_jsx_runtime.JSX.Element;
+
+/**
+ * Hook to access the Arbi instance from the nearest ArbiProvider.
+ */
+
+declare function useArbi(): Arbi;
+
+declare const arbiQueryKeys: {
+    workspaces: () => readonly ["arbi", "workspaces"];
+    documents: (workspaceId: string) => readonly ["arbi", "documents", string];
+    conversations: (workspaceId: string) => readonly ["arbi", "conversations", string];
+    tags: (workspaceId: string) => readonly ["arbi", "tags", string];
+    health: () => readonly ["arbi", "health"];
+};
+/** List the logged-in user's workspaces. */
+declare function useWorkspaces(options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    name: string;
+    description: string | null;
+    is_public: boolean;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    wrapped_key?: string | null;
+    shared_conversation_count: number;
+    private_conversation_count: number;
+    shared_document_count: number;
+    private_document_count: number;
+    user_files_mb: number;
+    users: _arbidocs_sdk.components["schemas"]["WorkspaceUserResponse"][];
+}[], Error>;
+/** List documents in a workspace. */
+declare function useDocuments(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    workspace_ext_id: string;
+    file_name?: string | null;
+    status?: string | null;
+    n_pages?: number | null;
+    n_chunks?: number | null;
+    tokens?: number | null;
+    file_type?: string | null;
+    file_size?: number | null;
+    storage_type?: string | null;
+    storage_uri?: string | null;
+    content_hash?: string | null;
+    shared?: boolean | null;
+    re_ocred?: boolean | null;
+    config_ext_id?: string | null;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    wp_type?: string | null;
+    doctags: _arbidocs_sdk.components["schemas"]["DocTagResponse"][];
+    doc_metadata?: _arbidocs_sdk.components["schemas"]["DocMetadata"] | null;
+}[], Error>;
+/** List conversations in a workspace. */
+declare function useConversations(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    title: string | null;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    is_shared: boolean;
+    message_count: number;
+}[], Error>;
+/** List tags in a workspace. */
+declare function useTags(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    workspace_ext_id: string;
+    name: string;
+    instruction?: string | null;
+    tag_type: _arbidocs_sdk.components["schemas"]["TagFormat"];
+    shared: boolean;
+    parent_ext_id?: string | null;
+    doctag_count: number;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+}[], Error>;
+/** Send a streaming query to the RAG assistant. Returns a mutation. */
+declare function useAssistantQuery(): _tanstack_react_query.UseMutationResult<SSEStreamResult, Error, {
+    question: string;
+    docIds: string[];
+    parentMessageExtId?: string | null;
+    configExtId?: string;
+    onToken?: (content: string) => void;
+    onStreamStart?: (data: {
+        assistant_message_ext_id?: string;
+    }) => void;
+    onAgentStep?: (data: {
+        focus?: string;
+        status?: string;
+    }) => void;
+    onError?: (message: string) => void;
+    onComplete?: () => void;
+}, unknown>;
+/** Check backend health. */
+declare function useHealth(options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    status: string;
+    backend_git_hash?: string | null;
+    frontend_docker_version?: string | null;
+    services: _arbidocs_sdk.components["schemas"]["ServiceStatus"][];
+    models_health?: _arbidocs_sdk.components["schemas"]["ModelsHealthStatus"] | null;
+    available_models: string[];
+}, Error>;
+
+export { ArbiProvider, type ArbiProviderProps, arbiQueryKeys, useArbi, useAssistantQuery, useConversations, useDocuments, useHealth, useTags, useWorkspaces };

package/dist/index.d.ts

@@ -0,0 +1,132 @@
+import * as react_jsx_runtime from 'react/jsx-runtime';
+import { ReactNode } from 'react';
+import { ArbiOptions, Arbi, SSEStreamResult } from '@arbidocs/core/browser';
+export { Arbi, ArbiError, ArbiOptions, QueryOptions, SSEStreamCallbacks, SSEStreamResult } from '@arbidocs/core/browser';
+import * as _tanstack_react_query from '@tanstack/react-query';
+import * as _arbidocs_sdk from '@arbidocs/sdk';
+
+interface ArbiProviderProps extends ArbiOptions {
+    children: ReactNode;
+}
+declare function ArbiProvider({ children, ...options }: ArbiProviderProps): react_jsx_runtime.JSX.Element;
+
+/**
+ * Hook to access the Arbi instance from the nearest ArbiProvider.
+ */
+
+declare function useArbi(): Arbi;
+
+declare const arbiQueryKeys: {
+    workspaces: () => readonly ["arbi", "workspaces"];
+    documents: (workspaceId: string) => readonly ["arbi", "documents", string];
+    conversations: (workspaceId: string) => readonly ["arbi", "conversations", string];
+    tags: (workspaceId: string) => readonly ["arbi", "tags", string];
+    health: () => readonly ["arbi", "health"];
+};
+/** List the logged-in user's workspaces. */
+declare function useWorkspaces(options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    name: string;
+    description: string | null;
+    is_public: boolean;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    wrapped_key?: string | null;
+    shared_conversation_count: number;
+    private_conversation_count: number;
+    shared_document_count: number;
+    private_document_count: number;
+    user_files_mb: number;
+    users: _arbidocs_sdk.components["schemas"]["WorkspaceUserResponse"][];
+}[], Error>;
+/** List documents in a workspace. */
+declare function useDocuments(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    workspace_ext_id: string;
+    file_name?: string | null;
+    status?: string | null;
+    n_pages?: number | null;
+    n_chunks?: number | null;
+    tokens?: number | null;
+    file_type?: string | null;
+    file_size?: number | null;
+    storage_type?: string | null;
+    storage_uri?: string | null;
+    content_hash?: string | null;
+    shared?: boolean | null;
+    re_ocred?: boolean | null;
+    config_ext_id?: string | null;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    wp_type?: string | null;
+    doctags: _arbidocs_sdk.components["schemas"]["DocTagResponse"][];
+    doc_metadata?: _arbidocs_sdk.components["schemas"]["DocMetadata"] | null;
+}[], Error>;
+/** List conversations in a workspace. */
+declare function useConversations(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    title: string | null;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+    is_shared: boolean;
+    message_count: number;
+}[], Error>;
+/** List tags in a workspace. */
+declare function useTags(workspaceId: string | undefined, options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    external_id: string;
+    workspace_ext_id: string;
+    name: string;
+    instruction?: string | null;
+    tag_type: _arbidocs_sdk.components["schemas"]["TagFormat"];
+    shared: boolean;
+    parent_ext_id?: string | null;
+    doctag_count: number;
+    created_by_ext_id: string;
+    updated_by_ext_id?: string | null;
+    created_at: string;
+    updated_at: string;
+}[], Error>;
+/** Send a streaming query to the RAG assistant. Returns a mutation. */
+declare function useAssistantQuery(): _tanstack_react_query.UseMutationResult<SSEStreamResult, Error, {
+    question: string;
+    docIds: string[];
+    parentMessageExtId?: string | null;
+    configExtId?: string;
+    onToken?: (content: string) => void;
+    onStreamStart?: (data: {
+        assistant_message_ext_id?: string;
+    }) => void;
+    onAgentStep?: (data: {
+        focus?: string;
+        status?: string;
+    }) => void;
+    onError?: (message: string) => void;
+    onComplete?: () => void;
+}, unknown>;
+/** Check backend health. */
+declare function useHealth(options?: {
+    enabled?: boolean;
+}): _tanstack_react_query.UseQueryResult<{
+    status: string;
+    backend_git_hash?: string | null;
+    frontend_docker_version?: string | null;
+    services: _arbidocs_sdk.components["schemas"]["ServiceStatus"][];
+    models_health?: _arbidocs_sdk.components["schemas"]["ModelsHealthStatus"] | null;
+    available_models: string[];
+}, Error>;
+
+export { ArbiProvider, type ArbiProviderProps, arbiQueryKeys, useArbi, useAssistantQuery, useConversations, useDocuments, useHealth, useTags, useWorkspaces };

package/dist/index.js

@@ -0,0 +1,81 @@
+import { createContext, useMemo, useContext } from 'react';
+import { Arbi } from '@arbidocs/core/browser';
+export { Arbi, ArbiError } from '@arbidocs/core/browser';
+import { jsx } from 'react/jsx-runtime';
+import { useQuery, useMutation } from '@tanstack/react-query';
+
+// src/ArbiProvider.tsx
+var ArbiContext = createContext(null);
+function ArbiProvider({ children, ...options }) {
+  const arbi = useMemo(
+    () => new Arbi(options),
+    // Stable on url + deploymentDomain — re-creating the instance on every render would be wrong
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [options.url, options.deploymentDomain, options.credentials]
+  );
+  return /* @__PURE__ */ jsx(ArbiContext.Provider, { value: arbi, children });
+}
+function useArbi() {
+  const arbi = useContext(ArbiContext);
+  if (!arbi) {
+    throw new Error("useArbi() must be used within an <ArbiProvider>");
+  }
+  return arbi;
+}
+var arbiQueryKeys = {
+  workspaces: () => ["arbi", "workspaces"],
+  documents: (workspaceId) => ["arbi", "documents", workspaceId],
+  conversations: (workspaceId) => ["arbi", "conversations", workspaceId],
+  tags: (workspaceId) => ["arbi", "tags", workspaceId],
+  health: () => ["arbi", "health"]
+};
+function useWorkspaces(options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.workspaces(),
+    queryFn: () => arbi.workspaces.list(),
+    enabled: options?.enabled ?? arbi.isLoggedIn
+  });
+}
+function useDocuments(workspaceId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.documents(workspaceId ?? ""),
+    queryFn: () => arbi.documents.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useConversations(workspaceId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.conversations(workspaceId ?? ""),
+    queryFn: () => arbi.conversations.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useTags(workspaceId, options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.tags(workspaceId ?? ""),
+    queryFn: () => arbi.tags.list(workspaceId),
+    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace
+  });
+}
+function useAssistantQuery() {
+  const arbi = useArbi();
+  return useMutation({
+    mutationFn: ({ question, ...options }) => arbi.assistant.query(question, options)
+  });
+}
+function useHealth(options) {
+  const arbi = useArbi();
+  return useQuery({
+    queryKey: arbiQueryKeys.health(),
+    queryFn: () => arbi.health.check(),
+    enabled: options?.enabled
+  });
+}
+
+export { ArbiProvider, arbiQueryKeys, useArbi, useAssistantQuery, useConversations, useDocuments, useHealth, useTags, useWorkspaces };
+//# sourceMappingURL=index.js.map
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/ArbiProvider.tsx","../src/useArbi.ts","../src/hooks.ts"],"names":[],"mappings":";;;;;;;AAYO,IAAM,WAAA,GAAc,cAA2B,IAAI,CAAA;AAMnD,SAAS,YAAA,CAAa,EAAE,QAAA,EAAU,GAAG,SAAQ,EAAsB;AACxE,EAAA,MAAM,IAAA,GAAO,OAAA;AAAA,IACX,MAAM,IAAI,IAAA,CAAK,OAAO,CAAA;AAAA;AAAA;AAAA,IAGtB,CAAC,OAAA,CAAQ,GAAA,EAAK,OAAA,CAAQ,gBAAA,EAAkB,QAAQ,WAAW;AAAA,GAC7D;AAEA,EAAA,2BAAQ,WAAA,CAAY,QAAA,EAAZ,EAAqB,KAAA,EAAO,MAAO,QAAA,EAAS,CAAA;AACtD;ACnBO,SAAS,OAAA,GAAgB;AAC9B,EAAA,MAAM,IAAA,GAAO,WAAW,WAAW,CAAA;AACnC,EAAA,IAAI,CAAC,IAAA,EAAM;AACT,IAAA,MAAM,IAAI,MAAM,iDAAiD,CAAA;AAAA,EACnE;AACA,EAAA,OAAO,IAAA;AACT;ACJO,IAAM,aAAA,GAAgB;AAAA,EAC3B,UAAA,EAAY,MAAM,CAAC,MAAA,EAAQ,YAAY,CAAA;AAAA,EACvC,WAAW,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,aAAa,WAAW,CAAA;AAAA,EACrE,eAAe,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,iBAAiB,WAAW,CAAA;AAAA,EAC7E,MAAM,CAAC,WAAA,KAAwB,CAAC,MAAA,EAAQ,QAAQ,WAAW,CAAA;AAAA,EAC3D,MAAA,EAAQ,MAAM,CAAC,MAAA,EAAQ,QAAQ;AACjC;AAKO,SAAS,cAAc,OAAA,EAAiC;AAC7D,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,cAAc,UAAA,EAAW;AAAA,IACnC,OAAA,EAAS,MAAM,IAAA,CAAK,UAAA,CAAW,IAAA,EAAK;AAAA,IACpC,OAAA,EAAS,OAAA,EAAS,OAAA,IAAW,IAAA,CAAK;AAAA,GACnC,CAAA;AACH;AAGO,SAAS,YAAA,CAAa,aAAiC,OAAA,EAAiC;AAC7F,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,SAAA,CAAU,WAAA,IAAe,EAAE,CAAA;AAAA,IACnD,OAAA,EAAS,MAAM,IAAA,CAAK,SAAA,CAAU,KAAK,WAAW,CAAA;AAAA,IAC9C,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,gBAAA,CAAiB,aAAiC,OAAA,EAAiC;AACjG,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,aAAA,CAAc,WAAA,IAAe,EAAE,CAAA;AAAA,IACvD,OAAA,EAAS,MAAM,IAAA,CAAK,aAAA,CAAc,KAAK,WAAW,CAAA;AAAA,IAClD,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,OAAA,CAAQ,aAAiC,OAAA,EAAiC;AACxF,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,aAAA,CAAc,IAAA,CAAK,WAAA,IAAe,EAAE,CAAA;AAAA,IAC9C,OAAA,EAAS,MAAM,IAAA,CAAK,IAAA,CAAK,KAAK,WAAW,CAAA;AAAA,IACzC,UAAU,OAAA,EAAS,OAAA,IAAW,SAAS,CAAC,CAAC,eAAe,IAAA,CAAK;AAAA,GAC9D,CAAA;AACH;AAGO,SAAS,iBAAA,GAAoB;AAClC,EAAA,MAAM,OAAO,OAAA,EAAQ;AAErB,EAAA,OAAO,WAAA,CAcL;AAAA,IACA,UAAA,EAAY,CAAC,EAAE,QAAA,EAAU,GAAG,OAAA,EAAQ,KAAM,IAAA,CAAK,SAAA,CAAU,KAAA,CAAM,QAAA,EAAU,OAAO;AAAA,GACjF,CAAA;AACH;AAGO,SAAS,UAAU,OAAA,EAAiC;AACzD,EAAA,MAAM,OAAO,OAAA,EAAQ;AACrB,EAAA,OAAO,QAAA,CAAS;AAAA,IACd,QAAA,EAAU,cAAc,MAAA,EAAO;AAAA,IAC/B,OAAA,EAAS,MAAM,IAAA,CAAK,MAAA,CAAO,KAAA,EAAM;AAAA,IACjC,SAAS,OAAA,EAAS;AAAA,GACnB,CAAA;AACH","file":"index.js","sourcesContent":["/**\n * ArbiProvider — React context that holds an Arbi instance.\n *\n * Usage:\n *   <ArbiProvider url=\"https://arbi.mycompany.com\">\n *     <App />\n *   </ArbiProvider>\n */\n\nimport { createContext, useMemo, type ReactNode } from 'react'\nimport { Arbi, type ArbiOptions } from '@arbidocs/core/browser'\n\nexport const ArbiContext = createContext<Arbi | null>(null)\n\nexport interface ArbiProviderProps extends ArbiOptions {\n  children: ReactNode\n}\n\nexport function ArbiProvider({ children, ...options }: ArbiProviderProps) {\n  const arbi = useMemo(\n    () => new Arbi(options),\n    // Stable on url + deploymentDomain — re-creating the instance on every render would be wrong\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [options.url, options.deploymentDomain, options.credentials]\n  )\n\n  return <ArbiContext.Provider value={arbi}>{children}</ArbiContext.Provider>\n}\n","/**\n * Hook to access the Arbi instance from the nearest ArbiProvider.\n */\n\nimport { useContext } from 'react'\nimport type { Arbi } from '@arbidocs/core/browser'\nimport { ArbiContext } from './ArbiProvider'\n\nexport function useArbi(): Arbi {\n  const arbi = useContext(ArbiContext)\n  if (!arbi) {\n    throw new Error('useArbi() must be used within an <ArbiProvider>')\n  }\n  return arbi\n}\n","/**\n * React Query hooks that use the Arbi instance from context.\n */\n\nimport { useQuery, useMutation } from '@tanstack/react-query'\nimport type { SSEStreamResult } from '@arbidocs/core/browser'\nimport { useArbi } from './useArbi'\n\n// ── Query keys ──────────────────────────────────────────────────────────────\n\nexport const arbiQueryKeys = {\n  workspaces: () => ['arbi', 'workspaces'] as const,\n  documents: (workspaceId: string) => ['arbi', 'documents', workspaceId] as const,\n  conversations: (workspaceId: string) => ['arbi', 'conversations', workspaceId] as const,\n  tags: (workspaceId: string) => ['arbi', 'tags', workspaceId] as const,\n  health: () => ['arbi', 'health'] as const,\n}\n\n// ── Hooks ───────────────────────────────────────────────────────────────────\n\n/** List the logged-in user's workspaces. */\nexport function useWorkspaces(options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.workspaces(),\n    queryFn: () => arbi.workspaces.list(),\n    enabled: options?.enabled ?? arbi.isLoggedIn,\n  })\n}\n\n/** List documents in a workspace. */\nexport function useDocuments(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.documents(workspaceId ?? ''),\n    queryFn: () => arbi.documents.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** List conversations in a workspace. */\nexport function useConversations(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.conversations(workspaceId ?? ''),\n    queryFn: () => arbi.conversations.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** List tags in a workspace. */\nexport function useTags(workspaceId: string | undefined, options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.tags(workspaceId ?? ''),\n    queryFn: () => arbi.tags.list(workspaceId),\n    enabled: (options?.enabled ?? true) && !!workspaceId && arbi.hasWorkspace,\n  })\n}\n\n/** Send a streaming query to the RAG assistant. Returns a mutation. */\nexport function useAssistantQuery() {\n  const arbi = useArbi()\n\n  return useMutation<\n    SSEStreamResult,\n    Error,\n    {\n      question: string\n      docIds: string[]\n      parentMessageExtId?: string | null\n      configExtId?: string\n      onToken?: (content: string) => void\n      onStreamStart?: (data: { assistant_message_ext_id?: string }) => void\n      onAgentStep?: (data: { focus?: string; status?: string }) => void\n      onError?: (message: string) => void\n      onComplete?: () => void\n    }\n  >({\n    mutationFn: ({ question, ...options }) => arbi.assistant.query(question, options),\n  })\n}\n\n/** Check backend health. */\nexport function useHealth(options?: { enabled?: boolean }) {\n  const arbi = useArbi()\n  return useQuery({\n    queryKey: arbiQueryKeys.health(),\n    queryFn: () => arbi.health.check(),\n    enabled: options?.enabled,\n  })\n}\n"]}

package/package.json

@@ -0,0 +1,60 @@
+{
+  "name": "@arbidocs/react",
+  "version": "0.1.0",
+  "description": "React hooks and provider for the ARBI SDK",
+  "type": "module",
+  "main": "dist/index.cjs",
+  "module": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "require": "./dist/index.cjs"
+    }
+  },
+  "files": [
+    "dist"
+  ],
+  "scripts": {
+    "build": "tsup",
+    "test": "vitest run",
+    "typecheck": "tsc --noEmit"
+  },
+  "dependencies": {
+    "@arbidocs/core": "^0.2.0",
+    "@arbidocs/sdk": "^0.1.0",
+    "@tanstack/react-query": ">=5"
+  },
+  "peerDependencies": {
+    "react": ">=18"
+  },
+  "devDependencies": {
+    "@tanstack/react-query": "^5.90.5",
+    "@testing-library/react": "^16.3.0",
+    "@types/react": "^19.2.2",
+    "@types/react-dom": "^19.2.2",
+    "jsdom": "^27.0.1",
+    "react": "^19.2.0",
+    "react-dom": "^19.2.0",
+    "tsup": "^8.5.1",
+    "typescript": "^5.9.3",
+    "vitest": "^4.0.18"
+  },
+  "license": "MIT",
+  "publishConfig": {
+    "access": "public"
+  },
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/arbitrationcity/ARBI-frontend.git",
+    "directory": "packages/arbi-react"
+  },
+  "keywords": [
+    "arbi",
+    "react",
+    "hooks",
+    "sdk",
+    "typescript"
+  ]
+}