@hasura/promptql 2.0.0-alpha.1 → 2.0.0-alpha.3

package/src/graphql/README.md

@@ -1,70 +0,0 @@
-# GraphQL Operations
-
-This directory contains GraphQL operations (queries, mutations, and fragments)
-that are used by the chat-handler service.
-
-## Directory Structure
-
-```
-graphql/
-├── queries/       # GraphQL queries
-├── mutations/     # GraphQL mutations
-└── fragments/     # Reusable GraphQL fragments
-```
-
-## How to Use
-
-### 1. Define Operations
-
-Create `.gql` files in the appropriate directory:
-
-**queries/getUser.gql**
-
-```graphql
-query GetUser($userId: uuid!) {
-  users_by_pk(id: $userId) {
-    id
-    email
-    name
-  }
-}
-```
-
-### 2. Generate Types
-
-Run the code generator to create TypeScript types:
-
-```bash
-bun run codegen
-```
-
-This will generate typed operations in `src/generated/graphql.ts`.
-
-### 3. Use in Code
-
-Import and use the generated SDK:
-
-```typescript
-import { createHGEClient } from "../hge-client";
-
-const sdk = createHGEClient();
-
-// Fully typed query with autocomplete and type checking
-const result = await sdk.GetUser({ userId: "123" });
-console.log(result.users_by_pk?.name);
-```
-
-## Benefits
-
-- **Type Safety**: All operations are fully typed
-- **Autocomplete**: IDE provides autocomplete for all fields
-- **Validation**: Operations are validated against the schema at build time
-- **Refactoring**: Renaming fields in the schema automatically updates all
-  usages
-
-## Environment Variables
-
-Make sure to set these environment variables:
-
-- `PLAYGROUND_GRAPHQL_URI`: The GraphQL API endpoint URL
-- `PLAYGROUND_GRAPHQL_ADMIN_SECRET`: The admin secret for authentication

package/src/graphql/mutations/sendThreadMessage.gql

@@ -1,16 +0,0 @@
-mutation SendThreadMessage(
-  $message: String!
-  $timezone: String!
-  $threadId: String!
-  $buildFqdn: String
-) {
-  send_thread_message(
-    threadId: $threadId
-    timezone: $timezone
-    message: $message
-    buildFqdn: $buildFqdn
-  ) {
-    thread_event_id
-    created_at
-  }
-}

package/src/graphql/mutations/startThread.gql

@@ -1,19 +0,0 @@
-mutation StartThread(
-  $message: String!
-  $projectId: String!
-  $timezone: String!
-  $buildId: String
-  $buildFqdn: String
-) {
-  start_thread(
-    message: $message
-    projectId: $projectId
-    timezone: $timezone
-    buildId: $buildId
-    buildFqdn: $buildFqdn
-  ) {
-    thread_id
-    title
-    created_at
-  }
-}

package/src/graphql/queries/getProjectInfo.gql

@@ -1,7 +0,0 @@
-query GetProjectInfo($projectId: uuid!) {
-  get_project_info(projectId: $projectId) {
-    projectId
-    projectName
-    promptqlConsoleUrl
-  }
-}

package/src/graphql/queries/thread.gql

@@ -1,39 +0,0 @@
-fragment Thread on threads_v2 {
-  thread_id
-  build_id
-  title
-  relevant_event_count
-  visibility
-  created_at
-  updated_at
-  user {
-    promptql_user_id
-    display_name
-    email
-    is_active
-    created_at
-    updated_at
-  }
-}
-
-query GetThreads(
-  $where: threads_v2_bool_exp!
-  $limit: Int
-  $offset: Int
-  $order_by: [threads_v2_order_by!]
-) {
-  threads_v2(
-    where: $where
-    limit: $limit
-    offset: $offset
-    order_by: $order_by
-  ) {
-    ...Thread
-  }
-}
-
-query GetThreadById($id: uuid!) {
-  threads_v2_by_pk(thread_id: $id) {
-    ...Thread
-  }
-}

package/src/graphql/subscriptions/subscribeThreadEventsByThreadId.gql

@@ -1,7 +0,0 @@
-subscription SubscribeThreadEvents($where: thread_events_bool_exp!) {
-  thread_events(where: $where, order_by: { thread_event_id: asc }) {
-    thread_event_id
-    event_data
-    created_at
-  }
-}

package/src/index.ts

@@ -1,13 +0,0 @@
-import type { ThreadEvent as ThreadEventData } from "./generated/ChatV2ServerTypes";
-import type { ThreadEvent } from "./sdk";
-
-export * from "./auth";
-export * from "./generated/ChatV2ServerTypes";
-export type {
-  GetThreadsQueryVariables,
-  Order_By,
-  ThreadFragment,
-} from "./generated/graphql";
-export * from "./sdk";
-
-export type { ThreadEvent, ThreadEventData };

package/src/sdk/apollo.ts

@@ -1,63 +0,0 @@
-import {
-  ApolloClient,
-  ApolloLink,
-  HttpLink,
-  InMemoryCache,
-} from "@apollo/client";
-import { SetContextLink } from "@apollo/client/link/context";
-import { GraphQLWsLink } from "@apollo/client/link/subscriptions";
-import { OperationTypeNode } from "graphql";
-import { createClient } from "graphql-ws";
-
-export type ApolloClientOptions = {
-  url: string;
-  headers?: Record<string, string>;
-  getAuthToken: () => Promise<string>;
-  fetch?: typeof fetch;
-};
-
-export const createApolloClient = (options: ApolloClientOptions) => {
-  const httpLink = new HttpLink({
-    uri: options.url,
-    fetch: options.fetch,
-    headers: options.headers,
-  });
-
-  const setAuthContext = async (headers?: Record<string, string>) => {
-    const token = await options.getAuthToken();
-
-    return {
-      headers: {
-        ...headers,
-        Authorization: `Bearer ${token}`,
-      },
-    };
-  };
-
-  const wsLink = new GraphQLWsLink(
-    createClient({
-      url: options.url,
-      connectionParams: () => setAuthContext(options.headers),
-    }),
-  );
-
-  const authLink = new SetContextLink(({ headers }) => setAuthContext(headers));
-
-  // The split function takes three parameters:
-  //
-  // * A function that's called for each operation to execute
-  // * The Link to use for an operation if the function returns a "truthy" value
-  // * The Link to use for an operation if the function returns a "falsy" value
-  const splitLink = ApolloLink.split(
-    ({ operationType }) => {
-      return operationType === OperationTypeNode.SUBSCRIPTION;
-    },
-    wsLink,
-    authLink.concat(httpLink),
-  );
-
-  return new ApolloClient({
-    link: splitLink,
-    cache: new InMemoryCache(),
-  });
-};

package/src/sdk/client.ts

@@ -1,273 +0,0 @@
-import type { ApolloClient, Observable } from "@apollo/client";
-import { map, takeWhile } from "rxjs/operators";
-import { createPromptQLAuthTokenGenerator } from "../auth/auth";
-import {
-  GetProjectInfoDocument,
-  type GetProjectInfoOutput,
-  GetThreadByIdDocument,
-  GetThreadsDocument,
-  type GetThreadsQueryVariables,
-  SendThreadMessageDocument,
-  StartThreadDocument,
-  SubscribeThreadEventsDocument,
-  Thread_Events_Bool_Exp,
-  type ThreadFragment,
-} from "../generated/graphql";
-import { normalizeBaseUrl } from "../utils";
-import { createApolloClient } from "./apollo";
-import type {
-  PromptQLSdkOptions,
-  SendMessageToThreadArguments,
-  SendMessageToThreadOutput,
-  StartThreadArguments,
-  StartThreadOutput,
-  ThreadEvent,
-} from "./types";
-
-/**
- * PromptQL SDK allows you to interact with PromptQL API.
- */
-export class PromptQLSdk {
-  constructor(options: PromptQLSdkOptions) {
-    const promptqlBaseUrl =
-      normalizeBaseUrl(options.promptqlBaseUrl) ||
-      "https://promptql.ddn.hasura.app";
-
-    if (!options.projectId) {
-      throw new Error(
-        "projectId must not be empty. You can find it in Project Settings > General Settings tab on https://promptql.console.hasura.io.",
-      );
-    }
-
-    if (!options.serviceAccountToken) {
-      throw new Error(
-        "serviceAccountToken must not be empty. If you haven't created any service account token yet, check out Hasura docs https://hasura.io/docs/3.0/project-configuration/project-management/service-accounts/#how-to-create-service-account",
-      );
-    }
-
-    const promptqlGraphQLUrl = promptqlBaseUrl.endsWith("/v1/graphql")
-      ? promptqlBaseUrl
-      : `${promptqlBaseUrl}/playground-v2-hge/v1/graphql`;
-
-    const authFetcher = createPromptQLAuthTokenGenerator({
-      promptqlGraphQLUrl,
-      authHost: options.authHost,
-      projectId: options.projectId,
-      serviceAccountToken: options.serviceAccountToken,
-      fetch: options.fetch,
-    });
-
-    this.projectId = options.projectId;
-    this.buildId = options.buildId;
-
-    this.client = createApolloClient({
-      getAuthToken: authFetcher,
-      url: promptqlGraphQLUrl,
-      fetch: options.fetch,
-      headers: options.headers,
-    });
-
-    this.defaultTimezone =
-      options.timezone || Intl.DateTimeFormat().resolvedOptions().timeZone;
-  }
-
-  private client: ApolloClient;
-  private projectId: string;
-  private buildId: string | undefined;
-  private defaultTimezone: string;
-
-  /**
-   * Get the basic information of the PromptQL project.
-   */
-  public getProjectInfo(): Promise<GetProjectInfoOutput> {
-    return this.client
-      .query({
-        query: GetProjectInfoDocument,
-        variables: {
-          projectId: this.projectId,
-        },
-      })
-      .then((result) => {
-        if (!result.data?.get_project_info) {
-          throw new Error(result.error?.message || "Project not found");
-        }
-
-        return result.data.get_project_info;
-      });
-  }
-
-  /**
-   * List threads of the current project. The default limit is 10.
-   */
-  public getThreads(
-    variables: GetThreadsQueryVariables,
-  ): Promise<ThreadFragment[]> {
-    return this.client
-      .query({
-        query: GetThreadsDocument,
-        variables: {
-          ...variables,
-          limit: variables.limit && variables.limit > 0 ? variables.limit : 10,
-        },
-        fetchPolicy: "no-cache",
-      })
-      .then((result) => result.data?.threads_v2 ?? []);
-  }
-
-  /**
-   * Get a thread by ID.
-   */
-  public getThreadById(
-    threadId: string,
-  ): Promise<ThreadFragment | null | undefined> {
-    if (!threadId) {
-      throw new Error("threadId is required");
-    }
-
-    return this.client
-      .query({
-        query: GetThreadByIdDocument,
-        variables: {
-          id: threadId,
-        },
-        fetchPolicy: "no-cache",
-      })
-      .then((result) => result.data?.threads_v2_by_pk);
-  }
-
-  /**
-   * Start a PromptQL thread.
-   */
-  public startThread(
-    variables: StartThreadArguments,
-  ): Promise<StartThreadOutput> {
-    if (!variables.message.trim()) {
-      throw new Error("message must not be empty");
-    }
-
-    return this.client
-      .mutate({
-        mutation: StartThreadDocument,
-        variables: {
-          ...variables,
-          projectId: this.projectId,
-          buildId: variables.buildId || this.buildId,
-          timezone: variables.timezone || this.defaultTimezone,
-        },
-      })
-      .then((result) => {
-        if (!result.data?.start_thread) {
-          throw new Error(result.error?.message || "Failed to start thread");
-        }
-
-        return result.data.start_thread;
-      });
-  }
-
-  /**
-   * Send a message to an existing thread.
-   */
-  public sendMessageToThread(
-    variables: SendMessageToThreadArguments,
-  ): Promise<SendMessageToThreadOutput> {
-    if (!variables.threadId) {
-      throw new Error("threadId is required");
-    }
-
-    if (!variables.message.trim()) {
-      throw new Error("message must not be empty");
-    }
-
-    return this.client
-      .mutate({
-        mutation: SendThreadMessageDocument,
-        variables: {
-          ...variables,
-          timezone: variables.timezone || this.defaultTimezone,
-        },
-      })
-      .then((result) => {
-        if (!result.data?.send_thread_message) {
-          throw new Error(result.error?.message || "Failed to continue thread");
-        }
-
-        return result.data.send_thread_message;
-      });
-  }
-
-  /**
-   * Subscribe to events of a thread by ID. The subscription won't stop even if the interaction is completed.
-   */
-  public subscribeThreadEventsByThreadId(
-    threadId: string,
-    threadEventId?: string | null,
-  ): Observable<ThreadEvent[]> {
-    if (!threadId) {
-      throw new Error("threadId is required");
-    }
-
-    if (threadEventId !== null && threadEventId !== undefined) {
-      try {
-        Number.parseInt(threadEventId);
-      } catch {
-        throw new Error("invalid threadEventId");
-      }
-    }
-
-    const where: Thread_Events_Bool_Exp = {
-      thread_id: { _eq: threadId },
-    };
-
-    if (threadEventId) {
-      where.thread_event_id = {
-        _gte: threadEventId,
-      };
-    }
-
-    return this.client
-      .subscribe({
-        query: SubscribeThreadEventsDocument,
-        variables: {
-          where,
-        },
-      })
-      .pipe(
-        map((value) => {
-          if (value.error) {
-            throw value.error;
-          }
-
-          return (value.data?.thread_events ?? []) as ThreadEvent[];
-        }),
-      );
-  }
-
-  /**
-   * Stream events of a thread by ID. Stop the subscription right after received the completed event.
-   */
-  public streamThreadEventsByThreadId(
-    threadId: string,
-    threadEventId?: string | null,
-  ): Observable<ThreadEvent[]> {
-    return this.subscribeThreadEventsByThreadId(threadId, threadEventId).pipe(
-      takeWhile((events) => {
-        for (let i = events.length - 1; i > 0; i--) {
-          const event = events[i]!;
-          if (
-            "AgentMessage" in event.event_data &&
-            "InteractionFinished" in event.event_data.AgentMessage.update
-          ) {
-            return false;
-          }
-        }
-
-        return true;
-      }, true),
-    );
-  }
-}
-
-/**
- * The interface of PromptQLSdk
- */
-export type IPromptQLSdk = InstanceType<typeof PromptQLSdk>;

package/src/sdk/index.ts

@@ -1,4 +0,0 @@
-export * from "./apollo";
-export * from "./client";
-export * from "./types";
-export * from "./utils";