prettier

Author: augustas1

.github/workflows/main.yaml

@@ -8,12 +8,13 @@ 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 lint
-    - run: pnpm build
-    - run: pnpm test
+      - uses: actions/checkout@v4
+      - uses: actions/setup-node@v4
+      - uses: pnpm/action-setup@v4
+        with:
+          version: 10
+      - run: pnpm install
+      - run: pnpm lint
+      - run: pnpm format
+      - run: pnpm build
+      - run: pnpm test

.github/workflows/release.yaml

@@ -4,17 +4,18 @@ name: Release
 on:
   push:
     tags:
+      - "*"
 
 jobs:
   publish:
     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 publish
-      env:
-        NODE_AUTH_TOKEN: ${{ secrets.NPM_TOKEN }}
+      - uses: actions/checkout@v4
+      - uses: actions/setup-node@v4
+      - uses: pnpm/action-setup@v4
+        with:
+          version: 10
+      - run: pnpm install
+      - run: pnpm publish
+        env:
+          NODE_AUTH_TOKEN: ${{ secrets.NPM_TOKEN }}

.prettierignore

@@ -0,0 +1 @@
+pnpm-lock.yaml

eslint.config.js

@@ -3,25 +3,24 @@ import globals from "globals";
 import tseslint from "typescript-eslint";
 import { defineConfig, globalIgnores } from "eslint/config";
 
-
 export default defineConfig([
   { files: ["**/*.{ts}"], plugins: { js }, extends: ["js/recommended"] },
   { files: ["**/*.{ts}"], languageOptions: { globals: globals.node } },
   tseslint.configs.recommended,
-  globalIgnores(['dist']),
+  globalIgnores(["dist"]),
   {
     rules: {
-      '@typescript-eslint/naming-convention': [
-          'error',
-          {
-            selector: 'default',
-            format: ['camelCase', 'PascalCase'],
-          },
-          {
-            selector: ['objectLiteralProperty', 'typeProperty'],
-            format: null,
-          },
-        ],
+      "@typescript-eslint/naming-convention": [
+        "error",
+        {
+          selector: "default",
+          format: ["camelCase", "PascalCase"],
+        },
+        {
+          selector: ["objectLiteralProperty", "typeProperty"],
+          format: null,
+        },
+      ],
     },
   },
 ]);

jest.config.js

@@ -4,7 +4,7 @@ import { createDefaultPreset } from "ts-jest";
 export default {
   ...createDefaultPreset(),
   moduleNameMapper: {
-    '^(\\.{1,2}/.*)\\.js$': '$1',
+    "^(\\.{1,2}/.*)\\.js$": "$1",
   },
-  testPathIgnorePatterns: ["dist"]
-};
+  testPathIgnorePatterns: ["dist"],
+};

package.json

@@ -9,6 +9,7 @@
   },
   "scripts": {
     "lint": "eslint",
+    "format": "prettier --check .",
     "build": "tsc",
     "test": "jest",
     "prepack": "pnpm build"
@@ -34,8 +35,9 @@
     "eslint": "^9.26.0",
     "globals": "^16.1.0",
     "jest": "^30.0.4",
+    "prettier": "^3.6.2",
     "ts-jest": "^29.4.0",
     "typescript": "^5.8.3",
     "typescript-eslint": "^8.32.1"
   }
-}
+}

pnpm-lock.yaml

@@ -47,7 +47,7 @@ it("runs the query agent", async () => {
             },
           ],
         }),
-    })
+    }),
   ) as jest.Mock;
 
   const agent = new QueryAgent(mockClient, {

src/query/agent.test.ts

@@ -41,7 +41,7 @@ export class QueryAgent {
       collections,
       systemPrompt,
       agentsHost = "https://api.agents.weaviate.io",
-    }: QueryAgentOptions = {}
+    }: QueryAgentOptions = {},
   ) {
     this.collections = collections;
     this.systemPrompt = systemPrompt;
@@ -57,7 +57,7 @@ export class QueryAgent {
    */
   async run(
     query: string,
-    { collections, context }: QueryAgentRunOptions = {}
+    { collections, context }: QueryAgentRunOptions = {},
   ): Promise<QueryAgentResponse> {
     const targetCollections = collections ?? this.collections;
     if (!targetCollections) {
@@ -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;
 
@@ -146,13 +163,13 @@ export class QueryAgent {
           include_progress: includeProgress ?? true,
           include_final_state: includeFinalState ?? true,
         }),
-      }
+      },
     );
 
     for await (const event of sseStream) {
       if (event.event === "error") {
         await handleError(event.data);
-      } 
+      }
 
       let output: ProgressMessage | StreamedTokens | QueryAgentResponse;
       if (event.event === "progress_message") {

src/query/agent.ts

@@ -1,5 +1,5 @@
 export const mapCollections = (
-  collections: (string | QueryAgentCollectionConfig)[]
+  collections: (string | QueryAgentCollectionConfig)[],
 ) =>
   collections.map((collection) =>
     typeof collection === "string"
@@ -9,7 +9,7 @@ export const mapCollections = (
           view_properties: collection.viewProperties,
           target_vector: collection.targetVector,
           tenant: collection.tenant,
-        }
+        },
   );
 
 /** Configuration for a collection to query. */