altair-graphql-plugin-ai 7.3.0 → 7.3.2

package/src/dev.tsx

@@ -1,37 +0,0 @@
-import React from 'react';
-import ReactDOM from 'react-dom/client';
-import Chat, { ChatProps } from './components/Chat/Chat';
-
-const chatProps: ChatProps = {
-  loggedIn: true,
-  loading: false,
-  sendMessageIsPending: true,
-  availableCredits: 0,
-  activeSession: {
-    id: '1',
-    isActive: true,
-    title: 'title',
-    userId: 'user-1',
-  },
-  messages: [
-    {
-      id: '1',
-      message:
-        'Can you generate a GraphQL query to fetch all users with their names and email addresses?',
-      role: 'USER',
-      sessionId: '1',
-    },
-    {
-      id: '2',
-      message:
-        'Certainly! Here is a GraphQL query to fetch all users along with their names and email addresses:\n\n```graphql\nquery {\n  users {\n    id\n    name\n    email\n  }\n}\n```\n\nMake sure your GraphQL server schema includes a `users` query that returns a list of user objects, each containing `id`, `name`, and `email` fields.',
-      role: 'ASSISTANT',
-      sessionId: '1',
-    },
-  ],
-};
-ReactDOM.createRoot(document.getElementById('root')!).render(
-  <React.StrictMode>
-    <Chat {...chatProps} />
-  </React.StrictMode>
-);

package/src/panel.tsx

@@ -1,194 +0,0 @@
-import { createRoot } from 'react-dom/client';
-import { PluginV3Context } from 'altair-graphql-core/build/plugin/v3/context';
-import { AltairV3Panel } from 'altair-graphql-core/build/plugin/v3/panel';
-import Chat, { ChatProps } from './components/Chat/Chat';
-import {
-  QueryClient,
-  QueryClientProvider,
-  useMutation,
-  useQuery,
-} from '@tanstack/react-query';
-import toast, { Toaster } from 'react-hot-toast';
-import { IMessage, ISendMessageDto } from 'altair-graphql-core/build/ai/types';
-
-const queryClient = new QueryClient({
-  defaultOptions: {
-    queries: {
-      refetchOnWindowFocus: false,
-      refetchOnReconnect: false,
-      retry: false,
-    },
-  },
-});
-
-interface PanelProps {
-  context: PluginV3Context;
-}
-const Panel = ({ context }: PanelProps) => {
-  const { data: userInfo, isLoading: userInfoIsLoading } = useQuery({
-    queryKey: ['userInfo'],
-    queryFn: () => context.getUserInfo(),
-  });
-
-  const { data: activeSession, isLoading: activeSessionIsLoading } = useQuery({
-    queryKey: ['activeSession'],
-    queryFn: () => context.getActiveAiSession(),
-    enabled: !!userInfo?.loggedIn,
-  });
-
-  const { data: availableCredits } = useQuery({
-    queryKey: ['availableCredits'],
-    queryFn: () => context.getAvailableCredits(),
-    enabled: !!userInfo?.loggedIn,
-  });
-
-  const { data: messages, isLoading: messagesIsLoading } = useQuery({
-    queryKey: ['sessionMessages', activeSession?.id],
-    queryFn: () =>
-      activeSession
-        ? context.getAiSessionMessages(activeSession.id)
-        : Promise.resolve([]),
-    enabled: !!activeSession?.id,
-  });
-
-  const { mutate: createAiSession, isPending: createSessionIsPending } = useMutation(
-    {
-      mutationKey: ['createAiSession'],
-      mutationFn: () => context.createAiSession(),
-      onSettled: async () => {
-        await queryClient.invalidateQueries({ queryKey: ['activeSession'] });
-        await queryClient.invalidateQueries({ queryKey: ['sessionMessages'] });
-      },
-    }
-  );
-
-  const {
-    mutate: sendMessage,
-    isPending: sendMessageIsPending,
-    error,
-  } = useMutation({
-    mutationKey: ['sendMessage'],
-    mutationFn: async (message: string) => {
-      if (!activeSession) {
-        throw new Error('No active session');
-      }
-
-      const windowState = await context.getCurrentWindowState();
-      let graphqlQuery = '';
-      let graphqlVariables = '';
-      let sdl = '';
-      if (windowState) {
-        graphqlQuery = windowState.query;
-        graphqlVariables = windowState.variables;
-        sdl = windowState.sdl;
-      }
-
-      // build message input
-      const input: ISendMessageDto = {
-        message,
-        graphqlQuery,
-        graphqlVariables,
-        sdl,
-      };
-      return context.sendMessageToAiSession(activeSession.id, input);
-    },
-    onMutate: async (message) => {
-      // Cancel any outgoing refetches
-      // (so they don't overwrite our optimistic update)
-      await queryClient.cancelQueries({ queryKey: ['sessionMessages'] });
-
-      // Snapshot the previous value
-      const previousMessages = queryClient.getQueryData<IMessage[]>([
-        'sessionMessages',
-      ]);
-
-      const sessionId = activeSession?.id ?? '';
-      const fakeMessage: IMessage = {
-        id: Math.random().toString(),
-        message: message,
-        role: 'USER',
-        sessionId,
-      };
-      // Optimistically update to the new value
-      queryClient.setQueryData<IMessage[]>(['sessionMessages', sessionId], (old) => [
-        ...(old ?? []),
-        fakeMessage,
-      ]);
-
-      // Return a context object with the snapshotted value
-      return { previousMessages };
-    },
-    // If the mutation fails,
-    // use the context returned from onMutate to roll back
-    onError: (_err, _message, context) => {
-      if (context?.previousMessages) {
-        queryClient.setQueryData(['todos'], context.previousMessages);
-      }
-    },
-    // Always refetch after error or success:
-    onSettled: async () => {
-      // wait for the refetching to complete before marking as not pending
-      await Promise.all([
-        queryClient.invalidateQueries({ queryKey: ['sessionMessages'] }),
-        queryClient.invalidateQueries({ queryKey: ['availableCredits'] }),
-      ]);
-    },
-  });
-
-  if (error) {
-    console.error(error);
-  }
-
-  const chatProps: ChatProps = {
-    loggedIn: !!userInfo?.loggedIn,
-    loading:
-      userInfoIsLoading ||
-      activeSessionIsLoading ||
-      messagesIsLoading ||
-      createSessionIsPending,
-    userInfo,
-    activeSession,
-    messages,
-    sendMessageIsPending,
-    availableCredits: availableCredits?.total ?? 0,
-    isPro: userInfo?.plan?.id === 'pro',
-    onStartNewSession() {
-      createAiSession();
-    },
-    onSendMessage(message) {
-      sendMessage(message);
-    },
-    async onUseQuery(query) {
-      const windowState = await context.getCurrentWindowState();
-      if (!windowState) {
-        return;
-      }
-      await context.setQuery(windowState.windowId, query);
-    },
-    async onRateMessage(messageId, rating) {
-      if (!activeSession) {
-        return;
-      }
-      await context.rateAiSessionMessage(activeSession.id, messageId, rating);
-      toast('Thank you for your feedback!');
-    },
-  };
-
-  return (
-    <>
-      <Chat {...chatProps} />
-      <Toaster />
-    </>
-  );
-};
-
-export class AiPluginPanel extends AltairV3Panel {
-  create(ctx: PluginV3Context, container: HTMLElement): void {
-    const root = createRoot(container);
-    root.render(
-      <QueryClientProvider client={queryClient}>
-        <Panel context={ctx} />
-      </QueryClientProvider>
-    );
-  }
-}

package/src/plugin.tsx

@@ -1,31 +0,0 @@
-import { PluginV3 } from 'altair-graphql-core/build/plugin/v3/plugin';
-import { PluginV3Context } from 'altair-graphql-core/build/plugin/v3/context';
-import { AiPluginPanel } from './panel';
-import { AltairPanelLocation } from 'altair-graphql-core/build/plugin/panel';
-
-class AiPlugin extends PluginV3 {
-  constructor() {
-    super({
-      panels: {
-        ai: new AiPluginPanel(),
-      },
-    });
-  }
-
-  async initialize(ctx: PluginV3Context) {
-    console.log('PLUGIN initialize', ctx);
-    console.log('PLUGIN isElectron', await ctx.getCurrentWindowState());
-    ctx.on('query.change', (x) => {
-      console.log('PLUGIN query.change', x);
-    });
-
-    await ctx.createPanel('ai', {
-      location: AltairPanelLocation.SIDEBAR,
-      width: 400,
-    });
-  }
-
-  async destroy() {}
-}
-console.log('AI plugin loaded!');
-new AiPlugin();

package/src/utils.ts

@@ -1,8 +0,0 @@
-export const randomSampling = ([...arr], n = 1) => {
-  let m = arr.length;
-  while (m) {
-    const i = Math.floor(Math.random() * m--);
-    [arr[m], arr[i]] = [arr[i], arr[m]];
-  }
-  return arr.slice(0, n);
-};

package/src/vite-env.d.ts

@@ -1 +0,0 @@
-/// <reference types="vite/client" />

package/tsconfig.app.json

@@ -1,27 +0,0 @@
-{
-  "compilerOptions": {
-    "composite": true,
-    "tsBuildInfoFile": "./node_modules/.tmp/tsconfig.app.tsbuildinfo",
-    "target": "ES2020",
-    "useDefineForClassFields": true,
-    "lib": ["ES2020", "DOM", "DOM.Iterable"],
-    "module": "ESNext",
-    "skipLibCheck": true,
-
-    /* Bundler mode */
-    "moduleResolution": "bundler",
-    "allowImportingTsExtensions": true,
-    "resolveJsonModule": true,
-    "isolatedModules": true,
-    "moduleDetection": "force",
-    "noEmit": true,
-    "jsx": "react-jsx",
-
-    /* Linting */
-    "strict": true,
-    "noUnusedLocals": true,
-    "noUnusedParameters": true,
-    "noFallthroughCasesInSwitch": true
-  },
-  "include": ["src"]
-}

package/tsconfig.json

@@ -1,11 +0,0 @@
-{
-  "files": [],
-  "references": [
-    {
-      "path": "./tsconfig.app.json"
-    },
-    {
-      "path": "./tsconfig.node.json"
-    }
-  ]
-}

package/tsconfig.node.json

@@ -1,13 +0,0 @@
-{
-  "compilerOptions": {
-    "composite": true,
-    "tsBuildInfoFile": "./node_modules/.tmp/tsconfig.node.tsbuildinfo",
-    "skipLibCheck": true,
-    "module": "ESNext",
-    "moduleResolution": "bundler",
-    "allowSyntheticDefaultImports": true,
-    "strict": true,
-    "noEmit": true
-  },
-  "include": ["vite.config.ts"]
-}

package/vite.config.ts

@@ -1,22 +0,0 @@
-import { defineConfig } from 'vite';
-import react from '@vitejs/plugin-react-swc';
-import { fileURLToPath } from 'url';
-
-// https://vitejs.dev/config/
-export default defineConfig({
-  plugins: [react()],
-  build: {
-    rollupOptions: {
-      input: {
-        plugin: fileURLToPath(new URL('./src/plugin.tsx', import.meta.url)),
-      },
-      output: {
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        manualChunks: false as any,
-        inlineDynamicImports: true,
-        entryFileNames: '[name].js', // currently does not work for the legacy bundle
-        assetFileNames: '[name].[ext]', // currently does not work for images
-      },
-    },
-  },
-});