ai-publish-sdk 1.4.0 → 1.5.0

package/README.md

@@ -24,6 +24,21 @@ await withTimeout(() => getUserInfo(), 10_000)
 
 ## Core API
 
+### getAllowedModels()
+
+Get the list of allowed AI models and the default model configured by the admin.
+
+```typescript
+getAllowedModels(config?: RpcConfig): Promise<AllowedModelsInfo>
+
+interface AllowedModelsInfo {
+  models: string[]
+  defaultModel: string | null
+}
+```
+
+Returns an empty `models` array when no models are configured.
+
 ### getUserInfo()
 
 Get current user info.
@@ -73,6 +88,7 @@ generateMessage(prompt: string, options: GenerateMessageOptions): Promise<string
 interface GenerateMessageOptions {
   withPageContext?: boolean // include current page content as context
   systemPrompt?: string    // guide AI behavior
+  model?: string           // model name from getAllowedModels()
 }
 ```
 

package/index.d.ts

@@ -1,5 +1,5 @@
-export { generateMessage, getBrandingAssets, getToken, getUserInfo } from './lib/functions';
+export { getAllowedModels, generateMessage, getBrandingAssets, getToken, getUserInfo } from './lib/functions';
 export { tcp } from './lib/tcp-socket';
 export type { TcpSocket } from './lib/tcp-socket';
 export { setGlobalTimeout, withTimeout } from './lib/rpc-bridge';
-export type { BrandingAssets, GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, UserInfo } from './lib/types';
+export type { AllowedModelsInfo, BrandingAssets, GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, UserInfo } from './lib/types';

package/index.js

@@ -1 +1 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});let u=15e3;const c=new Map;let a=!1;function d(e){u=e}let i;function m(e,t){i=t;try{return e()}finally{i=void 0}}function f(){a||(a=!0,window.addEventListener("message",e=>{const t=e.data;if(!t?.messageId||!("name"in t))return;const n=c.get(t.messageId);n&&(clearTimeout(n.timer),c.delete(t.messageId),t.error?n.reject(new Error(t.error.message??"RPC error")):n.resolve(t.data))}))}function r(e,t){return f(),new Promise((n,s)=>{const o=crypto.randomUUID(),l=setTimeout(()=>{c.delete(o),s(new Error(`RPC timeout: ${e}`))},i??u);c.set(o,{resolve:n,reject:s,timer:l});const g={action:e,params:t,messageId:o};window.parent.postMessage(g,"*")})}async function y(e,t){return r("generateMessage",[e,t])}async function p(e,t){return r("getToken",[e,t])}async function w(){return r("getUserInfo",[])}async function T(){return r("getBrandingAssets",[])}async function v(e){return r("tcpConnect",[e])}async function I(e,t){return r("tcpSend",[e,t])}async function C(e,t){return r("tcpReceive",[e,t])}async function b(e){return r("tcpClose",[e])}const M={async connect(e,t){const n=await v({host:e,port:t});return n===null?null:{send:async s=>(await I(n,{dataBase64:s}))?.bytesSent??null,receive:async()=>(await C(n))?.data??null,close:()=>b(n)}}};exports.generateMessage=y;exports.getBrandingAssets=T;exports.getToken=p;exports.getUserInfo=w;exports.setGlobalTimeout=d;exports.tcp=M;exports.withTimeout=m;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});let u=15e3;const c=new Map;let a=!1;function d(e){u=e}let i;function f(e,t){i=t;try{return e()}finally{i=void 0}}function m(){a||(a=!0,window.addEventListener("message",e=>{const t=e.data;if(!t?.messageId||!("name"in t))return;const n=c.get(t.messageId);n&&(clearTimeout(n.timer),c.delete(t.messageId),t.error?n.reject(new Error(t.error.message??"RPC error")):n.resolve(t.data))}))}function r(e,t){return m(),new Promise((n,s)=>{const o=crypto.randomUUID(),l=setTimeout(()=>{c.delete(o),s(new Error(`RPC timeout: ${e}`))},i??u);c.set(o,{resolve:n,reject:s,timer:l});const g={action:e,params:t,messageId:o};window.parent.postMessage(g,"*")})}async function y(){return r("getAllowedModels",[])}async function p(e,t){return r("generateMessage",[e,t])}async function w(e,t){return r("getToken",[e,t])}async function T(){return r("getUserInfo",[])}async function M(){return r("getBrandingAssets",[])}async function v(e){return r("tcpConnect",[e])}async function I(e,t){return r("tcpSend",[e,t])}async function C(e,t){return r("tcpReceive",[e,t])}async function b(e){return r("tcpClose",[e])}const A={async connect(e,t){const n=await v({host:e,port:t});return n===null?null:{send:async s=>(await I(n,{dataBase64:s}))?.bytesSent??null,receive:async()=>(await C(n))?.data??null,close:()=>b(n)}}};exports.generateMessage=p;exports.getAllowedModels=y;exports.getBrandingAssets=M;exports.getToken=w;exports.getUserInfo=T;exports.setGlobalTimeout=d;exports.tcp=A;exports.withTimeout=f;

package/index.mjs

@@ -13,7 +13,7 @@ function v(e, t) {
     a = void 0;
   }
 }
-function d() {
+function g() {
   i || (i = !0, window.addEventListener("message", (e) => {
     const t = e.data;
     if (!t?.messageId || !("name" in t)) return;
@@ -22,7 +22,7 @@ function d() {
   }));
 }
 function r(e, t) {
-  return d(), new Promise((n, s) => {
+  return g(), new Promise((n, s) => {
     const o = crypto.randomUUID(), l = setTimeout(() => {
       c.delete(o), s(new Error(`RPC timeout: ${e}`));
     }, a ?? u);
@@ -31,17 +31,20 @@ function r(e, t) {
       reject: s,
       timer: l
     });
-    const g = { action: e, params: t, messageId: o };
-    window.parent.postMessage(g, "*");
+    const d = { action: e, params: t, messageId: o };
+    window.parent.postMessage(d, "*");
   });
 }
-async function C(e, t) {
-  return r("generateMessage", [e, t]);
+async function C() {
+  return r("getAllowedModels", []);
 }
 async function I(e, t) {
+  return r("generateMessage", [e, t]);
+}
+async function M(e, t) {
   return r("getToken", [e, t]);
 }
-async function M() {
+async function A() {
   return r("getUserInfo", []);
 }
 async function R() {
@@ -53,10 +56,10 @@ async function f(e) {
 async function m(e, t) {
   return r("tcpSend", [e, t]);
 }
-async function p(e, t) {
+async function y(e, t) {
   return r("tcpReceive", [e, t]);
 }
-async function y(e) {
+async function p(e) {
   return r("tcpClose", [e]);
 }
 const U = {
@@ -72,16 +75,17 @@ const U = {
     const n = await f({ host: e, port: t });
     return n === null ? null : {
       send: async (s) => (await m(n, { dataBase64: s }))?.bytesSent ?? null,
-      receive: async () => (await p(n))?.data ?? null,
-      close: () => y(n)
+      receive: async () => (await y(n))?.data ?? null,
+      close: () => p(n)
     };
   }
 };
 export {
-  C as generateMessage,
+  I as generateMessage,
+  C as getAllowedModels,
   R as getBrandingAssets,
-  I as getToken,
-  M as getUserInfo,
+  M as getToken,
+  A as getUserInfo,
   T as setGlobalTimeout,
   U as tcp,
   v as withTimeout

package/lib/functions.d.ts

@@ -1,4 +1,5 @@
-import { BrandingAssets, GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, TcpConnectOptions, TcpConnectionSettings, TcpReceiveOptions, TcpReceiveResult, TcpSendMessage, TcpSendResult, UserInfo } from './types';
+import { AllowedModelsInfo, BrandingAssets, GenerateMessageOptions, GetTokenOptions, GetTokenResult, IntegrationAppName, TcpConnectOptions, TcpConnectionSettings, TcpReceiveOptions, TcpReceiveResult, TcpSendMessage, TcpSendResult, UserInfo } from './types';
+export declare function getAllowedModels(): Promise<AllowedModelsInfo>;
 export declare function generateMessage(prompt: string, options: GenerateMessageOptions): Promise<string | null>;
 export declare function getToken(appName: IntegrationAppName, options?: GetTokenOptions): Promise<GetTokenResult | null>;
 export declare function getUserInfo(): Promise<UserInfo | null>;

package/lib/types.d.ts

@@ -1,7 +1,12 @@
 export type IntegrationAppName = 'gmail' | 'jira' | 'googledrive' | 'salesforce' | 'googlecalendar' | 'slack';
+export interface AllowedModelsInfo {
+    models: string[];
+    defaultModel: string | null;
+}
 export interface GenerateMessageOptions {
     withPageContext?: boolean;
     systemPrompt?: string;
+    model?: string;
 }
 export interface GetTokenOptions {
     interactive?: boolean;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-publish-sdk",
-  "version": "1.4.0",
+  "version": "1.5.0",
   "module": "./index.mjs",
   "main": "./index.js",
   "types": "./index.d.ts",