ai-tool-set 1.1.0 → 1.2.1

package/README.md

@@ -1,6 +1,9 @@
 <div align='center'>
 
-# ai-tool-set
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="assets/logo-dark.png" />
+  <img src="assets/logo-light.png" alt="ai-tool-set logo" width="400" />
+</picture>
 
 <p align="center">Conditional tool activation for the AI SDK, fully type-safe</p>
 <p align="center">
@@ -66,9 +69,7 @@ Use `.activate()` and `.deactivate()` to statically control which tools are avai
 import { generateText } from 'ai';
 
 // Activate and deactivate tools
-const toolSet = createToolSet({ tools })
-  .deactivate(['cancel_order'])
-  .activate(['list_orders']);
+const toolSet = createToolSet({ tools }).deactivate(['cancel_order']).activate(['list_orders']);
 
 // Infer active tools
 const { tools, activeTools } = toolSet.inferTools();
@@ -142,11 +143,10 @@ const result = await generateText({ model, tools, activeTools, messages });
 You can also activate multiple tools at once:
 
 ```typescript
-const toolSet = createToolSet({ tools })
-  .activateWhen({
-    list_orders: ({ context }) => context?.isAuthenticated,
-    cancel_order: ({ messages }) => hasUnfulfilledOrders(messages),
-  });
+const toolSet = createToolSet({ tools }).activateWhen({
+  list_orders: ({ context }) => context?.isAuthenticated,
+  cancel_order: ({ messages }) => hasUnfulfilledOrders(messages),
+});
 ```
 
 ### Activation Defaults
@@ -184,11 +184,11 @@ Each activation method appends to an internal list. For each tool, the **last en
 ```typescript
 const toolSet = createToolSet({ tools })
   // cancel_order: activated
-  .activate(['cancel_order']) 
+  .activate(['cancel_order'])
   // cancel_order: deactivated
-  .deactivate(['cancel_order']) 
+  .deactivate(['cancel_order'])
   // cancel_order: deactivated with conditional activation
-  .activateWhen('cancel_order', ({ messages }) => hasUnfulfilledOrders(messages)); 
+  .activateWhen('cancel_order', ({ messages }) => hasUnfulfilledOrders(messages));
 ```
 
 ### Immutable vs Mutable
@@ -297,14 +297,12 @@ If you already have a custom `UIMessage` type, you can pass it as `MESSAGE` gene
 import { myTools } from './my-tools.js';
 import { MyUIMessage } from './my-ui-message.js';
 
-const toolSet = createToolSet<typeof myTools, MyUIMessage>({ tools: myTools })
-  .activateWhen(
-    'cancel_order',
-    ({ messages }) => hasUnfulfilledOrders(messages),
-    // ~~~~~~~~
-    // Messages are now typed as Array<MyUIMessage> | undefined  
-  );
-
+const toolSet = createToolSet<typeof myTools, MyUIMessage>({ tools: myTools }).activateWhen(
+  'cancel_order',
+  ({ messages }) => hasUnfulfilledOrders(messages),
+  // ~~~~~~~~
+  // Messages are now typed as Array<MyUIMessage> | undefined
+);
 
 const { tools, activeTools } = toolSet.inferTools({ messages });
 ```
@@ -319,14 +317,12 @@ import { MyUIMessage } from './my-ui-message.js';
 
 type MyContext = { userId: string; isAdmin: boolean };
 
-const toolSet = createToolSet<typeof myTools, MyUIMessage, MyContext>({ tools: myTools })
-  .activateWhen(
-    'cancel_order',
-    ({ context }) => context?.isAdmin,
-    // ~~~~~~~
-    // Context is typed as MyContext | undefined
-  );
-
+const toolSet = createToolSet<typeof myTools, MyUIMessage, MyContext>({ tools: myTools }).activateWhen(
+  'cancel_order',
+  ({ context }) => context?.isAdmin,
+  // ~~~~~~~
+  // Context is typed as MyContext | undefined
+);
 
 const { tools, activeTools } = toolSet.inferTools({
   messages,
@@ -521,6 +517,19 @@ type Inactive = InferInactiveTools<typeof toolSet>;
 // 'cancel_order'
 ```
 
+### `InferAllTools`
+
+Extract all tool names from a `ToolSet` instance, regardless of activation state. Works for both immutable and mutable toolsets since the tool record is statically known.
+
+```ts
+import type { InferAllTools } from 'ai-tool-set';
+
+const toolSet = createToolSet({ tools }).deactivate(['cancel_order']);
+
+type All = InferAllTools<typeof toolSet>;
+// 'search' | 'list_orders' | 'cancel_order'
+```
+
 ## License
 
 MIT

package/dist/index.d.mts

@@ -21,6 +21,11 @@ type InferActiveTools<TOOLSET extends AnyToolSet> = TOOLSET extends ImmutableToo
  * Returns `never` for MutableToolSet (cannot be determined at compile time).
  */
 type InferInactiveTools<TOOLSET extends AnyToolSet> = TOOLSET extends ImmutableToolSet<any, any, any, any, infer D> ? D : never;
+/**
+ * Extract all tool names from a ToolSet instance — both active and inactive.
+ * Works for both ImmutableToolSet and MutableToolSet since the tool record is statically known.
+ */
+type InferAllTools<TOOLSET extends AnyToolSet> = keyof InferToolSet<TOOLSET> & string;
 /**
  * Input passed to activation predicates.
  * Use `ActivationInput<MyMsg>` to get per-tool narrowing in callbacks.
@@ -176,4 +181,4 @@ declare function createToolSet<const TOOLS extends ToolRecord, MESSAGE extends M
   mutable?: false;
 }): ImmutableToolSet<TOOLS, MESSAGE, CONTEXT, keyof TOOLS & string>;
 //#endregion
-export { type ActivationInput, type InferActiveTools, type InferInactiveTools, type InferToolSet, type InferUIToolSet, type ToolSet, createToolSet };
+export { type ActivationInput, type InferActiveTools, type InferAllTools, type InferInactiveTools, type InferToolSet, type InferUIToolSet, type ToolSet, createToolSet };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-tool-set",
-  "version": "1.1.0",
+  "version": "1.2.1",
   "description": "Conditional tool activation for the AI SDK, fully type-safe",
   "keywords": [
     "ai",
@@ -28,6 +28,7 @@
     "@total-typescript/tsconfig": "^1.0.4",
     "@types/node": "^25.5.2",
     "ai": "^6.0.153",
+    "ai-test-kit": "^1.2.0",
     "husky": "^9.1.7",
     "lint-staged": "^16.4.0",
     "oxfmt": "^0.44.0",