@orpc/ai-sdk 1.14.9 → 1.14.11

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/README.md CHANGED
@@ -1,17 +1,16 @@
1
- <div align="center">
2
- <image align="center" src="https://orpc.dev/logo.webp" width=280 alt="oRPC logo" />
3
- </div>
4
-
5
- <h1></h1>
1
+ <h1 align="center">oRPC - Typesafe APIs Made Simple 🪄</h1>
6
2
 
7
3
  <div align="center">
8
4
  <a href="https://codecov.io/gh/middleapi/orpc">
9
- <img alt="codecov" src="https://codecov.io/gh/middleapi/orpc/branch/1.x/graph/badge.svg">
5
+ <img alt="codecov" src="https://codecov.io/gh/middleapi/orpc/branch/main/graph/badge.svg">
10
6
  </a>
11
7
  <a href="https://www.npmjs.com/package/@orpc/ai-sdk">
12
8
  <img alt="weekly downloads" src="https://img.shields.io/npm/dw/%40orpc%2Fai-sdk?logo=npm" />
13
9
  </a>
14
- <a href="https://github.com/middleapi/orpc/blob/1.x/LICENSE">
10
+ <a href="https://app.codspeed.io/middleapi/orpc?utm_source=badge">
11
+ <img alt="CodSpeed" src="https://img.shields.io/endpoint?url=https://codspeed.io/badge.json" />
12
+ </a>
13
+ <a href="https://github.com/middleapi/orpc/blob/main/LICENSE">
15
14
  <img alt="MIT License" src="https://img.shields.io/github/license/middleapi/orpc?logo=open-source-initiative" />
16
15
  </a>
17
16
  <a href="https://discord.gg/TXEbwRBvQn">
@@ -22,55 +21,53 @@
22
21
  </a>
23
22
  </div>
24
23
 
25
- <h3 align="center">Typesafe APIs Made Simple 🪄</h3>
24
+ ## Documentation
26
25
 
27
- **oRPC is a powerful combination of RPC and OpenAPI**, makes it easy to build APIs that are end-to-end type-safe and adhere to OpenAPI standards
26
+ You can read the documentation [here](https://orpc.dev).
28
27
 
29
- ---
28
+ ## Packages
30
29
 
31
- ## Highlights
30
+ **Core**
32
31
 
33
- - **🔗 End-to-End Type Safety**: Ensure type-safe inputs, outputs, and errors from client to server.
34
- - **📘 First-Class OpenAPI**: Built-in support that fully adheres to the OpenAPI standard.
35
- - **📝 Contract-First Development**: Optionally define your API contract before implementation.
36
- - **🔍 First-Class OpenTelemetry**: Seamlessly integrate with OpenTelemetry for observability.
37
- - **⚙️ Framework Integrations**: Seamlessly integrate with TanStack Query (React, Vue, Solid, Svelte, Angular), SWR, Pinia Colada, and more.
38
- - **🚀 Server Actions**: Fully compatible with React Server Actions on Next.js, TanStack Start, and other platforms.
39
- - **🔠 Standard Schema Support**: Works out of the box with Zod, Valibot, ArkType, and other schema validators.
40
- - **🗃️ Native Types**: Supports native types like Date, File, Blob, BigInt, URL, and more.
41
- - **⏱️ Lazy Router**: Enhance cold start times with our lazy routing feature.
42
- - **📡 SSE & Streaming**: Enjoy full type-safe support for SSE and streaming.
43
- - **🌍 Multi-Runtime Support**: Fast and lightweight on Cloudflare, Deno, Bun, Node.js, and beyond.
44
- - **🔌 Extendability**: Easily extend functionality with plugins, middleware, and interceptors.
32
+ - [@orpc/contract](https://www.npmjs.com/package/@orpc/contract): Define API contract as the single source of truth.
33
+ - [@orpc/server](https://www.npmjs.com/package/@orpc/server): Build APIs or implement contracts.
34
+ - [@orpc/client](https://www.npmjs.com/package/@orpc/client): Consume APIs with end-to-end type safety.
35
+ - [@orpc/openapi](https://www.npmjs.com/package/@orpc/openapi): Add OpenAPI compatibility to APIs.
45
36
 
46
- ## Documentation
37
+ **Schema validation**
47
38
 
48
- You can find the full documentation [here](https://orpc.dev).
39
+ - [@orpc/zod](https://www.npmjs.com/package/@orpc/zod): Integrate with [Zod](https://zod.dev/).
40
+ - [@orpc/valibot](https://www.npmjs.com/package/@orpc/valibot): Integrate with [Valibot](https://valibot.dev/).
41
+ - [@orpc/arktype](https://www.npmjs.com/package/@orpc/arktype): Integrate with [ArkType](https://arktype.io/).
49
42
 
50
- ## Packages
43
+ **Built-in features**
44
+
45
+ - [@orpc/publisher](https://www.npmjs.com/package/@orpc/publisher): Pub/Sub with memory, Redis, and Upstash adapters.
46
+ - [@orpc/ratelimit](https://www.npmjs.com/package/@orpc/ratelimit): Rate limiting with memory, Redis, and Upstash adapters.
47
+ - [@orpc/json-schema](https://www.npmjs.com/package/@orpc/json-schema): Smart coercion for OpenAPI requests.
51
48
 
52
- - [@orpc/contract](https://www.npmjs.com/package/@orpc/contract): Build your API contract.
53
- - [@orpc/server](https://www.npmjs.com/package/@orpc/server): Build your API or implement API contract.
54
- - [@orpc/client](https://www.npmjs.com/package/@orpc/client): Consume your API on the client with type-safety.
55
- - [@orpc/openapi](https://www.npmjs.com/package/@orpc/openapi): Generate OpenAPI specs and handle OpenAPI requests.
56
- - [@orpc/otel](https://www.npmjs.com/package/@orpc/otel): [OpenTelemetry](https://opentelemetry.io/) integration for observability.
57
- - [@orpc/nest](https://www.npmjs.com/package/@orpc/nest): Deeply integrate oRPC with [NestJS](https://nestjs.com/).
58
- - [@orpc/react](https://www.npmjs.com/package/@orpc/react): Utilities for integrating oRPC with React and React Server Actions.
59
- - [@orpc/tanstack-query](https://www.npmjs.com/package/@orpc/tanstack-query): [TanStack Query](https://tanstack.com/query/latest) integration.
60
- - [@orpc/experimental-react-swr](https://www.npmjs.com/package/@orpc/experimental-react-swr): [SWR](https://swr.vercel.app/) integration.
61
- - [@orpc/vue-colada](https://www.npmjs.com/package/@orpc/vue-colada): Integration with [Pinia Colada](https://pinia-colada.esm.dev/).
62
- - [@orpc/hey-api](https://www.npmjs.com/package/@orpc/hey-api): [Hey API](https://heyapi.dev/) integration.
63
- - [@orpc/zod](https://www.npmjs.com/package/@orpc/zod): More schemas that [Zod](https://zod.dev/) doesn't support yet.
64
- - [@orpc/valibot](https://www.npmjs.com/package/@orpc/valibot): OpenAPI spec generation from [Valibot](https://valibot.dev/).
65
- - [@orpc/arktype](https://www.npmjs.com/package/@orpc/arktype): OpenAPI spec generation from [ArkType](https://arktype.io/).
66
-
67
- ## `@orpc/ai-sdk`
68
-
69
- The [AI SDK](https://ai-sdk.dev/) integration for oRPC.
49
+ **Framework & ecosystem integrations**
50
+
51
+ - [@orpc/next](https://www.npmjs.com/package/@orpc/next): Integrate with [Next.js Server Functions](https://nextjs.org/docs/app/getting-started/mutating-data).
52
+ - [@orpc/ai-sdk](https://www.npmjs.com/package/@orpc/ai-sdk): Turn contracts and procedures into [AI SDK](https://ai-sdk.dev/) tools.
53
+ - [@orpc/tanstack-query](https://www.npmjs.com/package/@orpc/tanstack-query): Integrate with [TanStack Query](https://tanstack.com/query/latest).
54
+ - [@orpc/pinia-colada](https://www.npmjs.com/package/@orpc/pinia-colada): Integrate with [Pinia Colada](https://pinia-colada.esm.dev/).
55
+ - [@orpc/swr](https://www.npmjs.com/package/@orpc/swr): Integrate with [SWR](https://swr.vercel.app/).
56
+ - [@orpc/experimental-effect](https://www.npmjs.com/package/@orpc/experimental-effect): Integrate with [Effect](https://effect.website/).
57
+ - [@orpc/nest](https://www.npmjs.com/package/@orpc/nest): Implement your contract with [NestJS](https://nestjs.com/).
58
+ - [@orpc/bun](https://www.npmjs.com/package/@orpc/bun): Adapters for [Bun's Redis](https://bun.sh/).
59
+ - [@orpc/cloudflare](https://www.npmjs.com/package/@orpc/cloudflare): Adapters for [Cloudflare's RateLimit and Durable Objects](https://developers.cloudflare.com/workers/).
60
+ - [@orpc/trpc](https://www.npmjs.com/package/@orpc/trpc): Reuse existing [tRPC](https://trpc.io/) routers within oRPC.
61
+
62
+ **Observability**
63
+
64
+ - [@orpc/opentelemetry](https://www.npmjs.com/package/@orpc/opentelemetry): Integrate with [OpenTelemetry](https://opentelemetry.io/) for distributed tracing.
65
+ - [@orpc/pino](https://www.npmjs.com/package/@orpc/pino): Integrate with [Pino](https://getpino.io/) for logging.
66
+ - [@orpc/evlog](https://www.npmjs.com/package/@orpc/evlog): Integrate with [Evlog](https://evlog.dev/) for logging.
70
67
 
71
68
  ## Sponsors
72
69
 
73
- If you find oRPC valuable and would like to support its development, you can do so here: [GitHub Sponsors](https://github.com/sponsors/dinwwwh).
70
+ Like what we build over at [middleapi](https://github.com/middleapi)? You can help keep it going here: [GitHub Sponsors](https://github.com/sponsors/dinwwwh). Every bit helps! 🚀
74
71
 
75
72
  ### 🏆 Platinum Sponsor
76
73
 
@@ -112,11 +109,10 @@ If you find oRPC valuable and would like to support its development, you can do
112
109
  <td align="center"><a href="https://github.com/valerii15298?ref=orpc" target="_blank" rel="noopener" title="Valerii Petryniak"><img src="https://avatars.githubusercontent.com/u/44531564?u=88ac74d9bacd20401518441907acad21063cd397&amp;v=4" width="139" alt="Valerii Petryniak"/><br />Valerii Petryniak</a></td>
113
110
  <td align="center"><a href="https://github.com/letstri?ref=orpc" target="_blank" rel="noopener" title="Valerii Strilets"><img src="https://avatars.githubusercontent.com/u/13253748?u=c7b10399ccc8f8081e24db94ec32cd9858e86ac3&amp;v=4" width="139" alt="Valerii Strilets"/><br />Valerii Strilets</a></td>
114
111
  <td align="center"><a href="https://github.com/K-Mistele?ref=orpc" target="_blank" rel="noopener" title="Kyle Mistele"><img src="https://avatars.githubusercontent.com/u/18430555?u=3afebeb81de666e35aaac3ed46f14159d7603ffb&amp;v=4" width="139" alt="Kyle Mistele"/><br />Kyle Mistele</a></td>
115
- <td align="center"><a href="https://github.com/R44VC0RP?ref=orpc" target="_blank" rel="noopener" title="Ryan Vogel"><img src="https://avatars.githubusercontent.com/u/89211796?u=1857347b9787d8d8a7ea5bfc333f96be92d5a683&amp;v=4" width="139" alt="Ryan Vogel"/><br />Ryan Vogel</a></td>
116
112
  <td align="center"><a href="https://github.com/christ12938?ref=orpc" target="_blank" rel="noopener" title="christ12938"><img src="https://avatars.githubusercontent.com/u/25758598?v=4" width="139" alt="christ12938"/><br />christ12938</a></td>
113
+ <td align="center"><a href="https://github.com/Ryanjso?ref=orpc" target="_blank" rel="noopener" title="Ryan Soderberg"><img src="https://avatars.githubusercontent.com/u/39172778?u=5ed913c31d57e7221b75784abcad48c7ebddde27&amp;v=4" width="139" alt="Ryan Soderberg"/><br />Ryan Soderberg</a></td>
117
114
  </tr>
118
115
  <tr>
119
- <td align="center"><a href="https://github.com/Ryanjso?ref=orpc" target="_blank" rel="noopener" title="Ryan Soderberg"><img src="https://avatars.githubusercontent.com/u/39172778?u=5ed913c31d57e7221b75784abcad48c7ebddde27&amp;v=4" width="139" alt="Ryan Soderberg"/><br />Ryan Soderberg</a></td>
120
116
  <td align="center"><a href="https://github.com/itigoore01?ref=orpc" target="_blank" rel="noopener" title="shota"><img src="https://avatars.githubusercontent.com/u/11831107?u=c976a6dc7e055eb026304c46c99100ed22b0c8e0&amp;v=4" width="139" alt="shota"/><br />shota</a></td>
121
117
  </tr>
122
118
  </table>
@@ -176,6 +172,7 @@ If you find oRPC valuable and would like to support its development, you can do
176
172
  <a href="https://github.com/nguyenlc1993?ref=orpc" target="_blank" rel="noopener" title="Lê Cao Nguyên"><img src="https://avatars.githubusercontent.com/u/13871971?u=83c8b69d9e35b589c4e1f066cc113b1d9461386f&amp;v=4" width="32" height="32" alt="Lê Cao Nguyên" /></a>
177
173
  <a href="https://github.com/wobsoriano?ref=orpc" target="_blank" rel="noopener" title="Robert Soriano"><img src="https://avatars.githubusercontent.com/u/13049130?u=6d72104182e7c9ed25934815313fb69107332111&amp;v=4" width="32" height="32" alt="Robert Soriano" /></a>
178
174
  <a href="https://github.com/andrewpeters9?ref=orpc" target="_blank" rel="noopener" title="Andrew Peters"><img src="https://avatars.githubusercontent.com/u/36251325?v=4" width="32" height="32" alt="Andrew Peters" /></a>
175
+ <a href="https://github.com/R44VC0RP?ref=orpc" target="_blank" rel="noopener" title="Ryan Vogel"><img src="https://avatars.githubusercontent.com/u/89211796?u=1857347b9787d8d8a7ea5bfc333f96be92d5a683&amp;v=4" width="32" height="32" alt="Ryan Vogel" /></a>
179
176
  <a href="https://github.com/SKostyukovich?ref=orpc" target="_blank" rel="noopener" title="SKostyukovich"><img src="https://avatars.githubusercontent.com/u/10700067?v=4" width="32" height="32" alt="SKostyukovich" /></a>
180
177
  <a href="https://github.com/peter-adam-dy?ref=orpc" target="_blank" rel="noopener" title="Peter Adam"><img src="https://avatars.githubusercontent.com/u/132129459?u=4f3dbbb3b443990b56acb7d6a5d11ed2c555f6db&amp;v=4" width="32" height="32" alt="Peter Adam" /></a>
181
178
  <a href="https://github.com/FabworksHQ?ref=orpc" target="_blank" rel="noopener" title="Fabworks"><img src="https://avatars.githubusercontent.com/u/160179500?v=4" width="32" height="32" alt="Fabworks" /></a>
@@ -190,6 +187,13 @@ If you find oRPC valuable and would like to support its development, you can do
190
187
  <a href="https://github.com/plancraft?ref=orpc" target="_blank" rel="noopener" title="plancraft"><img src="https://avatars.githubusercontent.com/u/46482287?v=4" width="32" height="32" alt="plancraft" /></a>
191
188
  </p>
192
189
 
190
+ ## References
191
+
192
+ oRPC is inspired by existing solutions that prioritize type safety and developer experience. Special acknowledgments to:
193
+
194
+ - [tRPC](https://trpc.io): For pioneering the concept of end-to-end type-safe RPC and influencing the development of type-safe APIs.
195
+ - [ts-rest](https://ts-rest.com): For its emphasis on contract-first development and OpenAPI integration, which have greatly inspired oRPC's feature set.
196
+
193
197
  ## License
194
198
 
195
- Distributed under the MIT License. See [LICENSE](https://github.com/middleapi/orpc/blob/1.x/LICENSE) for more information.
199
+ Distributed under the MIT License. See [LICENSE](https://github.com/middleapi/orpc/blob/main/LICENSE) for more information.
package/dist/index.d.mts CHANGED
@@ -1,45 +1,92 @@
1
1
  import { ClientOptions } from '@orpc/client';
2
- import { Meta, ContractProcedure, Schema, AnySchema, ErrorMap, InferSchemaOutput, InferSchemaInput } from '@orpc/contract';
3
- import { Context, Procedure, CreateProcedureClientOptions } from '@orpc/server';
4
- import { MaybeOptionalOptions, SetOptional } from '@orpc/shared';
5
- export { AsyncIteratorClass, asyncIteratorToStream as eventIteratorToStream, asyncIteratorToUnproxiedDataStream as eventIteratorToUnproxiedDataStream, streamToAsyncIteratorClass as streamToEventIterator } from '@orpc/shared';
6
- import { Tool } from 'ai';
2
+ import { AnySchema, ErrorMap, InferSchemaOutput, InferSchemaInput, MetaPlugin, AnyProcedureContract, Schema, ProcedureContract } from '@orpc/contract';
3
+ import { Lazy, Context, ProcedureClientOptions, Procedure } from '@orpc/server';
4
+ import { MaybeOptionalOptions } from '@orpc/shared';
5
+ import { JSONValue, Tool } from 'ai';
7
6
 
8
- declare const AI_SDK_TOOL_META_SYMBOL: unique symbol;
9
- interface AiSdkToolMeta extends Meta {
10
- [AI_SDK_TOOL_META_SYMBOL]?: Partial<Tool<unknown, unknown>>;
7
+ /**
8
+ * The [function tool](https://ai-sdk.dev/docs/foundations/tools) member of the AI SDK `Tool` union,
9
+ * which is the kind of tool this package produces.
10
+ */
11
+ type FunctionTool<INPUT extends JSONValue | unknown | never = any, OUTPUT extends JSONValue | unknown | never = any> = Extract<Tool<INPUT, OUTPUT>, {
12
+ type?: 'function' | undefined;
13
+ }>;
14
+ /**
15
+ * Base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options attached to a procedure/contract,
16
+ * used as defaults when creating tools from it.
17
+ *
18
+ * **Note**: When defined multiple times, options are overridden by the most recent call.
19
+ *
20
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
21
+ */
22
+ type AiSdkToolMeta<TInputSchema extends AnySchema = AnySchema, TOutputSchema extends AnySchema = AnySchema> = Partial<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema' | 'execute'>>;
23
+ interface AiSdkToolMetaPlugin<TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap> extends MetaPlugin<TInputSchema, TOutputSchema, TErrorMap> {
24
+ name: '~ai-sdk/tool';
11
25
  }
12
- declare class CreateToolError extends Error {
26
+ interface AiSdkToolFunction {
27
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap>(meta: AiSdkToolMeta<TInputSchema, TOutputSchema>): AiSdkToolMetaPlugin<TInputSchema, TOutputSchema, TErrorMap>;
13
28
  }
14
29
  /**
15
- * Implements [procedure contract](https://orpc.dev/docs/contract-first/define-contract#procedure-contract)
16
- * as an [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing contract definitions.
30
+ * Attaches base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options to a procedure/contract,
31
+ * used as defaults when creating tools from it.
32
+ *
33
+ * **Note**: When defined multiple times, options are overridden by the most recent call.
17
34
  *
18
- * @warning Requires a contract with an `input` schema defined.
19
- * @info Standard [procedures](https://orpc.dev/docs/procedure) are also compatible with [procedure contracts](https://orpc.dev/docs/contract-first/define-contract).
35
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
20
36
  *
21
37
  * @example
22
38
  * ```ts
39
+ * const getWeatherContract = oc
40
+ * .meta(aiSdkTool({
41
+ * description: 'Get the weather in a location',
42
+ * metadata: { source: 'weather-service' },
43
+ * }))
44
+ * .input(z.object({ location: z.string() }))
45
+ * ```
46
+ */
47
+ declare const aiSdkTool: AiSdkToolFunction;
48
+ /**
49
+ * Reads the base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options
50
+ * attached to a procedure/contract by {@link aiSdkTool}.
51
+ *
52
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
53
+ */
54
+ declare function getAiSdkToolMeta(procedureOrLazy: AnyProcedureContract | Lazy<any>): AiSdkToolMeta | undefined;
55
+
56
+ /**
57
+ * Infers the value an AI SDK tool outputs for a given oRPC output type.
58
+ *
59
+ * For [AsyncIteratorObject](https://orpc.dev/docs/async-iterator-object) outputs, the tool streams
60
+ * each yielded event as a [preliminary result](https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#preliminary-tool-results)
61
+ * and the last event becomes the final result, so the tool output is the yield type.
62
+ *
63
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#streaming-tool-outputs AI SDK Streaming Tool Outputs Docs}
64
+ */
65
+ type ToolOutput<T> = T extends AsyncIteratorObject<infer TYield, any, any> ? TYield : T;
66
+ interface ImplementToolFactoryOptions {
67
+ }
68
+ interface ToolImplementer {
69
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema>(contract: ProcedureContract<TInputSchema, TOutputSchema, any>, ...rest: MaybeOptionalOptions<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema'>>): Tool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>;
70
+ }
71
+ /**
72
+ * Creates a builder that implements [procedure contracts](https://orpc.dev/docs/contract/procedure)
73
+ * as [AI SDK Tools](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing contract definitions.
74
+ *
75
+ * The factory accepts oRPC related options, and the resulting builder
76
+ * accepts a contract alongside AI SDK tool options.
77
+ *
78
+ * @info [procedures](https://orpc.dev/docs/procedure) are also compatible with [procedure contracts](https://orpc.dev/docs/contract/procedure).
79
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
80
+ *
81
+ * @example
82
+ * ```ts
83
+ * import { aiSdkTool, implementToolFactory } from '@orpc/ai-sdk'
23
84
  * import { oc } from '@orpc/contract'
24
- * import {
25
- * AI_SDK_TOOL_META_SYMBOL,
26
- * AiSdkToolMeta,
27
- * implementTool,
28
- * } from '@orpc/ai-sdk'
29
- * import { z } from 'zod'
30
- *
31
- * interface ORPCMeta extends AiSdkToolMeta {} // optional extend meta
32
- * const base = oc.$meta<ORPCMeta>({})
33
- *
34
- * const getWeatherContract = base
35
- * .meta({
36
- * [AI_SDK_TOOL_META_SYMBOL]: {
37
- * title: 'Get Weather', // AI SDK tool title
38
- * },
39
- * })
40
- * .route({
41
- * summary: 'Get the weather in a location', // AI SDK tool description
42
- * })
85
+ *
86
+ * const getWeatherContract = oc
87
+ * .meta(
88
+ * aiSdkTool({ description: 'Get the weather in a location' }), // default AI SDK tool options
89
+ * )
43
90
  * .input(
44
91
  * z.object({
45
92
  * location: z.string().describe('The location to get the weather for'),
@@ -52,6 +99,8 @@ declare class CreateToolError extends Error {
52
99
  * }),
53
100
  * )
54
101
  *
102
+ * const implementTool = implementToolFactory()
103
+ *
55
104
  * const getWeatherTool = implementTool(getWeatherContract, {
56
105
  * execute: async ({ location }) => ({
57
106
  * location,
@@ -60,35 +109,29 @@ declare class CreateToolError extends Error {
60
109
  * })
61
110
  * ```
62
111
  */
63
- declare function implementTool<TOutInput, TInOutput>(contract: ContractProcedure<Schema<any, TOutInput>, Schema<TInOutput, any>, any, AiSdkToolMeta>, ...rest: MaybeOptionalOptions<SetOptional<Tool<TOutInput, TInOutput>, 'inputSchema' | 'outputSchema'>>): Tool<TOutInput, TInOutput>;
112
+ declare function implementToolFactory(_options?: ImplementToolFactoryOptions): ToolImplementer;
113
+ type CreateToolFactoryOptions<TInitialContext extends Context> = ImplementToolFactoryOptions & ProcedureClientOptions<TInitialContext, Schema<unknown>, object, never, object> & Omit<ClientOptions<object>, 'context'>;
114
+ interface ToolFactory<TInitialContext extends Context> {
115
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema>(procedure: Procedure<TInitialContext, any, TInputSchema, TOutputSchema, any, any>, ...rest: MaybeOptionalOptions<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema' | 'execute'>>): Tool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>;
116
+ }
64
117
  /**
65
- * Converts a [procedure](https://orpc.dev/docs/procedure) into an [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools)
66
- * by leveraging existing procedure definitions.
118
+ * Creates a builder that converts [procedures](https://orpc.dev/docs/procedure)
119
+ * into [AI SDK Tools](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing procedure definitions.
120
+ *
121
+ * The factory accepts oRPC related options, and the resulting builder
122
+ * accepts a procedure alongside AI SDK tool options.
67
123
  *
68
- * @warning Requires a contract with an `input` schema defined.
124
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-factory AI SDK Tool Factory Docs}
69
125
  *
70
126
  * @example
71
127
  * ```ts
128
+ * import { aiSdkTool, createToolFactory } from '@orpc/ai-sdk'
72
129
  * import { os } from '@orpc/server'
73
- * import {
74
- * AI_SDK_TOOL_META_SYMBOL,
75
- * AiSdkToolMeta,
76
- * createTool
77
- * } from '@orpc/ai-sdk'
78
- * import { z } from 'zod'
79
- *
80
- * interface ORPCMeta extends AiSdkToolMeta {} // optional extend meta
81
- * const base = os.$meta<ORPCMeta>({})
82
- *
83
- * const getWeatherProcedure = base
84
- * .meta({
85
- * [AI_SDK_TOOL_META_SYMBOL]: {
86
- * title: 'Get Weather', // AI SDK tool title
87
- * },
88
- * })
89
- * .route({
90
- * summary: 'Get the weather in a location',
91
- * })
130
+ *
131
+ * const getWeatherProcedure = os
132
+ * .meta(
133
+ * aiSdkTool({ description: 'Get the weather in a location' }), // default AI SDK tool options
134
+ * )
92
135
  * .input(z.object({
93
136
  * location: z.string().describe('The location to get the weather for'),
94
137
  * }))
@@ -101,12 +144,16 @@ declare function implementTool<TOutInput, TInOutput>(contract: ContractProcedure
101
144
  * temperature: 72 + Math.floor(Math.random() * 21) - 10,
102
145
  * }))
103
146
  *
104
- * const getWeatherTool = createTool(getWeatherProcedure, {
147
+ * const createTool = createToolFactory({
105
148
  * context: {}, // provide initial context if needed
106
149
  * })
150
+ *
151
+ * const getWeatherTool = createTool(getWeatherProcedure, {
152
+ * // ...AI SDK tool options/overrides here if needed
153
+ * })
107
154
  * ```
108
155
  */
109
- declare function createTool<TInitialContext extends Context, TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap, TMeta extends AiSdkToolMeta>(procedure: Procedure<TInitialContext, any, TInputSchema, TOutputSchema, TErrorMap, TMeta>, ...rest: MaybeOptionalOptions<SetOptional<Tool<InferSchemaOutput<TInputSchema>, InferSchemaInput<TOutputSchema>>, 'inputSchema' | 'outputSchema' | 'execute'> & CreateProcedureClientOptions<TInitialContext, TOutputSchema, TErrorMap, TMeta, Record<never, never>> & Omit<ClientOptions<Record<never, never>>, 'context'>>): Tool<InferSchemaOutput<TInputSchema>, InferSchemaInput<TOutputSchema>>;
156
+ declare function createToolFactory<TInitialContext extends Context = object>(...rest: MaybeOptionalOptions<CreateToolFactoryOptions<TInitialContext>>): ToolFactory<TInitialContext>;
110
157
 
111
- export { AI_SDK_TOOL_META_SYMBOL, CreateToolError, createTool, implementTool };
112
- export type { AiSdkToolMeta };
158
+ export { aiSdkTool, createToolFactory, getAiSdkToolMeta, implementToolFactory };
159
+ export type { AiSdkToolFunction, AiSdkToolMeta, AiSdkToolMetaPlugin, CreateToolFactoryOptions, FunctionTool, ImplementToolFactoryOptions, ToolFactory, ToolImplementer, ToolOutput };
package/dist/index.d.ts CHANGED
@@ -1,45 +1,92 @@
1
1
  import { ClientOptions } from '@orpc/client';
2
- import { Meta, ContractProcedure, Schema, AnySchema, ErrorMap, InferSchemaOutput, InferSchemaInput } from '@orpc/contract';
3
- import { Context, Procedure, CreateProcedureClientOptions } from '@orpc/server';
4
- import { MaybeOptionalOptions, SetOptional } from '@orpc/shared';
5
- export { AsyncIteratorClass, asyncIteratorToStream as eventIteratorToStream, asyncIteratorToUnproxiedDataStream as eventIteratorToUnproxiedDataStream, streamToAsyncIteratorClass as streamToEventIterator } from '@orpc/shared';
6
- import { Tool } from 'ai';
2
+ import { AnySchema, ErrorMap, InferSchemaOutput, InferSchemaInput, MetaPlugin, AnyProcedureContract, Schema, ProcedureContract } from '@orpc/contract';
3
+ import { Lazy, Context, ProcedureClientOptions, Procedure } from '@orpc/server';
4
+ import { MaybeOptionalOptions } from '@orpc/shared';
5
+ import { JSONValue, Tool } from 'ai';
7
6
 
8
- declare const AI_SDK_TOOL_META_SYMBOL: unique symbol;
9
- interface AiSdkToolMeta extends Meta {
10
- [AI_SDK_TOOL_META_SYMBOL]?: Partial<Tool<unknown, unknown>>;
7
+ /**
8
+ * The [function tool](https://ai-sdk.dev/docs/foundations/tools) member of the AI SDK `Tool` union,
9
+ * which is the kind of tool this package produces.
10
+ */
11
+ type FunctionTool<INPUT extends JSONValue | unknown | never = any, OUTPUT extends JSONValue | unknown | never = any> = Extract<Tool<INPUT, OUTPUT>, {
12
+ type?: 'function' | undefined;
13
+ }>;
14
+ /**
15
+ * Base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options attached to a procedure/contract,
16
+ * used as defaults when creating tools from it.
17
+ *
18
+ * **Note**: When defined multiple times, options are overridden by the most recent call.
19
+ *
20
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
21
+ */
22
+ type AiSdkToolMeta<TInputSchema extends AnySchema = AnySchema, TOutputSchema extends AnySchema = AnySchema> = Partial<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema' | 'execute'>>;
23
+ interface AiSdkToolMetaPlugin<TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap> extends MetaPlugin<TInputSchema, TOutputSchema, TErrorMap> {
24
+ name: '~ai-sdk/tool';
11
25
  }
12
- declare class CreateToolError extends Error {
26
+ interface AiSdkToolFunction {
27
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap>(meta: AiSdkToolMeta<TInputSchema, TOutputSchema>): AiSdkToolMetaPlugin<TInputSchema, TOutputSchema, TErrorMap>;
13
28
  }
14
29
  /**
15
- * Implements [procedure contract](https://orpc.dev/docs/contract-first/define-contract#procedure-contract)
16
- * as an [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing contract definitions.
30
+ * Attaches base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options to a procedure/contract,
31
+ * used as defaults when creating tools from it.
32
+ *
33
+ * **Note**: When defined multiple times, options are overridden by the most recent call.
17
34
  *
18
- * @warning Requires a contract with an `input` schema defined.
19
- * @info Standard [procedures](https://orpc.dev/docs/procedure) are also compatible with [procedure contracts](https://orpc.dev/docs/contract-first/define-contract).
35
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
20
36
  *
21
37
  * @example
22
38
  * ```ts
39
+ * const getWeatherContract = oc
40
+ * .meta(aiSdkTool({
41
+ * description: 'Get the weather in a location',
42
+ * metadata: { source: 'weather-service' },
43
+ * }))
44
+ * .input(z.object({ location: z.string() }))
45
+ * ```
46
+ */
47
+ declare const aiSdkTool: AiSdkToolFunction;
48
+ /**
49
+ * Reads the base [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools) options
50
+ * attached to a procedure/contract by {@link aiSdkTool}.
51
+ *
52
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
53
+ */
54
+ declare function getAiSdkToolMeta(procedureOrLazy: AnyProcedureContract | Lazy<any>): AiSdkToolMeta | undefined;
55
+
56
+ /**
57
+ * Infers the value an AI SDK tool outputs for a given oRPC output type.
58
+ *
59
+ * For [AsyncIteratorObject](https://orpc.dev/docs/async-iterator-object) outputs, the tool streams
60
+ * each yielded event as a [preliminary result](https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#preliminary-tool-results)
61
+ * and the last event becomes the final result, so the tool output is the yield type.
62
+ *
63
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#streaming-tool-outputs AI SDK Streaming Tool Outputs Docs}
64
+ */
65
+ type ToolOutput<T> = T extends AsyncIteratorObject<infer TYield, any, any> ? TYield : T;
66
+ interface ImplementToolFactoryOptions {
67
+ }
68
+ interface ToolImplementer {
69
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema>(contract: ProcedureContract<TInputSchema, TOutputSchema, any>, ...rest: MaybeOptionalOptions<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema'>>): Tool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>;
70
+ }
71
+ /**
72
+ * Creates a builder that implements [procedure contracts](https://orpc.dev/docs/contract/procedure)
73
+ * as [AI SDK Tools](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing contract definitions.
74
+ *
75
+ * The factory accepts oRPC related options, and the resulting builder
76
+ * accepts a contract alongside AI SDK tool options.
77
+ *
78
+ * @info [procedures](https://orpc.dev/docs/procedure) are also compatible with [procedure contracts](https://orpc.dev/docs/contract/procedure).
79
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-implementer AI SDK Tool Implementer Docs}
80
+ *
81
+ * @example
82
+ * ```ts
83
+ * import { aiSdkTool, implementToolFactory } from '@orpc/ai-sdk'
23
84
  * import { oc } from '@orpc/contract'
24
- * import {
25
- * AI_SDK_TOOL_META_SYMBOL,
26
- * AiSdkToolMeta,
27
- * implementTool,
28
- * } from '@orpc/ai-sdk'
29
- * import { z } from 'zod'
30
- *
31
- * interface ORPCMeta extends AiSdkToolMeta {} // optional extend meta
32
- * const base = oc.$meta<ORPCMeta>({})
33
- *
34
- * const getWeatherContract = base
35
- * .meta({
36
- * [AI_SDK_TOOL_META_SYMBOL]: {
37
- * title: 'Get Weather', // AI SDK tool title
38
- * },
39
- * })
40
- * .route({
41
- * summary: 'Get the weather in a location', // AI SDK tool description
42
- * })
85
+ *
86
+ * const getWeatherContract = oc
87
+ * .meta(
88
+ * aiSdkTool({ description: 'Get the weather in a location' }), // default AI SDK tool options
89
+ * )
43
90
  * .input(
44
91
  * z.object({
45
92
  * location: z.string().describe('The location to get the weather for'),
@@ -52,6 +99,8 @@ declare class CreateToolError extends Error {
52
99
  * }),
53
100
  * )
54
101
  *
102
+ * const implementTool = implementToolFactory()
103
+ *
55
104
  * const getWeatherTool = implementTool(getWeatherContract, {
56
105
  * execute: async ({ location }) => ({
57
106
  * location,
@@ -60,35 +109,29 @@ declare class CreateToolError extends Error {
60
109
  * })
61
110
  * ```
62
111
  */
63
- declare function implementTool<TOutInput, TInOutput>(contract: ContractProcedure<Schema<any, TOutInput>, Schema<TInOutput, any>, any, AiSdkToolMeta>, ...rest: MaybeOptionalOptions<SetOptional<Tool<TOutInput, TInOutput>, 'inputSchema' | 'outputSchema'>>): Tool<TOutInput, TInOutput>;
112
+ declare function implementToolFactory(_options?: ImplementToolFactoryOptions): ToolImplementer;
113
+ type CreateToolFactoryOptions<TInitialContext extends Context> = ImplementToolFactoryOptions & ProcedureClientOptions<TInitialContext, Schema<unknown>, object, never, object> & Omit<ClientOptions<object>, 'context'>;
114
+ interface ToolFactory<TInitialContext extends Context> {
115
+ <TInputSchema extends AnySchema, TOutputSchema extends AnySchema>(procedure: Procedure<TInitialContext, any, TInputSchema, TOutputSchema, any, any>, ...rest: MaybeOptionalOptions<Omit<FunctionTool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>, 'inputSchema' | 'outputSchema' | 'execute'>>): Tool<InferSchemaOutput<TInputSchema>, ToolOutput<InferSchemaInput<TOutputSchema>>>;
116
+ }
64
117
  /**
65
- * Converts a [procedure](https://orpc.dev/docs/procedure) into an [AI SDK Tool](https://ai-sdk.dev/docs/foundations/tools)
66
- * by leveraging existing procedure definitions.
118
+ * Creates a builder that converts [procedures](https://orpc.dev/docs/procedure)
119
+ * into [AI SDK Tools](https://ai-sdk.dev/docs/foundations/tools) by leveraging existing procedure definitions.
120
+ *
121
+ * The factory accepts oRPC related options, and the resulting builder
122
+ * accepts a procedure alongside AI SDK tool options.
67
123
  *
68
- * @warning Requires a contract with an `input` schema defined.
124
+ * @see {@link https://orpc.dev/docs/integrations/ai-sdk#tool-factory AI SDK Tool Factory Docs}
69
125
  *
70
126
  * @example
71
127
  * ```ts
128
+ * import { aiSdkTool, createToolFactory } from '@orpc/ai-sdk'
72
129
  * import { os } from '@orpc/server'
73
- * import {
74
- * AI_SDK_TOOL_META_SYMBOL,
75
- * AiSdkToolMeta,
76
- * createTool
77
- * } from '@orpc/ai-sdk'
78
- * import { z } from 'zod'
79
- *
80
- * interface ORPCMeta extends AiSdkToolMeta {} // optional extend meta
81
- * const base = os.$meta<ORPCMeta>({})
82
- *
83
- * const getWeatherProcedure = base
84
- * .meta({
85
- * [AI_SDK_TOOL_META_SYMBOL]: {
86
- * title: 'Get Weather', // AI SDK tool title
87
- * },
88
- * })
89
- * .route({
90
- * summary: 'Get the weather in a location',
91
- * })
130
+ *
131
+ * const getWeatherProcedure = os
132
+ * .meta(
133
+ * aiSdkTool({ description: 'Get the weather in a location' }), // default AI SDK tool options
134
+ * )
92
135
  * .input(z.object({
93
136
  * location: z.string().describe('The location to get the weather for'),
94
137
  * }))
@@ -101,12 +144,16 @@ declare function implementTool<TOutInput, TInOutput>(contract: ContractProcedure
101
144
  * temperature: 72 + Math.floor(Math.random() * 21) - 10,
102
145
  * }))
103
146
  *
104
- * const getWeatherTool = createTool(getWeatherProcedure, {
147
+ * const createTool = createToolFactory({
105
148
  * context: {}, // provide initial context if needed
106
149
  * })
150
+ *
151
+ * const getWeatherTool = createTool(getWeatherProcedure, {
152
+ * // ...AI SDK tool options/overrides here if needed
153
+ * })
107
154
  * ```
108
155
  */
109
- declare function createTool<TInitialContext extends Context, TInputSchema extends AnySchema, TOutputSchema extends AnySchema, TErrorMap extends ErrorMap, TMeta extends AiSdkToolMeta>(procedure: Procedure<TInitialContext, any, TInputSchema, TOutputSchema, TErrorMap, TMeta>, ...rest: MaybeOptionalOptions<SetOptional<Tool<InferSchemaOutput<TInputSchema>, InferSchemaInput<TOutputSchema>>, 'inputSchema' | 'outputSchema' | 'execute'> & CreateProcedureClientOptions<TInitialContext, TOutputSchema, TErrorMap, TMeta, Record<never, never>> & Omit<ClientOptions<Record<never, never>>, 'context'>>): Tool<InferSchemaOutput<TInputSchema>, InferSchemaInput<TOutputSchema>>;
156
+ declare function createToolFactory<TInitialContext extends Context = object>(...rest: MaybeOptionalOptions<CreateToolFactoryOptions<TInitialContext>>): ToolFactory<TInitialContext>;
110
157
 
111
- export { AI_SDK_TOOL_META_SYMBOL, CreateToolError, createTool, implementTool };
112
- export type { AiSdkToolMeta };
158
+ export { aiSdkTool, createToolFactory, getAiSdkToolMeta, implementToolFactory };
159
+ export type { AiSdkToolFunction, AiSdkToolMeta, AiSdkToolMetaPlugin, CreateToolFactoryOptions, FunctionTool, ImplementToolFactoryOptions, ToolFactory, ToolImplementer, ToolOutput };
package/dist/index.mjs CHANGED
@@ -1,43 +1,142 @@
1
+ import { getAsyncIteratorObjectSchemaDetails } from '@orpc/contract';
2
+ import { combineJsonSchemasWithComposition } from '@orpc/json-schema';
1
3
  import { Procedure, call } from '@orpc/server';
2
- import { resolveMaybeOptionalOptions } from '@orpc/shared';
3
- export { AsyncIteratorClass, asyncIteratorToStream as eventIteratorToStream, asyncIteratorToUnproxiedDataStream as eventIteratorToUnproxiedDataStream, streamToAsyncIteratorClass as streamToEventIterator } from '@orpc/shared';
4
+ import { resolveMaybeOptionalOptions, toArray, isAsyncIteratorObject, ORPC_NAME } from '@orpc/shared';
4
5
  import { tool } from 'ai';
5
6
 
6
- const AI_SDK_TOOL_META_SYMBOL = Symbol("ORPC_AI_SDK_TOOL_META");
7
- class CreateToolError extends Error {
7
+ const aiSdkTool = (incoming) => ({
8
+ name: "~ai-sdk/tool",
9
+ init(meta) {
10
+ const existing = meta["~ai-sdk/tool"];
11
+ return {
12
+ ...meta,
13
+ "~ai-sdk/tool": { ...existing, ...incoming }
14
+ };
15
+ }
16
+ });
17
+ function getAiSdkToolMeta(procedureOrLazy) {
18
+ return procedureOrLazy["~orpc"].meta["~ai-sdk/tool"];
8
19
  }
9
- function implementTool(contract, ...rest) {
10
- if (contract["~orpc"].inputSchema === void 0) {
11
- throw new CreateToolError("Cannot implement tool from a contract procedure without input schema.");
20
+
21
+ const ANY_SCHEMA = {
22
+ "~standard": {
23
+ vendor: ORPC_NAME,
24
+ version: 1,
25
+ validate: (value) => ({ value }),
26
+ jsonSchema: {
27
+ input: () => ({}),
28
+ output: () => ({})
29
+ }
12
30
  }
13
- const options = resolveMaybeOptionalOptions(rest);
14
- return tool({
15
- inputSchema: contract["~orpc"].inputSchema,
16
- outputSchema: contract["~orpc"].outputSchema,
17
- description: contract["~orpc"].route.summary ?? contract["~orpc"].route.description,
18
- ...contract["~orpc"].meta[AI_SDK_TOOL_META_SYMBOL],
19
- ...options
31
+ };
32
+ function combineJsonSchemas(jsonSchemas) {
33
+ let $schema;
34
+ const branches = jsonSchemas.map((jsonSchema) => {
35
+ if ("$schema" in jsonSchema) {
36
+ const { $schema: branch$schema, ...rest } = jsonSchema;
37
+ $schema ??= branch$schema;
38
+ return rest;
39
+ }
40
+ return jsonSchema;
20
41
  });
42
+ const combined = combineJsonSchemasWithComposition("allOf", branches);
43
+ return { $schema, ...combined };
44
+ }
45
+ function combineSchemas(schemas) {
46
+ if (schemas.length === 0) {
47
+ return void 0;
48
+ }
49
+ if (schemas.length === 1) {
50
+ return schemas[0];
51
+ }
52
+ const standardJsonSchemas = schemas.filter((schema) => schema["~standard"].jsonSchema);
53
+ const jsonSchema = standardJsonSchemas.length ? {
54
+ input: (options) => combineJsonSchemas(standardJsonSchemas.map((schema) => schema["~standard"].jsonSchema.input(options))),
55
+ output: (options) => combineJsonSchemas(standardJsonSchemas.map((schema) => schema["~standard"].jsonSchema.output(options)))
56
+ } : void 0;
57
+ return {
58
+ "~standard": {
59
+ vendor: ORPC_NAME,
60
+ version: 1,
61
+ async validate(value) {
62
+ let current = value;
63
+ for (const schema of schemas) {
64
+ const result = await schema["~standard"].validate(current);
65
+ if (result.issues) {
66
+ return result;
67
+ }
68
+ current = result.value;
69
+ }
70
+ return { value: current };
71
+ },
72
+ jsonSchema
73
+ }
74
+ };
21
75
  }
22
- function createTool(procedure, ...rest) {
76
+ function getIteratorYieldSchemas(schemas) {
77
+ if (schemas.length === 0) {
78
+ return void 0;
79
+ }
80
+ const yieldSchemas = [];
81
+ for (const schema of schemas) {
82
+ const details = getAsyncIteratorObjectSchemaDetails(schema);
83
+ if (!details) {
84
+ return void 0;
85
+ }
86
+ yieldSchemas.push(details.yieldSchema);
87
+ }
88
+ return yieldSchemas;
89
+ }
90
+ function combineOutputSchemas(outputSchemas) {
91
+ const yieldSchemas = getIteratorYieldSchemas(outputSchemas);
92
+ return combineSchemas([...yieldSchemas ?? outputSchemas].reverse());
93
+ }
94
+ function implementToolFactory(_options = {}) {
95
+ const factory = (contract, ...rest) => {
96
+ const toolOptions = resolveMaybeOptionalOptions(rest);
97
+ const inputSchemas = toArray(contract["~orpc"].inputSchemas);
98
+ const outputSchemas = toArray(contract["~orpc"].outputSchemas);
99
+ const defaults = getAiSdkToolMeta(contract);
100
+ return tool({
101
+ ...defaults,
102
+ ...toolOptions,
103
+ inputSchema: combineSchemas(inputSchemas) ?? ANY_SCHEMA,
104
+ outputSchema: combineOutputSchemas(outputSchemas)
105
+ });
106
+ };
107
+ return factory;
108
+ }
109
+ function createToolFactory(...rest) {
23
110
  const options = resolveMaybeOptionalOptions(rest);
24
- return implementTool(procedure, {
25
- execute: ((input, callingOptions) => {
26
- const disabledValidation = new Procedure({
27
- ...procedure["~orpc"],
28
- inputValidationIndex: Number.NaN,
29
- // disable input validation
30
- outputValidationIndex: Number.NaN
31
- // disable output validation
32
- });
33
- return call(
34
- disabledValidation,
35
- input,
36
- { signal: callingOptions.abortSignal, ...options }
37
- );
38
- }),
39
- ...options
40
- });
111
+ const implementTool = implementToolFactory(options);
112
+ const factory = (procedure, ...rest2) => {
113
+ const toolOptions = resolveMaybeOptionalOptions(rest2);
114
+ const disabledValidation = new Procedure({
115
+ ...procedure["~orpc"],
116
+ disableInputValidation: true,
117
+ disableOutputValidation: true
118
+ });
119
+ const isIteratorOutput = getIteratorYieldSchemas(toArray(procedure["~orpc"].outputSchemas)) !== void 0;
120
+ return implementTool(procedure, {
121
+ ...toolOptions,
122
+ /**
123
+ * For `asyncIteratorObject` outputs, the tool streams each event as a
124
+ * [preliminary result](https://ai-sdk.dev/docs/ai-sdk-core/tools-and-tool-calling#preliminary-tool-results),
125
+ * and the last event becomes the final result.
126
+ */
127
+ execute: isIteratorOutput ? async function* (input, callingOptions) {
128
+ const output = await call(disabledValidation, input, { signal: callingOptions.abortSignal, ...options });
129
+ if (!isAsyncIteratorObject(output)) {
130
+ yield output;
131
+ return;
132
+ }
133
+ yield* output;
134
+ } : (input, callingOptions) => {
135
+ return call(disabledValidation, input, { signal: callingOptions.abortSignal, ...options });
136
+ }
137
+ });
138
+ };
139
+ return factory;
41
140
  }
42
141
 
43
- export { AI_SDK_TOOL_META_SYMBOL, CreateToolError, createTool, implementTool };
142
+ export { aiSdkTool, createToolFactory, getAiSdkToolMeta, implementToolFactory };
package/package.json CHANGED
@@ -1,7 +1,7 @@
1
1
  {
2
2
  "name": "@orpc/ai-sdk",
3
3
  "type": "module",
4
- "version": "1.14.9",
4
+ "version": "1.14.11",
5
5
  "license": "MIT",
6
6
  "homepage": "https://orpc.dev",
7
7
  "repository": {
@@ -10,11 +10,12 @@
10
10
  "directory": "packages/ai-sdk"
11
11
  },
12
12
  "keywords": [
13
- "ai-sdk",
14
- "orpc"
13
+ "orpc",
14
+ "ai-sdk"
15
15
  ],
16
16
  "sideEffects": false,
17
17
  "exports": {
18
+ "./package.json": "./package.json",
18
19
  ".": {
19
20
  "types": "./dist/index.d.mts",
20
21
  "import": "./dist/index.mjs",
@@ -25,21 +26,21 @@
25
26
  "dist"
26
27
  ],
27
28
  "peerDependencies": {
28
- "ai": ">=5.0.76"
29
+ "ai": ">=7.0.31"
29
30
  },
30
31
  "dependencies": {
31
- "@orpc/client": "1.14.9",
32
- "@orpc/contract": "1.14.9",
33
- "@orpc/server": "1.14.9",
34
- "@orpc/shared": "1.14.9"
32
+ "@orpc/client": "1.14.11",
33
+ "@orpc/json-schema": "1.14.11",
34
+ "@orpc/shared": "1.14.11",
35
+ "@orpc/server": "1.14.11",
36
+ "@orpc/contract": "1.14.11"
35
37
  },
36
38
  "devDependencies": {
37
- "ai": "^6.0.116",
38
- "zod": "^4.3.6"
39
+ "ai": "^7.0.37",
40
+ "zod": "^4.4.3"
39
41
  },
40
42
  "scripts": {
41
43
  "build": "unbuild",
42
- "build:watch": "pnpm run build --watch",
43
44
  "type:check": "tsc -b"
44
45
  }
45
46
  }