add test workflow

Author: augustas1

.github/workflows/main.yaml

@@ -0,0 +1,18 @@
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+
+jobs:
+  test:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v4
+    - uses: actions/setup-node@v4
+    - uses: pnpm/action-setup@v4
+      with:
+        version: 10
+    - run: pnpm install
+    # - run: pnpm test
+    - run: pnpm build

src/query/agent.ts

@@ -100,23 +100,40 @@ export class QueryAgent {
    */
   stream(
     query: string,
-    options: QueryAgentStreamOptions & { includeProgress: false; includeFinalState: false }
+    options: QueryAgentStreamOptions & {
+      includeProgress: false;
+      includeFinalState: false;
+    }
   ): AsyncGenerator<StreamedTokens>;
   stream(
     query: string,
-    options: QueryAgentStreamOptions & { includeProgress: false; includeFinalState?: true }
+    options: QueryAgentStreamOptions & {
+      includeProgress: false;
+      includeFinalState?: true;
+    }
   ): AsyncGenerator<StreamedTokens | QueryAgentResponse>;
   stream(
     query: string,
-    options: QueryAgentStreamOptions & { includeProgress?: true; includeFinalState: false }
+    options: QueryAgentStreamOptions & {
+      includeProgress?: true;
+      includeFinalState: false;
+    }
   ): AsyncGenerator<ProgressMessage | StreamedTokens>;
   stream(
     query: string,
-    options?: QueryAgentStreamOptions & { includeProgress?: true; includeFinalState?: true }
+    options?: QueryAgentStreamOptions & {
+      includeProgress?: true;
+      includeFinalState?: true;
+    }
   ): AsyncGenerator<ProgressMessage | StreamedTokens | QueryAgentResponse>;
   async *stream(
     query: string,
-    { collections, context, includeProgress, includeFinalState }: QueryAgentStreamOptions = {}
+    {
+      collections,
+      context,
+      includeProgress,
+      includeFinalState,
+    }: QueryAgentStreamOptions = {}
   ): AsyncGenerator<ProgressMessage | StreamedTokens | QueryAgentResponse> {
     const targetCollections = collections ?? this.collections;
 
@@ -152,10 +169,10 @@ export class QueryAgent {
     for await (const event of sseStream) {
       if (event.event === "error") {
         await handleError(event.data);
-      } 
+      }
 
       let output: ProgressMessage | StreamedTokens | QueryAgentResponse;
-      if (event.event === "progress_message") {
+      if (event.eevent === "progress_message") {
         output = mapProgressMessageFromSSE(event);
       } else if (event.event === "streamed_tokens") {
         output = mapStreamedTokensFromSSE(event);