react-native-chatbot-ai 0.0.6 → 0.1.1

package/src/context/ChatContext.tsx

@@ -5,14 +5,15 @@ import ChatBotAI from '../components/chat';
 export const ChatContext = createContext<ChatContextType>({
   apiAddress: '',
   userId: '',
+  cartButton: undefined,
 });
 
 export const useChatContext = () => useContext(ChatContext);
 
 export const ChatProvider = (props: ChatProviderProps) => {
-  const { apiAddress, userId } = props;
+  const { apiAddress, userId, cartButton } = props;
   return (
-    <ChatContext.Provider value={{ apiAddress, userId }}>
+    <ChatContext.Provider value={{ apiAddress, userId, cartButton }}>
       <ChatBotAI />
     </ChatContext.Provider>
   );

package/src/hooks/message/useMessage.ts

@@ -17,19 +17,21 @@ const initialState = {
   messages: [],
 };
 export const useMessage = () => {
-  const sessionId = useSessionStore(state => state.sessionId);
+  const sessionId = useSessionStore((state) => state.sessionId);
   const { mutateAsync: getSessionById } = useFetchSessionById();
-  const [loadState, setLoadState] = useState<SessionDetailResponse>(initialState);
+  const [loadState, setLoadState] =
+    useState<SessionDetailResponse>(initialState);
 
   const onGetSessionById = useCallback(async () => {
     if (!sessionId) return;
     const res = await getSessionById(sessionId);
     if (!res) return;
-    setLoadState(prev => ({
+    setLoadState((prev) => ({
       ...res,
-      messages: res?.messages?.length > 0 ? res.messages.reverse() : prev.messages,
+      messages:
+        res?.messages?.length > 0 ? res.messages.reverse() : prev.messages,
     }));
-  }, [sessionId]);
+  }, [sessionId, getSessionById]);
 
   useEffect(() => {
     if (!sessionId) {
@@ -37,13 +39,13 @@ export const useMessage = () => {
       return;
     }
     onGetSessionById();
-  }, [sessionId]);
+  }, [sessionId, onGetSessionById]);
 
   useEffect(() => {
     const subOneMessage = DeviceEventEmitter.addListener(
       events.updateOneMessage,
       (messageItem: IMessageItem) => {
-        setLoadState(prev => ({
+        setLoadState((prev) => ({
           ...prev,
           messages: [messageItem, ...prev.messages],
         }));
@@ -52,9 +54,11 @@ export const useMessage = () => {
     const subMultipleMessage = DeviceEventEmitter.addListener(
       events.updateMultipleMessage,
       (messageItems: IMessageItem[]) => {
-        setLoadState(prev => {
-          const existingIds = new Set(prev.messages.map(m => m?.id));
-          const filtered = messageItems.filter(m => !existingIds.has(m?.id)).reverse();
+        setLoadState((prev) => {
+          const existingIds = new Set(prev.messages.map((m) => m?.id));
+          const filtered = messageItems
+            .filter((m) => !existingIds.has(m?.id))
+            .reverse();
           console.log('🚀 ~ useEffect ~ filtered:', prev.messages);
           return {
             ...prev,
@@ -66,14 +70,14 @@ export const useMessage = () => {
     const subForceUpdateMessages = DeviceEventEmitter.addListener(
       events.forceUpdateMessages,
       (messages: IMessageItem[]) => {
-        setLoadState(prev => {
-          const merged = prev.messages.map(msg => {
-            const update = messages.find(u => u.id === msg.id);
+        setLoadState((prev) => {
+          const merged = prev.messages.map((msg) => {
+            const update = messages.find((u) => u.id === msg.id);
             return update ? { ...msg, ...update } : msg;
           });
 
-          const existingIds = new Set(prev.messages.map(m => m.id));
-          const newItems = messages.filter(u => !existingIds.has(u.id));
+          const existingIds = new Set(prev.messages.map((m) => m.id));
+          const newItems = messages.filter((u) => !existingIds.has(u.id));
           return {
             ...prev,
             messages: [...newItems, ...merged],

package/src/hooks/message/useSendMessage.ts

@@ -48,7 +48,7 @@ export const useSendMessage = () => {
         latestSessionId
       );
     },
-    [startStream]
+    [startStream, createSession]
   );
 
   return {

package/src/hooks/message/useStreamMessage.ts

@@ -10,12 +10,14 @@ import { events } from '../../constants/events';
 import useSessionStore from '../../store/session';
 
 export const useStreamMessage = () => {
-  const sessionId = useSessionStore(state => state.sessionId);
+  const sessionId = useSessionStore((state) => state.sessionId);
   const esRef = useRef<EventSource | null>(null);
   const sessionIdRef = useRef<string | undefined>(sessionId);
   const { apiAddress } = useChatContext();
-  const setStreamMessage = useStreamMessageStore(state => state.setStreamMessage);
-  const setIsStreaming = useStreamMessageStore(state => state.setIsStreaming);
+  const setStreamMessage = useStreamMessageStore(
+    (state) => state.setStreamMessage
+  );
+  const setIsStreaming = useStreamMessageStore((state) => state.setIsStreaming);
 
   useEffect(() => {
     sessionIdRef.current = sessionId;
@@ -39,7 +41,7 @@ export const useStreamMessage = () => {
         method: 'POST',
         headers: {
           'Content-Type': 'application/json',
-          Authorization: `Bearer ${token}`,
+          'Authorization': `Bearer ${token}`,
         },
         body: JSON.stringify(payload),
       });
@@ -56,7 +58,7 @@ export const useStreamMessage = () => {
         () => {
           DeviceEventEmitter.emit(
             events.updateMultipleMessage,
-            Object.values(buffers).map(m => ({ ...m, content: '' }))
+            Object.values(buffers).map((m) => ({ ...m, content: '' }))
           );
           setStreamMessage(buffers);
         },
@@ -64,11 +66,15 @@ export const useStreamMessage = () => {
         { leading: true, trailing: true }
       );
 
-      es.addEventListener('message', event => {
-        console.log('📩 Stream message:', typeof event.data, JSON?.parse?.(event?.data || '{}'));
+      es.addEventListener('message', (event) => {
+        console.log(
+          '📩 Stream message:',
+          typeof event.data,
+          JSON?.parse?.(event?.data || '{}')
+        );
         try {
           const data = JSON?.parse?.(event?.data || '{}');
-          if (!!data?.is_done) {
+          if (data?.is_done) {
             console.log('✅ Stream done');
             es.close();
             setIsStreaming(false);
@@ -78,7 +84,10 @@ export const useStreamMessage = () => {
             if (buffers?.[message?.id]) {
               buffers[message?.id] = {
                 ...message,
-                content: [buffers?.[message?.id]?.content || '', message?.content || ''].join(''),
+                content: [
+                  buffers?.[message?.id]?.content || '',
+                  message?.content || '',
+                ].join(''),
               };
             } else {
               buffers[message?.id] = message;
@@ -91,21 +100,24 @@ export const useStreamMessage = () => {
         }
       });
 
-      es.addEventListener('error', err => {
+      es.addEventListener('error', (err) => {
         console.log('❌ Stream error:', err);
         setIsStreaming(false);
       });
 
       es.addEventListener('close', () => {
         console.log('🔴 Stream closed');
-        DeviceEventEmitter.emit(events.forceUpdateMessages, Object.values(buffers));
+        DeviceEventEmitter.emit(
+          events.forceUpdateMessages,
+          Object.values(buffers)
+        );
         buffers = {};
         setIsStreaming(false);
       });
 
       esRef.current = es;
     },
-    [apiAddress]
+    [apiAddress, setIsStreaming, setStreamMessage]
   );
 
   const stopStream = useCallback(() => {
@@ -115,7 +127,7 @@ export const useStreamMessage = () => {
       esRef.current = null;
       setIsStreaming(false);
     }
-  }, []);
+  }, [setIsStreaming]);
 
   useEffect(() => {
     return () => {

package/src/services/endpoints.ts

@@ -1,7 +1,8 @@
 export const ENDPOINTS = {
   assistantService: {
     getSuggestions: '/assistant-service/v1/app/chat/prompt-suggestions',
-    getSessionById: (sessionId: string) => `/assistant-service/v1/app/chat/sessions/${sessionId}`,
+    getSessionById: (sessionId: string) =>
+      `/assistant-service/v1/app/chat/sessions/${sessionId}`,
     getStreamMessage: (sessionId: string) =>
       `/assistant-service/v1/app/chat/sessions/${sessionId}/stream`,
     createSession: '/assistant-service/v1/app/chat/sessions',

package/src/store/session.ts

@@ -1,9 +1,9 @@
 import { create } from 'zustand';
 import { SessionStore } from '../types/chat';
 
-const useSessionStore = create<SessionStore>(set => ({
+const useSessionStore = create<SessionStore>((set) => ({
   sessionId: undefined,
-  setSessionId: sessionId => set({ sessionId }),
+  setSessionId: (sessionId) => set({ sessionId }),
 }));
 
 export default useSessionStore;

package/src/store/streamMessage.ts

@@ -2,14 +2,14 @@ import { create } from 'zustand';
 import { StreamMessageStore } from '../types/chat';
 import { IMessageItem } from '../types';
 
-const useStreamMessageStore = create<StreamMessageStore>(set => ({
+const useStreamMessageStore = create<StreamMessageStore>((set) => ({
   isStreaming: false,
-  setIsStreaming: isStreaming => set({ isStreaming }),
+  setIsStreaming: (isStreaming) => set({ isStreaming }),
   streamMessage: {},
   setStreamMessage: (streamMessage: Record<string, IMessageItem>) => {
     if (!streamMessage || Object.keys(streamMessage).length === 0) return;
 
-    set(state => {
+    set((state) => {
       const updated = { ...state.streamMessage };
 
       for (const [id, msg] of Object.entries(streamMessage)) {

package/src/types/chat.ts

@@ -1,13 +1,16 @@
+import { JSX } from 'react';
 import { IMessageItem } from './dto';
 
 export interface ChatContextType {
   apiAddress: string;
   userId: string;
+  cartButton?: JSX.Element;
 }
 
 export interface ChatProviderProps {
   apiAddress: string;
   userId: string;
+  cartButton?: JSX.Element;
 }
 
 export interface SessionStore {

package/src/types/dto.ts

@@ -44,7 +44,12 @@ export interface IAttachment {
   name?: string;
   size?: number;
   loading?: boolean;
-  mime_type: 'image/jpeg' | 'image/png' | 'image/webp' | 'image/jpg' | 'application/pdf';
+  mime_type:
+    | 'image/jpeg'
+    | 'image/png'
+    | 'image/webp'
+    | 'image/jpg'
+    | 'application/pdf';
 }
 
 export interface StreamMessageRequest {