@allem-sdk/ai 0.1.0
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/LICENSE +21 -0
- package/README.md +105 -0
- package/dist/client.d.mts +49 -0
- package/dist/client.d.ts +49 -0
- package/dist/client.js +103 -0
- package/dist/client.mjs +76 -0
- package/dist/index.d.mts +84 -0
- package/dist/index.d.ts +84 -0
- package/dist/index.js +138 -0
- package/dist/index.mjs +112 -0
- package/dist/server.d.mts +38 -0
- package/dist/server.d.ts +38 -0
- package/dist/server.js +63 -0
- package/dist/server.mjs +39 -0
- package/package.json +86 -0
package/LICENSE
ADDED
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
MIT License
|
|
2
|
+
|
|
3
|
+
Copyright (c) 2026 Ahmed Allem
|
|
4
|
+
|
|
5
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
6
|
+
of this software and associated documentation files (the "Software"), to deal
|
|
7
|
+
in the Software without restriction, including without limitation the rights
|
|
8
|
+
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
9
|
+
copies of the Software, and to permit persons to whom the Software is
|
|
10
|
+
furnished to do so, subject to the following conditions:
|
|
11
|
+
|
|
12
|
+
The above copyright notice and this permission notice shall be included in all
|
|
13
|
+
copies or substantial portions of the Software.
|
|
14
|
+
|
|
15
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
16
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
17
|
+
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
18
|
+
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
19
|
+
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
20
|
+
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
21
|
+
SOFTWARE.
|
package/README.md
ADDED
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
<p align="center">
|
|
2
|
+
<img src="https://raw.githubusercontent.com/kingofmit/allem-sdk/main/.github/AllemSDK.png" alt="Allem SDK" />
|
|
3
|
+
</p>
|
|
4
|
+
|
|
5
|
+
<p align="center">
|
|
6
|
+
<a href="https://github.com/kingofmit/allem-sdk/blob/main/LICENSE"><img src="https://img.shields.io/badge/license-MIT-blue.svg" alt="License" /></a>
|
|
7
|
+
<img src="https://img.shields.io/badge/react-18+-61dafb" alt="React 18+" />
|
|
8
|
+
<img src="https://img.shields.io/badge/typescript-strict-blue" alt="TypeScript" />
|
|
9
|
+
<img src="https://img.shields.io/badge/AI_SDK-v6-black" alt="AI SDK v6" />
|
|
10
|
+
</p>
|
|
11
|
+
|
|
12
|
+
# @allem-sdk/ai
|
|
13
|
+
|
|
14
|
+
AI hooks for React built on [Vercel AI SDK v6](https://sdk.vercel.ai). Multi-provider support (Google, Anthropic, OpenAI) with a unified API.
|
|
15
|
+
|
|
16
|
+
## Installation
|
|
17
|
+
|
|
18
|
+
```bash
|
|
19
|
+
npm install @allem-sdk/ai ai @ai-sdk/react
|
|
20
|
+
|
|
21
|
+
# Install at least one provider
|
|
22
|
+
npm install @ai-sdk/google # Google Gemini
|
|
23
|
+
npm install @ai-sdk/anthropic # Anthropic Claude
|
|
24
|
+
npm install @ai-sdk/openai # OpenAI GPT
|
|
25
|
+
```
|
|
26
|
+
|
|
27
|
+
## Client Usage
|
|
28
|
+
|
|
29
|
+
```tsx
|
|
30
|
+
import { useState } from "react";
|
|
31
|
+
import { AllemAIProvider, useAllemChat } from "@allem-sdk/ai";
|
|
32
|
+
|
|
33
|
+
function App() {
|
|
34
|
+
return (
|
|
35
|
+
<AllemAIProvider api="/api/chat" provider="google">
|
|
36
|
+
<Chat />
|
|
37
|
+
</AllemAIProvider>
|
|
38
|
+
);
|
|
39
|
+
}
|
|
40
|
+
|
|
41
|
+
function Chat() {
|
|
42
|
+
const [input, setInput] = useState("");
|
|
43
|
+
const { messages, sendMessage, status } = useAllemChat();
|
|
44
|
+
const isLoading = status === "submitted" || status === "streaming";
|
|
45
|
+
|
|
46
|
+
return (
|
|
47
|
+
<div>
|
|
48
|
+
{messages.map((m) => (
|
|
49
|
+
<div key={m.id}>
|
|
50
|
+
{m.parts?.filter((p) => p.type === "text").map((p) => p.text).join("")}
|
|
51
|
+
</div>
|
|
52
|
+
))}
|
|
53
|
+
<form onSubmit={async (e) => {
|
|
54
|
+
e.preventDefault();
|
|
55
|
+
const text = input;
|
|
56
|
+
setInput("");
|
|
57
|
+
await sendMessage({ text });
|
|
58
|
+
}}>
|
|
59
|
+
<input value={input} onChange={(e) => setInput(e.target.value)} disabled={isLoading} />
|
|
60
|
+
<button type="submit">Send</button>
|
|
61
|
+
</form>
|
|
62
|
+
</div>
|
|
63
|
+
);
|
|
64
|
+
}
|
|
65
|
+
```
|
|
66
|
+
|
|
67
|
+
## Server Usage
|
|
68
|
+
|
|
69
|
+
```ts
|
|
70
|
+
// app/api/chat/route.ts (Next.js)
|
|
71
|
+
import { createAllemChatHandler } from "@allem-sdk/ai";
|
|
72
|
+
import { google } from "@ai-sdk/google";
|
|
73
|
+
import { anthropic } from "@ai-sdk/anthropic";
|
|
74
|
+
|
|
75
|
+
export const POST = createAllemChatHandler({
|
|
76
|
+
providers: {
|
|
77
|
+
google: (model) => google(model ?? "gemini-2.0-flash"),
|
|
78
|
+
anthropic: (model) => anthropic(model ?? "claude-sonnet-4-20250514"),
|
|
79
|
+
},
|
|
80
|
+
defaultProvider: "google",
|
|
81
|
+
});
|
|
82
|
+
```
|
|
83
|
+
|
|
84
|
+
## Exports
|
|
85
|
+
|
|
86
|
+
| Export | Type | Description |
|
|
87
|
+
|--------|------|-------------|
|
|
88
|
+
| `useAllemChat` | Hook | Chat hook with streaming, provider context, and message history |
|
|
89
|
+
| `useAllemCompletion` | Hook | Text completion hook for single-turn prompts |
|
|
90
|
+
| `AllemAIProvider` | Component | React context provider for default API and provider config |
|
|
91
|
+
| `createAllemChatHandler` | Server | Server-side API route handler with multi-provider routing |
|
|
92
|
+
|
|
93
|
+
## Part of [Allem SDK](https://github.com/kingofmit/allem-sdk)
|
|
94
|
+
|
|
95
|
+
This package can be used standalone or as part of the full SDK. Install `allem-sdk` to get all packages in one install.
|
|
96
|
+
|
|
97
|
+
## Support
|
|
98
|
+
|
|
99
|
+
If you find Allem SDK useful, consider supporting its development:
|
|
100
|
+
|
|
101
|
+
<a href="https://buymeacoffee.com/kingofmit" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/v2/default-yellow.png" alt="Buy Me A Coffee" height="50" /></a>
|
|
102
|
+
|
|
103
|
+
## License
|
|
104
|
+
|
|
105
|
+
[MIT](https://github.com/kingofmit/allem-sdk/blob/main/LICENSE) - [Ahmed Allem](https://kingallem.com)
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { UseChatHelpers, UseCompletionOptions, UseCompletionHelpers } from '@ai-sdk/react';
|
|
2
|
+
export { UseChatHelpers, UseCompletionHelpers } from '@ai-sdk/react';
|
|
3
|
+
import { ChatInit, UIMessage } from 'ai';
|
|
4
|
+
export { UIMessage } from 'ai';
|
|
5
|
+
import * as react_jsx_runtime from 'react/jsx-runtime';
|
|
6
|
+
import { ReactNode } from 'react';
|
|
7
|
+
|
|
8
|
+
type AllemProvider = "google" | "anthropic" | "openai";
|
|
9
|
+
interface UseAllemChatOptions extends Omit<ChatInit<UIMessage>, "transport"> {
|
|
10
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
11
|
+
api?: string;
|
|
12
|
+
/** Model identifier sent in the request body (e.g. "gemini-2.0-flash"). */
|
|
13
|
+
model?: string;
|
|
14
|
+
/** Provider name sent in the request body. */
|
|
15
|
+
provider?: AllemProvider;
|
|
16
|
+
/** System prompt sent in the request body. */
|
|
17
|
+
systemPrompt?: string;
|
|
18
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
19
|
+
headers?: Record<string, string>;
|
|
20
|
+
/** Custom throttle wait in ms for updates. */
|
|
21
|
+
experimental_throttle?: number;
|
|
22
|
+
}
|
|
23
|
+
declare function useAllemChat(options?: UseAllemChatOptions): UseChatHelpers<UIMessage>;
|
|
24
|
+
|
|
25
|
+
interface UseAllemCompletionOptions extends UseCompletionOptions {
|
|
26
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
27
|
+
api?: string;
|
|
28
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
29
|
+
headers?: Record<string, string>;
|
|
30
|
+
}
|
|
31
|
+
declare function useAllemCompletion(options?: UseAllemCompletionOptions): UseCompletionHelpers;
|
|
32
|
+
|
|
33
|
+
interface AllemAIConfig {
|
|
34
|
+
/** API endpoint for chat. Default: "/api/chat" */
|
|
35
|
+
api: string;
|
|
36
|
+
/** Default model identifier (e.g. "gemini-2.0-flash", "claude-sonnet-4-20250514"). */
|
|
37
|
+
model?: string;
|
|
38
|
+
/** Default provider. */
|
|
39
|
+
provider?: "google" | "anthropic" | "openai";
|
|
40
|
+
/** Extra headers sent with every request. */
|
|
41
|
+
headers?: Record<string, string>;
|
|
42
|
+
}
|
|
43
|
+
interface AllemAIProviderProps extends Partial<AllemAIConfig> {
|
|
44
|
+
children: ReactNode;
|
|
45
|
+
}
|
|
46
|
+
declare function AllemAIProvider({ children, ...config }: AllemAIProviderProps): react_jsx_runtime.JSX.Element;
|
|
47
|
+
declare function useAllemAIConfig(): AllemAIConfig;
|
|
48
|
+
|
|
49
|
+
export { type AllemAIConfig, AllemAIProvider, type AllemAIProviderProps, type AllemProvider, type UseAllemChatOptions, type UseAllemCompletionOptions, useAllemAIConfig, useAllemChat, useAllemCompletion };
|
package/dist/client.d.ts
ADDED
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { UseChatHelpers, UseCompletionOptions, UseCompletionHelpers } from '@ai-sdk/react';
|
|
2
|
+
export { UseChatHelpers, UseCompletionHelpers } from '@ai-sdk/react';
|
|
3
|
+
import { ChatInit, UIMessage } from 'ai';
|
|
4
|
+
export { UIMessage } from 'ai';
|
|
5
|
+
import * as react_jsx_runtime from 'react/jsx-runtime';
|
|
6
|
+
import { ReactNode } from 'react';
|
|
7
|
+
|
|
8
|
+
type AllemProvider = "google" | "anthropic" | "openai";
|
|
9
|
+
interface UseAllemChatOptions extends Omit<ChatInit<UIMessage>, "transport"> {
|
|
10
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
11
|
+
api?: string;
|
|
12
|
+
/** Model identifier sent in the request body (e.g. "gemini-2.0-flash"). */
|
|
13
|
+
model?: string;
|
|
14
|
+
/** Provider name sent in the request body. */
|
|
15
|
+
provider?: AllemProvider;
|
|
16
|
+
/** System prompt sent in the request body. */
|
|
17
|
+
systemPrompt?: string;
|
|
18
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
19
|
+
headers?: Record<string, string>;
|
|
20
|
+
/** Custom throttle wait in ms for updates. */
|
|
21
|
+
experimental_throttle?: number;
|
|
22
|
+
}
|
|
23
|
+
declare function useAllemChat(options?: UseAllemChatOptions): UseChatHelpers<UIMessage>;
|
|
24
|
+
|
|
25
|
+
interface UseAllemCompletionOptions extends UseCompletionOptions {
|
|
26
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
27
|
+
api?: string;
|
|
28
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
29
|
+
headers?: Record<string, string>;
|
|
30
|
+
}
|
|
31
|
+
declare function useAllemCompletion(options?: UseAllemCompletionOptions): UseCompletionHelpers;
|
|
32
|
+
|
|
33
|
+
interface AllemAIConfig {
|
|
34
|
+
/** API endpoint for chat. Default: "/api/chat" */
|
|
35
|
+
api: string;
|
|
36
|
+
/** Default model identifier (e.g. "gemini-2.0-flash", "claude-sonnet-4-20250514"). */
|
|
37
|
+
model?: string;
|
|
38
|
+
/** Default provider. */
|
|
39
|
+
provider?: "google" | "anthropic" | "openai";
|
|
40
|
+
/** Extra headers sent with every request. */
|
|
41
|
+
headers?: Record<string, string>;
|
|
42
|
+
}
|
|
43
|
+
interface AllemAIProviderProps extends Partial<AllemAIConfig> {
|
|
44
|
+
children: ReactNode;
|
|
45
|
+
}
|
|
46
|
+
declare function AllemAIProvider({ children, ...config }: AllemAIProviderProps): react_jsx_runtime.JSX.Element;
|
|
47
|
+
declare function useAllemAIConfig(): AllemAIConfig;
|
|
48
|
+
|
|
49
|
+
export { type AllemAIConfig, AllemAIProvider, type AllemAIProviderProps, type AllemProvider, type UseAllemChatOptions, type UseAllemCompletionOptions, useAllemAIConfig, useAllemChat, useAllemCompletion };
|
package/dist/client.js
ADDED
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
"use strict";
|
|
3
|
+
var __defProp = Object.defineProperty;
|
|
4
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
5
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
6
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
7
|
+
var __export = (target, all) => {
|
|
8
|
+
for (var name in all)
|
|
9
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
10
|
+
};
|
|
11
|
+
var __copyProps = (to, from, except, desc) => {
|
|
12
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
13
|
+
for (let key of __getOwnPropNames(from))
|
|
14
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
15
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
16
|
+
}
|
|
17
|
+
return to;
|
|
18
|
+
};
|
|
19
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
20
|
+
|
|
21
|
+
// src/client.ts
|
|
22
|
+
var client_exports = {};
|
|
23
|
+
__export(client_exports, {
|
|
24
|
+
AllemAIProvider: () => AllemAIProvider,
|
|
25
|
+
useAllemAIConfig: () => useAllemAIConfig,
|
|
26
|
+
useAllemChat: () => useAllemChat,
|
|
27
|
+
useAllemCompletion: () => useAllemCompletion
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(client_exports);
|
|
30
|
+
|
|
31
|
+
// src/useAllemChat.ts
|
|
32
|
+
var import_react2 = require("@ai-sdk/react");
|
|
33
|
+
var import_ai = require("ai");
|
|
34
|
+
|
|
35
|
+
// src/AllemAIProvider.tsx
|
|
36
|
+
var import_react = require("react");
|
|
37
|
+
var import_jsx_runtime = require("react/jsx-runtime");
|
|
38
|
+
var defaults = {
|
|
39
|
+
api: "/api/chat",
|
|
40
|
+
provider: "google"
|
|
41
|
+
};
|
|
42
|
+
var AllemAIContext = (0, import_react.createContext)(defaults);
|
|
43
|
+
function AllemAIProvider({
|
|
44
|
+
children,
|
|
45
|
+
...config
|
|
46
|
+
}) {
|
|
47
|
+
const value = { ...defaults, ...config };
|
|
48
|
+
return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(AllemAIContext.Provider, { value, children });
|
|
49
|
+
}
|
|
50
|
+
function useAllemAIConfig() {
|
|
51
|
+
return (0, import_react.useContext)(AllemAIContext);
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
// src/useAllemChat.ts
|
|
55
|
+
function useAllemChat(options = {}) {
|
|
56
|
+
const config = useAllemAIConfig();
|
|
57
|
+
const {
|
|
58
|
+
api,
|
|
59
|
+
model,
|
|
60
|
+
provider,
|
|
61
|
+
systemPrompt,
|
|
62
|
+
headers,
|
|
63
|
+
experimental_throttle,
|
|
64
|
+
...restOptions
|
|
65
|
+
} = options;
|
|
66
|
+
const resolvedApi = api ?? config.api;
|
|
67
|
+
const resolvedModel = model ?? config.model;
|
|
68
|
+
const resolvedProvider = provider ?? config.provider;
|
|
69
|
+
const resolvedHeaders = { ...config.headers, ...headers };
|
|
70
|
+
const transport = new import_ai.DefaultChatTransport({
|
|
71
|
+
api: resolvedApi,
|
|
72
|
+
headers: resolvedHeaders,
|
|
73
|
+
body: {
|
|
74
|
+
...resolvedModel && { model: resolvedModel },
|
|
75
|
+
...resolvedProvider && { provider: resolvedProvider },
|
|
76
|
+
...systemPrompt && { systemPrompt }
|
|
77
|
+
}
|
|
78
|
+
});
|
|
79
|
+
return (0, import_react2.useChat)({
|
|
80
|
+
transport,
|
|
81
|
+
experimental_throttle,
|
|
82
|
+
...restOptions
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
// src/useAllemCompletion.ts
|
|
87
|
+
var import_react3 = require("@ai-sdk/react");
|
|
88
|
+
function useAllemCompletion(options = {}) {
|
|
89
|
+
const config = useAllemAIConfig();
|
|
90
|
+
const { api, headers, ...restOptions } = options;
|
|
91
|
+
return (0, import_react3.useCompletion)({
|
|
92
|
+
api: api ?? config.api,
|
|
93
|
+
headers: { ...config.headers, ...headers },
|
|
94
|
+
...restOptions
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
98
|
+
0 && (module.exports = {
|
|
99
|
+
AllemAIProvider,
|
|
100
|
+
useAllemAIConfig,
|
|
101
|
+
useAllemChat,
|
|
102
|
+
useAllemCompletion
|
|
103
|
+
});
|
package/dist/client.mjs
ADDED
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
|
|
3
|
+
// src/useAllemChat.ts
|
|
4
|
+
import { useChat } from "@ai-sdk/react";
|
|
5
|
+
import { DefaultChatTransport } from "ai";
|
|
6
|
+
|
|
7
|
+
// src/AllemAIProvider.tsx
|
|
8
|
+
import { createContext, useContext } from "react";
|
|
9
|
+
import { jsx } from "react/jsx-runtime";
|
|
10
|
+
var defaults = {
|
|
11
|
+
api: "/api/chat",
|
|
12
|
+
provider: "google"
|
|
13
|
+
};
|
|
14
|
+
var AllemAIContext = createContext(defaults);
|
|
15
|
+
function AllemAIProvider({
|
|
16
|
+
children,
|
|
17
|
+
...config
|
|
18
|
+
}) {
|
|
19
|
+
const value = { ...defaults, ...config };
|
|
20
|
+
return /* @__PURE__ */ jsx(AllemAIContext.Provider, { value, children });
|
|
21
|
+
}
|
|
22
|
+
function useAllemAIConfig() {
|
|
23
|
+
return useContext(AllemAIContext);
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
// src/useAllemChat.ts
|
|
27
|
+
function useAllemChat(options = {}) {
|
|
28
|
+
const config = useAllemAIConfig();
|
|
29
|
+
const {
|
|
30
|
+
api,
|
|
31
|
+
model,
|
|
32
|
+
provider,
|
|
33
|
+
systemPrompt,
|
|
34
|
+
headers,
|
|
35
|
+
experimental_throttle,
|
|
36
|
+
...restOptions
|
|
37
|
+
} = options;
|
|
38
|
+
const resolvedApi = api ?? config.api;
|
|
39
|
+
const resolvedModel = model ?? config.model;
|
|
40
|
+
const resolvedProvider = provider ?? config.provider;
|
|
41
|
+
const resolvedHeaders = { ...config.headers, ...headers };
|
|
42
|
+
const transport = new DefaultChatTransport({
|
|
43
|
+
api: resolvedApi,
|
|
44
|
+
headers: resolvedHeaders,
|
|
45
|
+
body: {
|
|
46
|
+
...resolvedModel && { model: resolvedModel },
|
|
47
|
+
...resolvedProvider && { provider: resolvedProvider },
|
|
48
|
+
...systemPrompt && { systemPrompt }
|
|
49
|
+
}
|
|
50
|
+
});
|
|
51
|
+
return useChat({
|
|
52
|
+
transport,
|
|
53
|
+
experimental_throttle,
|
|
54
|
+
...restOptions
|
|
55
|
+
});
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
// src/useAllemCompletion.ts
|
|
59
|
+
import {
|
|
60
|
+
useCompletion
|
|
61
|
+
} from "@ai-sdk/react";
|
|
62
|
+
function useAllemCompletion(options = {}) {
|
|
63
|
+
const config = useAllemAIConfig();
|
|
64
|
+
const { api, headers, ...restOptions } = options;
|
|
65
|
+
return useCompletion({
|
|
66
|
+
api: api ?? config.api,
|
|
67
|
+
headers: { ...config.headers, ...headers },
|
|
68
|
+
...restOptions
|
|
69
|
+
});
|
|
70
|
+
}
|
|
71
|
+
export {
|
|
72
|
+
AllemAIProvider,
|
|
73
|
+
useAllemAIConfig,
|
|
74
|
+
useAllemChat,
|
|
75
|
+
useAllemCompletion
|
|
76
|
+
};
|
package/dist/index.d.mts
ADDED
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import { UseChatHelpers, UseCompletionOptions, UseCompletionHelpers } from '@ai-sdk/react';
|
|
2
|
+
export { UseChatHelpers, UseCompletionHelpers } from '@ai-sdk/react';
|
|
3
|
+
import { ChatInit, UIMessage, LanguageModel, ToolSet } from 'ai';
|
|
4
|
+
export { UIMessage } from 'ai';
|
|
5
|
+
import * as react_jsx_runtime from 'react/jsx-runtime';
|
|
6
|
+
import { ReactNode } from 'react';
|
|
7
|
+
|
|
8
|
+
type AllemProvider = "google" | "anthropic" | "openai";
|
|
9
|
+
interface UseAllemChatOptions extends Omit<ChatInit<UIMessage>, "transport"> {
|
|
10
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
11
|
+
api?: string;
|
|
12
|
+
/** Model identifier sent in the request body (e.g. "gemini-2.0-flash"). */
|
|
13
|
+
model?: string;
|
|
14
|
+
/** Provider name sent in the request body. */
|
|
15
|
+
provider?: AllemProvider;
|
|
16
|
+
/** System prompt sent in the request body. */
|
|
17
|
+
systemPrompt?: string;
|
|
18
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
19
|
+
headers?: Record<string, string>;
|
|
20
|
+
/** Custom throttle wait in ms for updates. */
|
|
21
|
+
experimental_throttle?: number;
|
|
22
|
+
}
|
|
23
|
+
declare function useAllemChat(options?: UseAllemChatOptions): UseChatHelpers<UIMessage>;
|
|
24
|
+
|
|
25
|
+
interface UseAllemCompletionOptions extends UseCompletionOptions {
|
|
26
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
27
|
+
api?: string;
|
|
28
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
29
|
+
headers?: Record<string, string>;
|
|
30
|
+
}
|
|
31
|
+
declare function useAllemCompletion(options?: UseAllemCompletionOptions): UseCompletionHelpers;
|
|
32
|
+
|
|
33
|
+
interface AllemAIConfig {
|
|
34
|
+
/** API endpoint for chat. Default: "/api/chat" */
|
|
35
|
+
api: string;
|
|
36
|
+
/** Default model identifier (e.g. "gemini-2.0-flash", "claude-sonnet-4-20250514"). */
|
|
37
|
+
model?: string;
|
|
38
|
+
/** Default provider. */
|
|
39
|
+
provider?: "google" | "anthropic" | "openai";
|
|
40
|
+
/** Extra headers sent with every request. */
|
|
41
|
+
headers?: Record<string, string>;
|
|
42
|
+
}
|
|
43
|
+
interface AllemAIProviderProps extends Partial<AllemAIConfig> {
|
|
44
|
+
children: ReactNode;
|
|
45
|
+
}
|
|
46
|
+
declare function AllemAIProvider({ children, ...config }: AllemAIProviderProps): react_jsx_runtime.JSX.Element;
|
|
47
|
+
declare function useAllemAIConfig(): AllemAIConfig;
|
|
48
|
+
|
|
49
|
+
type AllemProviderName = "google" | "anthropic" | "openai";
|
|
50
|
+
interface AllemChatHandlerConfig {
|
|
51
|
+
/** Map of provider name to a function that returns a LanguageModel for a given model ID. */
|
|
52
|
+
providers: Partial<Record<AllemProviderName, (modelId?: string) => LanguageModel>>;
|
|
53
|
+
/** Default provider when none is specified in the request. Default: "google". */
|
|
54
|
+
defaultProvider?: AllemProviderName;
|
|
55
|
+
/** Default model ID when none is specified in the request. */
|
|
56
|
+
defaultModel?: string;
|
|
57
|
+
/** Default system prompt. */
|
|
58
|
+
systemPrompt?: string;
|
|
59
|
+
/** Tools available to the model. */
|
|
60
|
+
tools?: ToolSet;
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Creates an API route handler for AI chat.
|
|
64
|
+
* Works with Next.js App Router, Remix, and any framework that uses
|
|
65
|
+
* the Web Request/Response API.
|
|
66
|
+
*
|
|
67
|
+
* @example
|
|
68
|
+
* ```ts
|
|
69
|
+
* // app/api/chat/route.ts (Next.js)
|
|
70
|
+
* import { createAllemChatHandler } from "@allem-sdk/ai";
|
|
71
|
+
* import { google } from "@ai-sdk/google";
|
|
72
|
+
* import { anthropic } from "@ai-sdk/anthropic";
|
|
73
|
+
*
|
|
74
|
+
* export const POST = createAllemChatHandler({
|
|
75
|
+
* providers: {
|
|
76
|
+
* google: (model) => google(model ?? "gemini-2.0-flash"),
|
|
77
|
+
* anthropic: (model) => anthropic(model ?? "claude-sonnet-4-20250514"),
|
|
78
|
+
* },
|
|
79
|
+
* });
|
|
80
|
+
* ```
|
|
81
|
+
*/
|
|
82
|
+
declare function createAllemChatHandler(config: AllemChatHandlerConfig): (req: Request) => Promise<Response>;
|
|
83
|
+
|
|
84
|
+
export { type AllemAIConfig, AllemAIProvider, type AllemAIProviderProps, type AllemChatHandlerConfig, type AllemProvider, type AllemProviderName, type UseAllemChatOptions, type UseAllemCompletionOptions, createAllemChatHandler, useAllemAIConfig, useAllemChat, useAllemCompletion };
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import { UseChatHelpers, UseCompletionOptions, UseCompletionHelpers } from '@ai-sdk/react';
|
|
2
|
+
export { UseChatHelpers, UseCompletionHelpers } from '@ai-sdk/react';
|
|
3
|
+
import { ChatInit, UIMessage, LanguageModel, ToolSet } from 'ai';
|
|
4
|
+
export { UIMessage } from 'ai';
|
|
5
|
+
import * as react_jsx_runtime from 'react/jsx-runtime';
|
|
6
|
+
import { ReactNode } from 'react';
|
|
7
|
+
|
|
8
|
+
type AllemProvider = "google" | "anthropic" | "openai";
|
|
9
|
+
interface UseAllemChatOptions extends Omit<ChatInit<UIMessage>, "transport"> {
|
|
10
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
11
|
+
api?: string;
|
|
12
|
+
/** Model identifier sent in the request body (e.g. "gemini-2.0-flash"). */
|
|
13
|
+
model?: string;
|
|
14
|
+
/** Provider name sent in the request body. */
|
|
15
|
+
provider?: AllemProvider;
|
|
16
|
+
/** System prompt sent in the request body. */
|
|
17
|
+
systemPrompt?: string;
|
|
18
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
19
|
+
headers?: Record<string, string>;
|
|
20
|
+
/** Custom throttle wait in ms for updates. */
|
|
21
|
+
experimental_throttle?: number;
|
|
22
|
+
}
|
|
23
|
+
declare function useAllemChat(options?: UseAllemChatOptions): UseChatHelpers<UIMessage>;
|
|
24
|
+
|
|
25
|
+
interface UseAllemCompletionOptions extends UseCompletionOptions {
|
|
26
|
+
/** Override the API endpoint from AllemAIProvider. */
|
|
27
|
+
api?: string;
|
|
28
|
+
/** Extra headers merged with AllemAIProvider headers. */
|
|
29
|
+
headers?: Record<string, string>;
|
|
30
|
+
}
|
|
31
|
+
declare function useAllemCompletion(options?: UseAllemCompletionOptions): UseCompletionHelpers;
|
|
32
|
+
|
|
33
|
+
interface AllemAIConfig {
|
|
34
|
+
/** API endpoint for chat. Default: "/api/chat" */
|
|
35
|
+
api: string;
|
|
36
|
+
/** Default model identifier (e.g. "gemini-2.0-flash", "claude-sonnet-4-20250514"). */
|
|
37
|
+
model?: string;
|
|
38
|
+
/** Default provider. */
|
|
39
|
+
provider?: "google" | "anthropic" | "openai";
|
|
40
|
+
/** Extra headers sent with every request. */
|
|
41
|
+
headers?: Record<string, string>;
|
|
42
|
+
}
|
|
43
|
+
interface AllemAIProviderProps extends Partial<AllemAIConfig> {
|
|
44
|
+
children: ReactNode;
|
|
45
|
+
}
|
|
46
|
+
declare function AllemAIProvider({ children, ...config }: AllemAIProviderProps): react_jsx_runtime.JSX.Element;
|
|
47
|
+
declare function useAllemAIConfig(): AllemAIConfig;
|
|
48
|
+
|
|
49
|
+
type AllemProviderName = "google" | "anthropic" | "openai";
|
|
50
|
+
interface AllemChatHandlerConfig {
|
|
51
|
+
/** Map of provider name to a function that returns a LanguageModel for a given model ID. */
|
|
52
|
+
providers: Partial<Record<AllemProviderName, (modelId?: string) => LanguageModel>>;
|
|
53
|
+
/** Default provider when none is specified in the request. Default: "google". */
|
|
54
|
+
defaultProvider?: AllemProviderName;
|
|
55
|
+
/** Default model ID when none is specified in the request. */
|
|
56
|
+
defaultModel?: string;
|
|
57
|
+
/** Default system prompt. */
|
|
58
|
+
systemPrompt?: string;
|
|
59
|
+
/** Tools available to the model. */
|
|
60
|
+
tools?: ToolSet;
|
|
61
|
+
}
|
|
62
|
+
/**
|
|
63
|
+
* Creates an API route handler for AI chat.
|
|
64
|
+
* Works with Next.js App Router, Remix, and any framework that uses
|
|
65
|
+
* the Web Request/Response API.
|
|
66
|
+
*
|
|
67
|
+
* @example
|
|
68
|
+
* ```ts
|
|
69
|
+
* // app/api/chat/route.ts (Next.js)
|
|
70
|
+
* import { createAllemChatHandler } from "@allem-sdk/ai";
|
|
71
|
+
* import { google } from "@ai-sdk/google";
|
|
72
|
+
* import { anthropic } from "@ai-sdk/anthropic";
|
|
73
|
+
*
|
|
74
|
+
* export const POST = createAllemChatHandler({
|
|
75
|
+
* providers: {
|
|
76
|
+
* google: (model) => google(model ?? "gemini-2.0-flash"),
|
|
77
|
+
* anthropic: (model) => anthropic(model ?? "claude-sonnet-4-20250514"),
|
|
78
|
+
* },
|
|
79
|
+
* });
|
|
80
|
+
* ```
|
|
81
|
+
*/
|
|
82
|
+
declare function createAllemChatHandler(config: AllemChatHandlerConfig): (req: Request) => Promise<Response>;
|
|
83
|
+
|
|
84
|
+
export { type AllemAIConfig, AllemAIProvider, type AllemAIProviderProps, type AllemChatHandlerConfig, type AllemProvider, type AllemProviderName, type UseAllemChatOptions, type UseAllemCompletionOptions, createAllemChatHandler, useAllemAIConfig, useAllemChat, useAllemCompletion };
|
package/dist/index.js
ADDED
|
@@ -0,0 +1,138 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __defProp = Object.defineProperty;
|
|
3
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
4
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
5
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
6
|
+
var __export = (target, all) => {
|
|
7
|
+
for (var name in all)
|
|
8
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
9
|
+
};
|
|
10
|
+
var __copyProps = (to, from, except, desc) => {
|
|
11
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
12
|
+
for (let key of __getOwnPropNames(from))
|
|
13
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
14
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
15
|
+
}
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
19
|
+
|
|
20
|
+
// src/index.ts
|
|
21
|
+
var src_exports = {};
|
|
22
|
+
__export(src_exports, {
|
|
23
|
+
AllemAIProvider: () => AllemAIProvider,
|
|
24
|
+
createAllemChatHandler: () => createAllemChatHandler,
|
|
25
|
+
useAllemAIConfig: () => useAllemAIConfig,
|
|
26
|
+
useAllemChat: () => useAllemChat,
|
|
27
|
+
useAllemCompletion: () => useAllemCompletion
|
|
28
|
+
});
|
|
29
|
+
module.exports = __toCommonJS(src_exports);
|
|
30
|
+
|
|
31
|
+
// src/useAllemChat.ts
|
|
32
|
+
var import_react2 = require("@ai-sdk/react");
|
|
33
|
+
var import_ai = require("ai");
|
|
34
|
+
|
|
35
|
+
// src/AllemAIProvider.tsx
|
|
36
|
+
var import_react = require("react");
|
|
37
|
+
var import_jsx_runtime = require("react/jsx-runtime");
|
|
38
|
+
var defaults = {
|
|
39
|
+
api: "/api/chat",
|
|
40
|
+
provider: "google"
|
|
41
|
+
};
|
|
42
|
+
var AllemAIContext = (0, import_react.createContext)(defaults);
|
|
43
|
+
function AllemAIProvider({
|
|
44
|
+
children,
|
|
45
|
+
...config
|
|
46
|
+
}) {
|
|
47
|
+
const value = { ...defaults, ...config };
|
|
48
|
+
return /* @__PURE__ */ (0, import_jsx_runtime.jsx)(AllemAIContext.Provider, { value, children });
|
|
49
|
+
}
|
|
50
|
+
function useAllemAIConfig() {
|
|
51
|
+
return (0, import_react.useContext)(AllemAIContext);
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
// src/useAllemChat.ts
|
|
55
|
+
function useAllemChat(options = {}) {
|
|
56
|
+
const config = useAllemAIConfig();
|
|
57
|
+
const {
|
|
58
|
+
api,
|
|
59
|
+
model,
|
|
60
|
+
provider,
|
|
61
|
+
systemPrompt,
|
|
62
|
+
headers,
|
|
63
|
+
experimental_throttle,
|
|
64
|
+
...restOptions
|
|
65
|
+
} = options;
|
|
66
|
+
const resolvedApi = api ?? config.api;
|
|
67
|
+
const resolvedModel = model ?? config.model;
|
|
68
|
+
const resolvedProvider = provider ?? config.provider;
|
|
69
|
+
const resolvedHeaders = { ...config.headers, ...headers };
|
|
70
|
+
const transport = new import_ai.DefaultChatTransport({
|
|
71
|
+
api: resolvedApi,
|
|
72
|
+
headers: resolvedHeaders,
|
|
73
|
+
body: {
|
|
74
|
+
...resolvedModel && { model: resolvedModel },
|
|
75
|
+
...resolvedProvider && { provider: resolvedProvider },
|
|
76
|
+
...systemPrompt && { systemPrompt }
|
|
77
|
+
}
|
|
78
|
+
});
|
|
79
|
+
return (0, import_react2.useChat)({
|
|
80
|
+
transport,
|
|
81
|
+
experimental_throttle,
|
|
82
|
+
...restOptions
|
|
83
|
+
});
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
// src/useAllemCompletion.ts
|
|
87
|
+
var import_react3 = require("@ai-sdk/react");
|
|
88
|
+
function useAllemCompletion(options = {}) {
|
|
89
|
+
const config = useAllemAIConfig();
|
|
90
|
+
const { api, headers, ...restOptions } = options;
|
|
91
|
+
return (0, import_react3.useCompletion)({
|
|
92
|
+
api: api ?? config.api,
|
|
93
|
+
headers: { ...config.headers, ...headers },
|
|
94
|
+
...restOptions
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
// src/createAllemChatHandler.ts
|
|
99
|
+
var import_ai2 = require("ai");
|
|
100
|
+
function createAllemChatHandler(config) {
|
|
101
|
+
const {
|
|
102
|
+
providers,
|
|
103
|
+
defaultProvider = "google",
|
|
104
|
+
defaultModel,
|
|
105
|
+
systemPrompt: defaultSystemPrompt,
|
|
106
|
+
tools
|
|
107
|
+
} = config;
|
|
108
|
+
return async function handler(req) {
|
|
109
|
+
const body = await req.json();
|
|
110
|
+
const providerName = body.provider ?? defaultProvider;
|
|
111
|
+
const modelId = body.model ?? defaultModel;
|
|
112
|
+
const system = body.systemPrompt ?? defaultSystemPrompt;
|
|
113
|
+
const providerFactory = providers[providerName];
|
|
114
|
+
if (!providerFactory) {
|
|
115
|
+
return new Response(
|
|
116
|
+
JSON.stringify({ error: `Provider "${providerName}" is not configured.` }),
|
|
117
|
+
{ status: 400, headers: { "Content-Type": "application/json" } }
|
|
118
|
+
);
|
|
119
|
+
}
|
|
120
|
+
const model = providerFactory(modelId);
|
|
121
|
+
const modelMessages = await (0, import_ai2.convertToModelMessages)(body.messages);
|
|
122
|
+
const result = (0, import_ai2.streamText)({
|
|
123
|
+
model,
|
|
124
|
+
system,
|
|
125
|
+
messages: modelMessages,
|
|
126
|
+
tools
|
|
127
|
+
});
|
|
128
|
+
return result.toUIMessageStreamResponse();
|
|
129
|
+
};
|
|
130
|
+
}
|
|
131
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
132
|
+
0 && (module.exports = {
|
|
133
|
+
AllemAIProvider,
|
|
134
|
+
createAllemChatHandler,
|
|
135
|
+
useAllemAIConfig,
|
|
136
|
+
useAllemChat,
|
|
137
|
+
useAllemCompletion
|
|
138
|
+
});
|
package/dist/index.mjs
ADDED
|
@@ -0,0 +1,112 @@
|
|
|
1
|
+
// src/useAllemChat.ts
|
|
2
|
+
import { useChat } from "@ai-sdk/react";
|
|
3
|
+
import { DefaultChatTransport } from "ai";
|
|
4
|
+
|
|
5
|
+
// src/AllemAIProvider.tsx
|
|
6
|
+
import { createContext, useContext } from "react";
|
|
7
|
+
import { jsx } from "react/jsx-runtime";
|
|
8
|
+
var defaults = {
|
|
9
|
+
api: "/api/chat",
|
|
10
|
+
provider: "google"
|
|
11
|
+
};
|
|
12
|
+
var AllemAIContext = createContext(defaults);
|
|
13
|
+
function AllemAIProvider({
|
|
14
|
+
children,
|
|
15
|
+
...config
|
|
16
|
+
}) {
|
|
17
|
+
const value = { ...defaults, ...config };
|
|
18
|
+
return /* @__PURE__ */ jsx(AllemAIContext.Provider, { value, children });
|
|
19
|
+
}
|
|
20
|
+
function useAllemAIConfig() {
|
|
21
|
+
return useContext(AllemAIContext);
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
// src/useAllemChat.ts
|
|
25
|
+
function useAllemChat(options = {}) {
|
|
26
|
+
const config = useAllemAIConfig();
|
|
27
|
+
const {
|
|
28
|
+
api,
|
|
29
|
+
model,
|
|
30
|
+
provider,
|
|
31
|
+
systemPrompt,
|
|
32
|
+
headers,
|
|
33
|
+
experimental_throttle,
|
|
34
|
+
...restOptions
|
|
35
|
+
} = options;
|
|
36
|
+
const resolvedApi = api ?? config.api;
|
|
37
|
+
const resolvedModel = model ?? config.model;
|
|
38
|
+
const resolvedProvider = provider ?? config.provider;
|
|
39
|
+
const resolvedHeaders = { ...config.headers, ...headers };
|
|
40
|
+
const transport = new DefaultChatTransport({
|
|
41
|
+
api: resolvedApi,
|
|
42
|
+
headers: resolvedHeaders,
|
|
43
|
+
body: {
|
|
44
|
+
...resolvedModel && { model: resolvedModel },
|
|
45
|
+
...resolvedProvider && { provider: resolvedProvider },
|
|
46
|
+
...systemPrompt && { systemPrompt }
|
|
47
|
+
}
|
|
48
|
+
});
|
|
49
|
+
return useChat({
|
|
50
|
+
transport,
|
|
51
|
+
experimental_throttle,
|
|
52
|
+
...restOptions
|
|
53
|
+
});
|
|
54
|
+
}
|
|
55
|
+
|
|
56
|
+
// src/useAllemCompletion.ts
|
|
57
|
+
import {
|
|
58
|
+
useCompletion
|
|
59
|
+
} from "@ai-sdk/react";
|
|
60
|
+
function useAllemCompletion(options = {}) {
|
|
61
|
+
const config = useAllemAIConfig();
|
|
62
|
+
const { api, headers, ...restOptions } = options;
|
|
63
|
+
return useCompletion({
|
|
64
|
+
api: api ?? config.api,
|
|
65
|
+
headers: { ...config.headers, ...headers },
|
|
66
|
+
...restOptions
|
|
67
|
+
});
|
|
68
|
+
}
|
|
69
|
+
|
|
70
|
+
// src/createAllemChatHandler.ts
|
|
71
|
+
import {
|
|
72
|
+
streamText,
|
|
73
|
+
convertToModelMessages
|
|
74
|
+
} from "ai";
|
|
75
|
+
function createAllemChatHandler(config) {
|
|
76
|
+
const {
|
|
77
|
+
providers,
|
|
78
|
+
defaultProvider = "google",
|
|
79
|
+
defaultModel,
|
|
80
|
+
systemPrompt: defaultSystemPrompt,
|
|
81
|
+
tools
|
|
82
|
+
} = config;
|
|
83
|
+
return async function handler(req) {
|
|
84
|
+
const body = await req.json();
|
|
85
|
+
const providerName = body.provider ?? defaultProvider;
|
|
86
|
+
const modelId = body.model ?? defaultModel;
|
|
87
|
+
const system = body.systemPrompt ?? defaultSystemPrompt;
|
|
88
|
+
const providerFactory = providers[providerName];
|
|
89
|
+
if (!providerFactory) {
|
|
90
|
+
return new Response(
|
|
91
|
+
JSON.stringify({ error: `Provider "${providerName}" is not configured.` }),
|
|
92
|
+
{ status: 400, headers: { "Content-Type": "application/json" } }
|
|
93
|
+
);
|
|
94
|
+
}
|
|
95
|
+
const model = providerFactory(modelId);
|
|
96
|
+
const modelMessages = await convertToModelMessages(body.messages);
|
|
97
|
+
const result = streamText({
|
|
98
|
+
model,
|
|
99
|
+
system,
|
|
100
|
+
messages: modelMessages,
|
|
101
|
+
tools
|
|
102
|
+
});
|
|
103
|
+
return result.toUIMessageStreamResponse();
|
|
104
|
+
};
|
|
105
|
+
}
|
|
106
|
+
export {
|
|
107
|
+
AllemAIProvider,
|
|
108
|
+
createAllemChatHandler,
|
|
109
|
+
useAllemAIConfig,
|
|
110
|
+
useAllemChat,
|
|
111
|
+
useAllemCompletion
|
|
112
|
+
};
|
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { LanguageModel, ToolSet } from 'ai';
|
|
2
|
+
|
|
3
|
+
type AllemProviderName = "google" | "anthropic" | "openai";
|
|
4
|
+
interface AllemChatHandlerConfig {
|
|
5
|
+
/** Map of provider name to a function that returns a LanguageModel for a given model ID. */
|
|
6
|
+
providers: Partial<Record<AllemProviderName, (modelId?: string) => LanguageModel>>;
|
|
7
|
+
/** Default provider when none is specified in the request. Default: "google". */
|
|
8
|
+
defaultProvider?: AllemProviderName;
|
|
9
|
+
/** Default model ID when none is specified in the request. */
|
|
10
|
+
defaultModel?: string;
|
|
11
|
+
/** Default system prompt. */
|
|
12
|
+
systemPrompt?: string;
|
|
13
|
+
/** Tools available to the model. */
|
|
14
|
+
tools?: ToolSet;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Creates an API route handler for AI chat.
|
|
18
|
+
* Works with Next.js App Router, Remix, and any framework that uses
|
|
19
|
+
* the Web Request/Response API.
|
|
20
|
+
*
|
|
21
|
+
* @example
|
|
22
|
+
* ```ts
|
|
23
|
+
* // app/api/chat/route.ts (Next.js)
|
|
24
|
+
* import { createAllemChatHandler } from "@allem-sdk/ai";
|
|
25
|
+
* import { google } from "@ai-sdk/google";
|
|
26
|
+
* import { anthropic } from "@ai-sdk/anthropic";
|
|
27
|
+
*
|
|
28
|
+
* export const POST = createAllemChatHandler({
|
|
29
|
+
* providers: {
|
|
30
|
+
* google: (model) => google(model ?? "gemini-2.0-flash"),
|
|
31
|
+
* anthropic: (model) => anthropic(model ?? "claude-sonnet-4-20250514"),
|
|
32
|
+
* },
|
|
33
|
+
* });
|
|
34
|
+
* ```
|
|
35
|
+
*/
|
|
36
|
+
declare function createAllemChatHandler(config: AllemChatHandlerConfig): (req: Request) => Promise<Response>;
|
|
37
|
+
|
|
38
|
+
export { type AllemChatHandlerConfig, type AllemProviderName, createAllemChatHandler };
|
package/dist/server.d.ts
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
import { LanguageModel, ToolSet } from 'ai';
|
|
2
|
+
|
|
3
|
+
type AllemProviderName = "google" | "anthropic" | "openai";
|
|
4
|
+
interface AllemChatHandlerConfig {
|
|
5
|
+
/** Map of provider name to a function that returns a LanguageModel for a given model ID. */
|
|
6
|
+
providers: Partial<Record<AllemProviderName, (modelId?: string) => LanguageModel>>;
|
|
7
|
+
/** Default provider when none is specified in the request. Default: "google". */
|
|
8
|
+
defaultProvider?: AllemProviderName;
|
|
9
|
+
/** Default model ID when none is specified in the request. */
|
|
10
|
+
defaultModel?: string;
|
|
11
|
+
/** Default system prompt. */
|
|
12
|
+
systemPrompt?: string;
|
|
13
|
+
/** Tools available to the model. */
|
|
14
|
+
tools?: ToolSet;
|
|
15
|
+
}
|
|
16
|
+
/**
|
|
17
|
+
* Creates an API route handler for AI chat.
|
|
18
|
+
* Works with Next.js App Router, Remix, and any framework that uses
|
|
19
|
+
* the Web Request/Response API.
|
|
20
|
+
*
|
|
21
|
+
* @example
|
|
22
|
+
* ```ts
|
|
23
|
+
* // app/api/chat/route.ts (Next.js)
|
|
24
|
+
* import { createAllemChatHandler } from "@allem-sdk/ai";
|
|
25
|
+
* import { google } from "@ai-sdk/google";
|
|
26
|
+
* import { anthropic } from "@ai-sdk/anthropic";
|
|
27
|
+
*
|
|
28
|
+
* export const POST = createAllemChatHandler({
|
|
29
|
+
* providers: {
|
|
30
|
+
* google: (model) => google(model ?? "gemini-2.0-flash"),
|
|
31
|
+
* anthropic: (model) => anthropic(model ?? "claude-sonnet-4-20250514"),
|
|
32
|
+
* },
|
|
33
|
+
* });
|
|
34
|
+
* ```
|
|
35
|
+
*/
|
|
36
|
+
declare function createAllemChatHandler(config: AllemChatHandlerConfig): (req: Request) => Promise<Response>;
|
|
37
|
+
|
|
38
|
+
export { type AllemChatHandlerConfig, type AllemProviderName, createAllemChatHandler };
|
package/dist/server.js
ADDED
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __defProp = Object.defineProperty;
|
|
3
|
+
var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
|
|
4
|
+
var __getOwnPropNames = Object.getOwnPropertyNames;
|
|
5
|
+
var __hasOwnProp = Object.prototype.hasOwnProperty;
|
|
6
|
+
var __export = (target, all) => {
|
|
7
|
+
for (var name in all)
|
|
8
|
+
__defProp(target, name, { get: all[name], enumerable: true });
|
|
9
|
+
};
|
|
10
|
+
var __copyProps = (to, from, except, desc) => {
|
|
11
|
+
if (from && typeof from === "object" || typeof from === "function") {
|
|
12
|
+
for (let key of __getOwnPropNames(from))
|
|
13
|
+
if (!__hasOwnProp.call(to, key) && key !== except)
|
|
14
|
+
__defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
|
|
15
|
+
}
|
|
16
|
+
return to;
|
|
17
|
+
};
|
|
18
|
+
var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
|
|
19
|
+
|
|
20
|
+
// src/server.ts
|
|
21
|
+
var server_exports = {};
|
|
22
|
+
__export(server_exports, {
|
|
23
|
+
createAllemChatHandler: () => createAllemChatHandler
|
|
24
|
+
});
|
|
25
|
+
module.exports = __toCommonJS(server_exports);
|
|
26
|
+
|
|
27
|
+
// src/createAllemChatHandler.ts
|
|
28
|
+
var import_ai = require("ai");
|
|
29
|
+
function createAllemChatHandler(config) {
|
|
30
|
+
const {
|
|
31
|
+
providers,
|
|
32
|
+
defaultProvider = "google",
|
|
33
|
+
defaultModel,
|
|
34
|
+
systemPrompt: defaultSystemPrompt,
|
|
35
|
+
tools
|
|
36
|
+
} = config;
|
|
37
|
+
return async function handler(req) {
|
|
38
|
+
const body = await req.json();
|
|
39
|
+
const providerName = body.provider ?? defaultProvider;
|
|
40
|
+
const modelId = body.model ?? defaultModel;
|
|
41
|
+
const system = body.systemPrompt ?? defaultSystemPrompt;
|
|
42
|
+
const providerFactory = providers[providerName];
|
|
43
|
+
if (!providerFactory) {
|
|
44
|
+
return new Response(
|
|
45
|
+
JSON.stringify({ error: `Provider "${providerName}" is not configured.` }),
|
|
46
|
+
{ status: 400, headers: { "Content-Type": "application/json" } }
|
|
47
|
+
);
|
|
48
|
+
}
|
|
49
|
+
const model = providerFactory(modelId);
|
|
50
|
+
const modelMessages = await (0, import_ai.convertToModelMessages)(body.messages);
|
|
51
|
+
const result = (0, import_ai.streamText)({
|
|
52
|
+
model,
|
|
53
|
+
system,
|
|
54
|
+
messages: modelMessages,
|
|
55
|
+
tools
|
|
56
|
+
});
|
|
57
|
+
return result.toUIMessageStreamResponse();
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
// Annotate the CommonJS export names for ESM import in node:
|
|
61
|
+
0 && (module.exports = {
|
|
62
|
+
createAllemChatHandler
|
|
63
|
+
});
|
package/dist/server.mjs
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
1
|
+
// src/createAllemChatHandler.ts
|
|
2
|
+
import {
|
|
3
|
+
streamText,
|
|
4
|
+
convertToModelMessages
|
|
5
|
+
} from "ai";
|
|
6
|
+
function createAllemChatHandler(config) {
|
|
7
|
+
const {
|
|
8
|
+
providers,
|
|
9
|
+
defaultProvider = "google",
|
|
10
|
+
defaultModel,
|
|
11
|
+
systemPrompt: defaultSystemPrompt,
|
|
12
|
+
tools
|
|
13
|
+
} = config;
|
|
14
|
+
return async function handler(req) {
|
|
15
|
+
const body = await req.json();
|
|
16
|
+
const providerName = body.provider ?? defaultProvider;
|
|
17
|
+
const modelId = body.model ?? defaultModel;
|
|
18
|
+
const system = body.systemPrompt ?? defaultSystemPrompt;
|
|
19
|
+
const providerFactory = providers[providerName];
|
|
20
|
+
if (!providerFactory) {
|
|
21
|
+
return new Response(
|
|
22
|
+
JSON.stringify({ error: `Provider "${providerName}" is not configured.` }),
|
|
23
|
+
{ status: 400, headers: { "Content-Type": "application/json" } }
|
|
24
|
+
);
|
|
25
|
+
}
|
|
26
|
+
const model = providerFactory(modelId);
|
|
27
|
+
const modelMessages = await convertToModelMessages(body.messages);
|
|
28
|
+
const result = streamText({
|
|
29
|
+
model,
|
|
30
|
+
system,
|
|
31
|
+
messages: modelMessages,
|
|
32
|
+
tools
|
|
33
|
+
});
|
|
34
|
+
return result.toUIMessageStreamResponse();
|
|
35
|
+
};
|
|
36
|
+
}
|
|
37
|
+
export {
|
|
38
|
+
createAllemChatHandler
|
|
39
|
+
};
|
package/package.json
ADDED
|
@@ -0,0 +1,86 @@
|
|
|
1
|
+
{
|
|
2
|
+
"name": "@allem-sdk/ai",
|
|
3
|
+
"version": "0.1.0",
|
|
4
|
+
"description": "AI hooks for React — useChat, useAgent, useStreaming. Built on Vercel AI SDK.",
|
|
5
|
+
"license": "MIT",
|
|
6
|
+
"author": "Ahmed Allem",
|
|
7
|
+
"main": "./dist/index.js",
|
|
8
|
+
"module": "./dist/index.mjs",
|
|
9
|
+
"types": "./dist/index.d.ts",
|
|
10
|
+
"exports": {
|
|
11
|
+
".": {
|
|
12
|
+
"types": "./dist/index.d.ts",
|
|
13
|
+
"import": "./dist/index.mjs",
|
|
14
|
+
"require": "./dist/index.js"
|
|
15
|
+
},
|
|
16
|
+
"./client": {
|
|
17
|
+
"types": "./dist/client.d.ts",
|
|
18
|
+
"import": "./dist/client.mjs",
|
|
19
|
+
"require": "./dist/client.js"
|
|
20
|
+
},
|
|
21
|
+
"./server": {
|
|
22
|
+
"types": "./dist/server.d.ts",
|
|
23
|
+
"import": "./dist/server.mjs",
|
|
24
|
+
"require": "./dist/server.js"
|
|
25
|
+
}
|
|
26
|
+
},
|
|
27
|
+
"files": [
|
|
28
|
+
"dist"
|
|
29
|
+
],
|
|
30
|
+
"sideEffects": false,
|
|
31
|
+
"devDependencies": {
|
|
32
|
+
"@ai-sdk/anthropic": "^3.0.78",
|
|
33
|
+
"@ai-sdk/google": "^3.0.75",
|
|
34
|
+
"@ai-sdk/openai": "^3.0.64",
|
|
35
|
+
"@ai-sdk/react": "^3.0.187",
|
|
36
|
+
"@types/react": "^19.0.0",
|
|
37
|
+
"ai": "^6.0.185",
|
|
38
|
+
"react": "^19.0.0",
|
|
39
|
+
"tsup": "^8.4.0",
|
|
40
|
+
"typescript": "^5.8.0",
|
|
41
|
+
"vitest": "^3.0.0"
|
|
42
|
+
},
|
|
43
|
+
"peerDependencies": {
|
|
44
|
+
"@ai-sdk/anthropic": ">=3.0.0",
|
|
45
|
+
"@ai-sdk/google": ">=3.0.0",
|
|
46
|
+
"@ai-sdk/openai": ">=3.0.0",
|
|
47
|
+
"@ai-sdk/react": ">=3.0.0",
|
|
48
|
+
"ai": ">=6.0.0",
|
|
49
|
+
"react": ">=18.0.0"
|
|
50
|
+
},
|
|
51
|
+
"peerDependenciesMeta": {
|
|
52
|
+
"@ai-sdk/google": {
|
|
53
|
+
"optional": true
|
|
54
|
+
},
|
|
55
|
+
"@ai-sdk/anthropic": {
|
|
56
|
+
"optional": true
|
|
57
|
+
},
|
|
58
|
+
"@ai-sdk/openai": {
|
|
59
|
+
"optional": true
|
|
60
|
+
}
|
|
61
|
+
},
|
|
62
|
+
"repository": {
|
|
63
|
+
"type": "git",
|
|
64
|
+
"url": "https://github.com/kingofmit/allem-sdk.git",
|
|
65
|
+
"directory": "packages/ai"
|
|
66
|
+
},
|
|
67
|
+
"keywords": [
|
|
68
|
+
"allem-sdk",
|
|
69
|
+
"ai",
|
|
70
|
+
"react",
|
|
71
|
+
"hooks",
|
|
72
|
+
"useChat",
|
|
73
|
+
"useAgent",
|
|
74
|
+
"vercel-ai-sdk",
|
|
75
|
+
"llm"
|
|
76
|
+
],
|
|
77
|
+
"publishConfig": {
|
|
78
|
+
"access": "public"
|
|
79
|
+
},
|
|
80
|
+
"scripts": {
|
|
81
|
+
"build": "tsup",
|
|
82
|
+
"dev": "tsup --watch",
|
|
83
|
+
"clean": "rm -rf dist",
|
|
84
|
+
"test": "vitest run"
|
|
85
|
+
}
|
|
86
|
+
}
|