@rimori/react-client 0.4.8-next.0 → 0.4.8-next.1

package/dist/components/ai/Avatar.d.ts

@@ -9,6 +9,7 @@ interface Props {
     children?: React.ReactNode;
     autoStartConversation?: FirstMessages;
     className?: string;
+    knowledgeId?: string;
 }
-export declare function Avatar({ avatarImageUrl, voiceId, agentTools, autoStartConversation, children, circleSize, className, cache, }: Props): import("react/jsx-runtime").JSX.Element;
+export declare function Avatar({ avatarImageUrl, voiceId, agentTools, autoStartConversation, children, circleSize, className, cache, knowledgeId, }: Props): import("react/jsx-runtime").JSX.Element;
 export {};

package/dist/components/ai/Avatar.js

@@ -7,13 +7,13 @@ import { useChat } from '../../hooks/UseChatHook';
 import { useRimori } from '../../providers/PluginProvider';
 import { getFirstMessages } from './utils';
 import { useTheme } from '../../hooks/ThemeSetter';
-export function Avatar({ avatarImageUrl, voiceId, agentTools, autoStartConversation, children, circleSize = '300px', className, cache = false, }) {
+export function Avatar({ avatarImageUrl, voiceId, agentTools, autoStartConversation, children, circleSize = '300px', className, cache = false, knowledgeId, }) {
     const { ai, event, plugin } = useRimori();
     const { isDark: isDarkThemeValue } = useTheme(plugin.theme);
     const [agentReplying, setAgentReplying] = useState(false);
     const [isProcessingMessage, setIsProcessingMessage] = useState(false);
     const sender = useMemo(() => new MessageSender((...args) => ai.getVoice(...args), voiceId, cache), [voiceId, ai, cache]);
-    const { messages, append, isLoading, lastMessage, setMessages } = useChat(agentTools);
+    const { messages, append, isLoading, lastMessage, setMessages } = useChat(agentTools, { knowledgeId });
     useEffect(() => {
         console.log('messages', messages);
     }, [messages]);

package/dist/hooks/UseChatHook.d.ts

@@ -1,7 +1,9 @@
 import React from 'react';
 import { Tool } from '@rimori/client';
 import { Message } from '@rimori/client';
-export declare function useChat(tools?: Tool[]): {
+export declare function useChat(tools?: Tool[], options?: {
+    knowledgeId?: string | (() => string | undefined);
+}): {
     messages: Message[];
     append: (appendMessages: Message[]) => void;
     isLoading: boolean;

package/dist/hooks/UseChatHook.js

@@ -1,12 +1,13 @@
 import React from 'react';
 import { useRimori } from '../providers/PluginProvider';
-export function useChat(tools) {
+export function useChat(tools, options) {
     const [messages, setMessages] = React.useState([]);
     const [isLoading, setIsLoading] = React.useState(false);
     const { ai } = useRimori();
     const append = (appendMessages) => {
         const allMessages = [...messages, ...appendMessages];
         setMessages(allMessages);
+        const knowledgeId = typeof (options === null || options === void 0 ? void 0 : options.knowledgeId) === 'function' ? options.knowledgeId() : options === null || options === void 0 ? void 0 : options.knowledgeId;
         ai.getSteamedText(allMessages, (id, message, finished, toolInvocations) => {
             const lastMessage = messages[messages.length - 1];
             setIsLoading(!finished);
@@ -17,7 +18,7 @@ export function useChat(tools) {
             else {
                 setMessages([...allMessages, { id, role: 'assistant', content: message, toolCalls: toolInvocations }]);
             }
-        }, tools);
+        }, tools, false, undefined, knowledgeId);
     };
     return {
         messages,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@rimori/react-client",
-  "version": "0.4.8-next.0",
+  "version": "0.4.8-next.1",
   "license": "Apache-2.0",
   "repository": {
     "type": "git",
@@ -23,7 +23,7 @@
     "format": "prettier --write ."
   },
   "peerDependencies": {
-    "@rimori/client": "^2.5.13",
+    "@rimori/client": "2.5.14-next.3",
     "react": "^18.1.0",
     "react-dom": "^18.1.0"
   },
@@ -34,7 +34,7 @@
   },
   "devDependencies": {
     "@eslint/js": "^9.37.0",
-    "@rimori/client": "^2.5.13",
+    "@rimori/client": "2.5.14-next.3",
     "@types/react": "^18.3.21",
     "eslint-config-prettier": "^10.1.8",
     "eslint-plugin-prettier": "^5.5.4",

package/src/components/ai/Avatar.tsx

@@ -18,6 +18,7 @@ interface Props {
   children?: React.ReactNode;
   autoStartConversation?: FirstMessages;
   className?: string;
+  knowledgeId?: string;
 }
 
 export function Avatar({
@@ -29,6 +30,7 @@ export function Avatar({
   circleSize = '300px',
   className,
   cache = false,
+  knowledgeId,
 }: Props) {
   const { ai, event, plugin } = useRimori();
   const { isDark: isDarkThemeValue } = useTheme(plugin.theme);
@@ -38,7 +40,7 @@ export function Avatar({
     () => new MessageSender((...args) => ai.getVoice(...args), voiceId, cache),
     [voiceId, ai, cache],
   );
-  const { messages, append, isLoading, lastMessage, setMessages } = useChat(agentTools);
+  const { messages, append, isLoading, lastMessage, setMessages } = useChat(agentTools, { knowledgeId });
 
   useEffect(() => {
     console.log('messages', messages);

package/src/hooks/UseChatHook.ts

@@ -3,7 +3,7 @@ import { Tool } from '@rimori/client';
 import { useRimori } from '../providers/PluginProvider';
 import { Message, ToolInvocation } from '@rimori/client';
 
-export function useChat(tools?: Tool[]) {
+export function useChat(tools?: Tool[], options?: { knowledgeId?: string | (() => string | undefined) }) {
   const [messages, setMessages] = React.useState<Message[]>([]);
   const [isLoading, setIsLoading] = React.useState(false);
   const { ai } = useRimori();
@@ -11,6 +11,7 @@ export function useChat(tools?: Tool[]) {
   const append = (appendMessages: Message[]) => {
     const allMessages = [...messages, ...appendMessages];
     setMessages(allMessages);
+    const knowledgeId = typeof options?.knowledgeId === 'function' ? options.knowledgeId() : options?.knowledgeId;
     ai.getSteamedText(
       allMessages,
       (id, message, finished: boolean, toolInvocations?: ToolInvocation[]) => {
@@ -25,6 +26,9 @@ export function useChat(tools?: Tool[]) {
         }
       },
       tools,
+      false,
+      undefined,
+      knowledgeId,
     );
   };