chat-nest-sdk 1.1.0 → 1.1.2

package/README.md

@@ -45,6 +45,9 @@ import { useAiChat } from "chat-nest-sdk";
 function App() {
   const chat = useAiChat({
     endpoint: "http://localhost:3001/api/chat",
+    initialProfile: "balanced",
+    dailyTokenLimit: 50_000,    // optional: cap daily tokens
+    maxTokensPerRequest: 4096,  // optional: cap per request
   });
 
   return (
@@ -64,6 +67,11 @@ function App() {
       <button onClick={chat.cancel}>
         Cancel
       </button>
+      <select value={chat.profile} onChange={(e) => chat.setProfile(e.target.value)}>
+        <option value="constrained">Constrained</option>
+        <option value="balanced">Balanced</option>
+        <option value="expanded">Expanded</option>
+      </select>
     </>
   );
 }
@@ -75,17 +83,27 @@ function App() {
 
 ### useAiChat(options)
 
-| Field    | Type   | Description          |
-| -------- | ------ | -------------------- |
-| endpoint | string | Backend API endpoint |
-
-| Field             | Description              |
-| ----------------- | ------------------------ |
-| messages          | Chat message list        |
-| sendMessage(text) | Sends user message       |
-| cancel()          | Cancels active request   |
-| isStreaming       | Whether stream is active |
-| error             | Last error               |
+| Field               | Type   | Description                                                                 |
+| ------------------- | ------ | --------------------------------------------------------------------------- |
+| endpoint            | string | Backend API endpoint                                                        |
+| initialMessages     | Message[] | Optional. Initial messages                                              |
+| maxMessages         | number | Optional. Max messages to keep in context (default 10)                      |
+| initialProfile      | AiUsageProfile | Optional. Profile: `constrained`, `balanced`, `expanded`              |
+| dailyTokenLimit     | number | Optional. Cap daily tokens; server applies min(profile limit, this)        |
+| maxTokensPerRequest | number | Optional. Cap tokens per request (input + output); server applies min      |
+
+### Return value
+
+| Field             | Description                   |
+| ----------------- | ----------------------------- |
+| messages          | Chat message list             |
+| sendMessage(text) | Sends user message            |
+| cancel()          | Cancels active request        |
+| isStreaming       | Whether stream is active      |
+| error             | Last error                    |
+| profile           | Current profile               |
+| setProfile(p)     | Set profile; persisted to localStorage |
+| reset()           | Reset messages and error      |
 
 ---
 

package/dist/index.d.ts

@@ -1,4 +1,5 @@
-import { UseAIChatOptions, UseAIChatReturn, AIClient, AiClientConfig, Message, StreamCallbacks } from 'chat-nest-core';
+import { UseAIChatOptions, UseAIChatReturn, AIClient, AiClientConfig, Message, StreamCallbacks, AiUsageProfile, ChatRequestOptions } from 'chat-nest-core';
+export { AiUsageProfile } from 'chat-nest-core';
 
 declare function useAiChat(options: UseAIChatOptions): UseAIChatReturn;
 
@@ -6,7 +7,7 @@ declare class FetchAiClient implements AIClient {
     private abortController?;
     private config;
     constructor(config: AiClientConfig);
-    streamChat(messages: Message[], callbacks: StreamCallbacks): Promise<void>;
+    streamChat(messages: Message[], callbacks: StreamCallbacks, profile: AiUsageProfile, options?: ChatRequestOptions): Promise<void>;
     cancel(): void;
     private executeStream;
     private normalizeError;

package/dist/index.js

@@ -4,6 +4,16 @@ import {
   MessageRole
 } from "chat-nest-core";
 
+// src/core/promptPresets.ts
+var PROFILE_SYSTEM_PROMPTS = {
+  constrained: "You are a concise assistant. Answer in the fewest words possible. No explanations or preamble. Direct answers only. No examples needed",
+  balanced: "You are a helpful assistant. Be clear and concise. Give brief explanations when they add value, but keep responses focused.",
+  expanded: "You are a thorough assistant. Explain your reasoning step by step when useful. Include relevant examples and detail. Prioritize clarity and completeness."
+};
+function getSystemPromptForProfile(profile) {
+  return PROFILE_SYSTEM_PROMPTS[profile];
+}
+
 // src/core/aiClient.ts
 var DEFAULT_TIMEOUT = 3e4;
 var DEFAULT_RETRIES = 2;
@@ -22,11 +32,11 @@ var FetchAiClient = class {
       maxRetries: config.maxRetries ?? DEFAULT_RETRIES
     };
   }
-  async streamChat(messages, callbacks) {
+  async streamChat(messages, callbacks, profile, options) {
     let attempt = 0;
     while (attempt <= this.config.maxRetries) {
       try {
-        await this.executeStream(messages, callbacks);
+        await this.executeStream(messages, callbacks, profile, options);
         return;
       } catch (error) {
         if (error?.name === "AbortError") {
@@ -48,7 +58,7 @@ var FetchAiClient = class {
   cancel() {
     this.abortController?.abort();
   }
-  async executeStream(messages, callbacks) {
+  async executeStream(messages, callbacks, profile, options) {
     this.abortController = new AbortController();
     const timeoutId = setTimeout(
       () => this.abortController?.abort(),
@@ -63,7 +73,17 @@ var FetchAiClient = class {
           ...this.config.headers
         },
         signal: this.abortController.signal,
-        body: JSON.stringify({ messages })
+        body: JSON.stringify({
+          messages,
+          profile,
+          systemPrompt: getSystemPromptForProfile(profile),
+          ...options?.dailyTokenLimit != null && {
+            dailyTokenLimit: options.dailyTokenLimit
+          },
+          ...options?.maxTokensPerRequest != null && {
+            maxTokensPerRequest: options.maxTokensPerRequest
+          }
+        })
       });
       if (!response.ok) {
         if (response.status >= 400 && response.status < 500) {
@@ -159,15 +179,42 @@ function generateId() {
 }
 
 // src/react/useAiChat.ts
+var STORAGE_KEY = "aiUsageProfile";
+var VALID_PROFILES = ["constrained", "balanced", "expanded"];
+var DEFAULT_PROFILE = "balanced";
+function readStoredProfile() {
+  if (typeof window === "undefined") return DEFAULT_PROFILE;
+  try {
+    const raw = localStorage.getItem(STORAGE_KEY);
+    if (raw && VALID_PROFILES.includes(raw)) {
+      return raw;
+    }
+  } catch {
+  }
+  return DEFAULT_PROFILE;
+}
 function useAiChat(options) {
   const {
     endpoint,
     initialMessages = [],
-    maxMessages = 10
+    maxMessages = 10,
+    initialProfile,
+    dailyTokenLimit,
+    maxTokensPerRequest
   } = options;
-  const [messages, setMessages] = useState(
-    initialMessages
+  const [profile, setProfileState] = useState(
+    () => initialProfile ?? readStoredProfile()
   );
+  const setProfile = useCallback((next) => {
+    setProfileState(next);
+    try {
+      if (typeof window !== "undefined") {
+        localStorage.setItem(STORAGE_KEY, next);
+      }
+    } catch {
+    }
+  }, []);
+  const [messages, setMessages] = useState(initialMessages);
   const [isStreaming, setIsStreaming] = useState(false);
   const [error, setError] = useState();
   const clientRef = useRef(null);
@@ -196,30 +243,43 @@ function useAiChat(options) {
     setIsStreaming(true);
     const history = [...messages, userMessage];
     try {
-      await clientRef.current.streamChat(history, {
-        onToken(token) {
-          setMessages(
-            (prev) => prev.map(
-              (msg) => msg.id === assistantMessage.id ? {
-                ...msg,
-                content: msg.content + token
-              } : msg
-            )
-          );
-        },
-        onComplete() {
-          setIsStreaming(false);
+      await clientRef.current.streamChat(
+        history,
+        {
+          onToken(token) {
+            setMessages(
+              (prev) => prev.map(
+                (msg) => msg.id === assistantMessage.id ? {
+                  ...msg,
+                  content: msg.content + token
+                } : msg
+              )
+            );
+          },
+          onComplete() {
+            setIsStreaming(false);
+          },
+          onError(err) {
+            setError(err.message);
+            setIsStreaming(false);
+          }
         },
-        onError(err) {
-          setError(err.message);
-          setIsStreaming(false);
-        }
-      });
+        profile,
+        dailyTokenLimit != null || maxTokensPerRequest != null ? { dailyTokenLimit: dailyTokenLimit ?? void 0, maxTokensPerRequest: maxTokensPerRequest ?? void 0 } : void 0
+      );
     } catch (err) {
       setError(err.message);
       setIsStreaming(false);
     }
-  }, [endpoint, isStreaming, maxMessages, messages]);
+  }, [
+    endpoint,
+    isStreaming,
+    maxMessages,
+    messages,
+    profile,
+    dailyTokenLimit,
+    maxTokensPerRequest
+  ]);
   const cancel = useCallback(() => {
     clientRef.current?.cancel();
     setIsStreaming(false);
@@ -236,9 +296,11 @@ function useAiChat(options) {
       cancel,
       reset,
       isStreaming,
-      error
+      error,
+      profile,
+      setProfile
     }),
-    [messages, sendMessage, cancel, reset, isStreaming, error]
+    [messages, sendMessage, cancel, reset, isStreaming, error, profile, setProfile]
   );
 }
 export {

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "chat-nest-sdk",
 	"description": "Frontend React SDK for consuming streaming AI APIs with cancellation and retry safety.",
-	"version": "1.1.0",
+	"version": "1.1.2",
 	"type": "module",
 	"main": "dist/index.js",
 	"types": "dist/index.d.ts",