ai-publish-sdk 1.3.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,9 +88,20 @@ 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()
 }
 ```
 
+### getBrandingAssets()
+
+Get tenant branding assets for white-label styling.
+
+```typescript
+getBrandingAssets(): Promise<BrandingAssets | null>
+```
+
+Returns `BrandingAssets` with fields: `logoUrl`, `darkLogoUrl`, `brandColor` (all `string | null`).
+
 ### executeScript(code)
 
 Execute JavaScript on the host page.

package/index.d.ts

@@ -1,5 +1,5 @@
-export { generateMessage, 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 { 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 m(e){u=e}let i;function d(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 p(e,t){return r("generateMessage",[e,t])}async function y(e,t){return r("getToken",[e,t])}async function w(){return r("getUserInfo",[])}async function T(e){return r("tcpConnect",[e])}async function v(e,t){return r("tcpSend",[e,t])}async function I(e,t){return r("tcpReceive",[e,t])}async function C(e){return r("tcpClose",[e])}const b={async connect(e,t){const n=await T({host:e,port:t});return n===null?null:{send:async s=>(await v(n,{dataBase64:s}))?.bytesSent??null,receive:async()=>(await I(n))?.data??null,close:()=>C(n)}}};exports.generateMessage=p;exports.getToken=y;exports.getUserInfo=w;exports.setGlobalTimeout=m;exports.tcp=b;exports.withTimeout=d;
+"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 m() {
+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 m() {
   }));
 }
 function r(e, t) {
-  return m(), 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);
@@ -35,28 +35,34 @@ function r(e, t) {
     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() {
+  return r("getBrandingAssets", []);
+}
 async function f(e) {
   return r("tcpConnect", [e]);
 }
-async function g(e, t) {
+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 R = {
+const U = {
   /**
    * Opens a TCP connection and returns a `TcpSocket` wrapper.
    *
@@ -68,17 +74,19 @@ const R = {
   async connect(e, t) {
     const n = await f({ host: e, port: t });
     return n === null ? null : {
-      send: async (s) => (await g(n, { dataBase64: s }))?.bytesSent ?? null,
-      receive: async () => (await p(n))?.data ?? null,
-      close: () => y(n)
+      send: async (s) => (await m(n, { dataBase64: s }))?.bytesSent ?? null,
+      receive: async () => (await y(n))?.data ?? null,
+      close: () => p(n)
     };
   }
 };
 export {
-  C as generateMessage,
-  I as getToken,
-  M as getUserInfo,
+  I as generateMessage,
+  C as getAllowedModels,
+  R as getBrandingAssets,
+  M as getToken,
+  A as getUserInfo,
   T as setGlobalTimeout,
-  R as tcp,
+  U as tcp,
   v as withTimeout
 };

package/lib/functions.d.ts

@@ -1,7 +1,9 @@
-import { 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>;
+export declare function getBrandingAssets(): Promise<BrandingAssets | null>;
 export declare function executeScript(code: string): Promise<void>;
 export declare function tcpConnect(options: TcpConnectOptions): Promise<TcpConnectionSettings | null>;
 export declare function tcpSend(connection: TcpConnectionSettings, message: TcpSendMessage): Promise<TcpSendResult | 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;
@@ -21,6 +26,11 @@ export interface UserInfo {
     email?: string;
     picture?: string;
 }
+export interface BrandingAssets {
+    logoUrl: string | null;
+    darkLogoUrl: string | null;
+    brandColor: string | null;
+}
 export interface TcpConnectOptions {
     host: string;
     port: number;

package/package.json

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