@trainly/react 1.4.1 → 1.4.3

package/dist/TrainlyProvider.js

@@ -246,7 +246,7 @@ export function TrainlyProvider(_a) {
             }
         });
     }); };
-    var ask = function (question) { return __awaiter(_this, void 0, void 0, function () {
+    var ask = function (question, options) { return __awaiter(_this, void 0, void 0, function () {
         var response, err_3, errorMessage, newToken, response, refreshError_1, error_3;
         return __generator(this, function (_a) {
             switch (_a.label) {
@@ -254,10 +254,10 @@ export function TrainlyProvider(_a) {
                     _a.trys.push([0, 2, 10, 11]);
                     setIsLoading(true);
                     setError(null);
-                    return [4 /*yield*/, client.ask(question)];
+                    return [4 /*yield*/, client.ask(question, options || {})];
                 case 1:
                     response = _a.sent();
-                    return [2 /*return*/, response.answer];
+                    return [2 /*return*/, response];
                 case 2:
                     err_3 = _a.sent();
                     errorMessage = err_3 instanceof Error ? err_3.message : String(err_3);
@@ -277,11 +277,11 @@ export function TrainlyProvider(_a) {
                     return [4 /*yield*/, client.connectWithOAuthToken(newToken)];
                 case 5:
                     _a.sent();
-                    return [4 /*yield*/, client.ask(question)];
+                    return [4 /*yield*/, client.ask(question, options || {})];
                 case 6:
                     response = _a.sent();
                     console.log("✅ Query succeeded after token refresh");
-                    return [2 /*return*/, response.answer];
+                    return [2 /*return*/, response];
                 case 7: return [3 /*break*/, 9];
                 case 8:
                     refreshError_1 = _a.sent();
@@ -302,39 +302,23 @@ export function TrainlyProvider(_a) {
             }
         });
     }); };
+    // Legacy wrapper for backward compatibility (deprecated - use ask() with options instead)
     var askWithCitations = function (question) { return __awaiter(_this, void 0, void 0, function () {
-        var response, err_4, error_4;
+        var response;
         return __generator(this, function (_a) {
             switch (_a.label) {
-                case 0:
-                    _a.trys.push([0, 2, 3, 4]);
-                    setIsLoading(true);
-                    setError(null);
-                    return [4 /*yield*/, client.ask(question, { includeCitations: true })];
+                case 0: return [4 /*yield*/, ask(question, { includeCitations: true })];
                 case 1:
                     response = _a.sent();
                     return [2 /*return*/, {
                             answer: response.answer,
                             citations: response.citations || [],
                         }];
-                case 2:
-                    err_4 = _a.sent();
-                    error_4 = {
-                        code: "QUERY_FAILED",
-                        message: "Failed to get answer with citations",
-                        details: err_4,
-                    };
-                    setError(error_4);
-                    throw error_4;
-                case 3:
-                    setIsLoading(false);
-                    return [7 /*endfinally*/];
-                case 4: return [2 /*return*/];
             }
         });
     }); };
     var upload = function (file, scopeValues) { return __awaiter(_this, void 0, void 0, function () {
-        var result, err_5, errorMessage, newToken, result, refreshError_2, error_5;
+        var result, err_4, errorMessage, newToken, result, refreshError_2, error_4;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -346,8 +330,8 @@ export function TrainlyProvider(_a) {
                     result = _a.sent();
                     return [2 /*return*/, result];
                 case 2:
-                    err_5 = _a.sent();
-                    errorMessage = err_5 instanceof Error ? err_5.message : String(err_5);
+                    err_4 = _a.sent();
+                    errorMessage = err_4 instanceof Error ? err_4.message : String(err_4);
                     if (!(getToken &&
                         appId &&
                         (errorMessage.includes("401") ||
@@ -375,13 +359,13 @@ export function TrainlyProvider(_a) {
                     console.error("❌ Token refresh failed:", refreshError_2);
                     return [3 /*break*/, 9];
                 case 9:
-                    error_5 = {
+                    error_4 = {
                         code: "UPLOAD_FAILED",
                         message: "Failed to upload file",
-                        details: err_5,
+                        details: err_4,
                     };
-                    setError(error_5);
-                    throw error_5;
+                    setError(error_4);
+                    throw error_4;
                 case 10:
                     setIsLoading(false);
                     return [7 /*endfinally*/];
@@ -390,7 +374,7 @@ export function TrainlyProvider(_a) {
         });
     }); };
     var bulkUploadFiles = function (files, scopeValues) { return __awaiter(_this, void 0, void 0, function () {
-        var result, err_6, errorMessage, newToken, result, refreshError_3, error_6;
+        var result, err_5, errorMessage, newToken, result, refreshError_3, error_5;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -402,8 +386,8 @@ export function TrainlyProvider(_a) {
                     result = _a.sent();
                     return [2 /*return*/, result];
                 case 2:
-                    err_6 = _a.sent();
-                    errorMessage = err_6 instanceof Error ? err_6.message : String(err_6);
+                    err_5 = _a.sent();
+                    errorMessage = err_5 instanceof Error ? err_5.message : String(err_5);
                     if (!(getToken &&
                         appId &&
                         (errorMessage.includes("401") ||
@@ -431,13 +415,13 @@ export function TrainlyProvider(_a) {
                     console.error("❌ Token refresh failed during bulk upload:", refreshError_3);
                     return [3 /*break*/, 9];
                 case 9:
-                    error_6 = {
+                    error_5 = {
                         code: "BULK_UPLOAD_FAILED",
                         message: "Failed to upload files",
-                        details: err_6,
+                        details: err_5,
                     };
-                    setError(error_6);
-                    throw error_6;
+                    setError(error_5);
+                    throw error_5;
                 case 10:
                     setIsLoading(false);
                     return [7 /*endfinally*/];
@@ -446,7 +430,7 @@ export function TrainlyProvider(_a) {
         });
     }); };
     var listFiles = function () { return __awaiter(_this, void 0, void 0, function () {
-        var result, err_7, errorMessage, newToken, result, refreshError_4, error_7;
+        var result, err_6, errorMessage, newToken, result, refreshError_4, error_6;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -458,8 +442,8 @@ export function TrainlyProvider(_a) {
                     result = _a.sent();
                     return [2 /*return*/, result];
                 case 2:
-                    err_7 = _a.sent();
-                    errorMessage = err_7 instanceof Error ? err_7.message : String(err_7);
+                    err_6 = _a.sent();
+                    errorMessage = err_6 instanceof Error ? err_6.message : String(err_6);
                     if (!(getToken &&
                         appId &&
                         (errorMessage.includes("401") ||
@@ -487,13 +471,13 @@ export function TrainlyProvider(_a) {
                     console.error("❌ Token refresh failed:", refreshError_4);
                     return [3 /*break*/, 9];
                 case 9:
-                    error_7 = {
+                    error_6 = {
                         code: "LIST_FILES_FAILED",
                         message: "Failed to list files",
-                        details: err_7,
+                        details: err_6,
                     };
-                    setError(error_7);
-                    throw error_7;
+                    setError(error_6);
+                    throw error_6;
                 case 10:
                     setIsLoading(false);
                     return [7 /*endfinally*/];
@@ -502,7 +486,7 @@ export function TrainlyProvider(_a) {
         });
     }); };
     var deleteFile = function (fileId) { return __awaiter(_this, void 0, void 0, function () {
-        var result, err_8, errorMessage, newToken, result, refreshError_5, error_8;
+        var result, err_7, errorMessage, newToken, result, refreshError_5, error_7;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -514,8 +498,8 @@ export function TrainlyProvider(_a) {
                     result = _a.sent();
                     return [2 /*return*/, result];
                 case 2:
-                    err_8 = _a.sent();
-                    errorMessage = err_8 instanceof Error ? err_8.message : String(err_8);
+                    err_7 = _a.sent();
+                    errorMessage = err_7 instanceof Error ? err_7.message : String(err_7);
                     if (!(getToken &&
                         appId &&
                         (errorMessage.includes("401") ||
@@ -543,13 +527,13 @@ export function TrainlyProvider(_a) {
                     console.error("❌ Token refresh failed:", refreshError_5);
                     return [3 /*break*/, 9];
                 case 9:
-                    error_8 = {
+                    error_7 = {
                         code: "DELETE_FILE_FAILED",
                         message: "Failed to delete file",
-                        details: err_8,
+                        details: err_7,
                     };
-                    setError(error_8);
-                    throw error_8;
+                    setError(error_7);
+                    throw error_7;
                 case 10:
                     setIsLoading(false);
                     return [7 /*endfinally*/];
@@ -558,7 +542,7 @@ export function TrainlyProvider(_a) {
         });
     }); };
     var sendMessage = function (content) { return __awaiter(_this, void 0, void 0, function () {
-        var userMessage, response, assistantMessage_1, err_9;
+        var userMessage, response, assistantMessage_1, err_8;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -585,9 +569,9 @@ export function TrainlyProvider(_a) {
                     setMessages(function (prev) { return __spreadArray(__spreadArray([], prev, true), [assistantMessage_1], false); });
                     return [3 /*break*/, 4];
                 case 3:
-                    err_9 = _a.sent();
+                    err_8 = _a.sent();
                     // Error is already set by askWithCitations
-                    console.error("Failed to send message:", err_9);
+                    console.error("Failed to send message:", err_8);
                     return [3 /*break*/, 4];
                 case 4: return [2 /*return*/];
             }
@@ -598,7 +582,7 @@ export function TrainlyProvider(_a) {
     var clearMessages = function () { return setMessages([]); };
     var value = {
         ask: ask,
-        askWithCitations: askWithCitations,
+        askWithCitations: askWithCitations, // Deprecated - kept for backward compatibility
         upload: upload,
         bulkUploadFiles: bulkUploadFiles, // NEW: Bulk file upload method
         listFiles: listFiles, // NEW: File management methods

package/dist/components/TrainlyChat.d.ts

@@ -5,10 +5,11 @@ export interface TrainlyChatProps {
     showCitations?: boolean;
     enableFileUpload?: boolean;
     theme?: "light" | "dark" | "auto";
+    scopeFilters?: Record<string, string | number | boolean>;
     onMessage?: (message: {
         role: "user" | "assistant";
         content: string;
     }) => void;
     onError?: (error: string) => void;
 }
-export declare function TrainlyChat({ height, className, placeholder, showCitations, enableFileUpload, theme, onMessage, onError, }: TrainlyChatProps): import("react/jsx-runtime").JSX.Element;
+export declare function TrainlyChat({ height, className, placeholder, showCitations, enableFileUpload, theme, scopeFilters, onMessage, onError, }: TrainlyChatProps): import("react/jsx-runtime").JSX.Element;

package/dist/components/TrainlyChat.js

@@ -35,14 +35,26 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
         if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
 };
+var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
+    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
+        if (ar || !(i in from)) {
+            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
+            ar[i] = from[i];
+        }
+    }
+    return to.concat(ar || Array.prototype.slice.call(from));
+};
 import { jsx as _jsx, jsxs as _jsxs, Fragment as _Fragment } from "react/jsx-runtime";
 import * as React from "react";
 import { useTrainly } from "../useTrainly";
 export function TrainlyChat(_a) {
     var _this = this;
-    var _b = _a.height, height = _b === void 0 ? "400px" : _b, _c = _a.className, className = _c === void 0 ? "" : _c, _d = _a.placeholder, placeholder = _d === void 0 ? "Ask me anything..." : _d, _e = _a.showCitations, showCitations = _e === void 0 ? true : _e, _f = _a.enableFileUpload, enableFileUpload = _f === void 0 ? true : _f, _g = _a.theme, theme = _g === void 0 ? "auto" : _g, onMessage = _a.onMessage, onError = _a.onError;
-    var _h = useTrainly(), messages = _h.messages, sendMessage = _h.sendMessage, isLoading = _h.isLoading, error = _h.error, clearError = _h.clearError;
-    var _j = React.useState(""), input = _j[0], setInput = _j[1];
+    var _b = _a.height, height = _b === void 0 ? "400px" : _b, _c = _a.className, className = _c === void 0 ? "" : _c, _d = _a.placeholder, placeholder = _d === void 0 ? "Ask me anything..." : _d, _e = _a.showCitations, showCitations = _e === void 0 ? true : _e, _f = _a.enableFileUpload, enableFileUpload = _f === void 0 ? true : _f, _g = _a.theme, theme = _g === void 0 ? "auto" : _g, scopeFilters = _a.scopeFilters, onMessage = _a.onMessage, onError = _a.onError;
+    var ask = useTrainly().ask;
+    var _h = React.useState(""), input = _h[0], setInput = _h[1];
+    var _j = React.useState([]), messages = _j[0], setMessages = _j[1];
+    var _k = React.useState(false), isLoading = _k[0], setIsLoading = _k[1];
+    var _l = React.useState(null), error = _l[0], setError = _l[1];
     var messagesEndRef = React.useRef(null);
     var fileInputRef = React.useRef(null);
     // Auto-scroll to bottom
@@ -56,8 +68,9 @@ export function TrainlyChat(_a) {
             onError(error.message);
         }
     }, [error, onError]);
+    var clearError = function () { return setError(null); };
     var handleSubmit = function (e) { return __awaiter(_this, void 0, void 0, function () {
-        var messageContent, err_1;
+        var messageContent, userMessage, response, assistantMessage_1, err_1, errorMsg;
         return __generator(this, function (_a) {
             switch (_a.label) {
                 case 0:
@@ -66,19 +79,43 @@ export function TrainlyChat(_a) {
                         return [2 /*return*/];
                     messageContent = input.trim();
                     setInput("");
+                    setError(null);
+                    userMessage = {
+                        id: Date.now().toString(),
+                        role: "user",
+                        content: messageContent,
+                    };
+                    setMessages(function (prev) { return __spreadArray(__spreadArray([], prev, true), [userMessage], false); });
+                    onMessage === null || onMessage === void 0 ? void 0 : onMessage(userMessage);
+                    setIsLoading(true);
                     _a.label = 1;
                 case 1:
-                    _a.trys.push([1, 3, , 4]);
-                    return [4 /*yield*/, sendMessage(messageContent)];
+                    _a.trys.push([1, 3, 4, 5]);
+                    return [4 /*yield*/, ask(messageContent, {
+                            includeCitations: showCitations,
+                            scope_filters: scopeFilters,
+                        })];
                 case 2:
-                    _a.sent();
-                    onMessage === null || onMessage === void 0 ? void 0 : onMessage({ role: "user", content: messageContent });
-                    return [3 /*break*/, 4];
+                    response = _a.sent();
+                    assistantMessage_1 = {
+                        id: (Date.now() + 1).toString(),
+                        role: "assistant",
+                        content: response.answer,
+                        citations: response.citations,
+                    };
+                    setMessages(function (prev) { return __spreadArray(__spreadArray([], prev, true), [assistantMessage_1], false); });
+                    onMessage === null || onMessage === void 0 ? void 0 : onMessage(assistantMessage_1);
+                    return [3 /*break*/, 5];
                 case 3:
                     err_1 = _a.sent();
+                    errorMsg = err_1 instanceof Error ? err_1.message : "Failed to send message";
+                    setError({ message: errorMsg });
                     console.error("Failed to send message:", err_1);
-                    return [3 /*break*/, 4];
-                case 4: return [2 /*return*/];
+                    return [3 /*break*/, 5];
+                case 4:
+                    setIsLoading(false);
+                    return [7 /*endfinally*/];
+                case 5: return [2 /*return*/];
             }
         });
     }); };

package/dist/types.d.ts

@@ -88,7 +88,13 @@ export interface TrainlyFileManagerProps {
     maxFileSize?: number;
 }
 export interface TrainlyContextValue {
-    ask: (question: string) => Promise<string>;
+    ask: (question: string, options?: {
+        includeCitations?: boolean;
+        scope_filters?: Record<string, string | number | boolean>;
+    }) => Promise<{
+        answer: string;
+        citations?: Citation[];
+    }>;
     askWithCitations: (question: string) => Promise<{
         answer: string;
         citations: Citation[];

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@trainly/react",
-  "version": "1.4.1",
+  "version": "1.4.3",
   "description": "Dead simple RAG integration for React apps with OAuth authentication and custom scopes",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",