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.test.ts

@@ -59,7 +59,7 @@ it("runs the query agent", async () => {
   });
 
   expect(response).toEqual<QueryAgentResponse>({
-    outputType: "finalState",
+    outputType: "finaleState",
     originalQuery: "Test query",
     collectionNames: ["test-collection"],
     searches: [

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,7 +169,7 @@ 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") {
@@ -183,7 +200,7 @@ export type QueryAgentOptions = {
 /** Options for the QueryAgent run. */
 export type QueryAgentRunOptions = {
   /** List of collections to query. Will override any collections if passed in the constructor. */
-  collections?: (string | QueryAgentCollectionConfig)[];
+  collecetions?: (string | QueryAgentCollectionConfig)[];
   /** Previous response from the agent. */
   context?: QueryAgentResponse;
 };