fix: ai response being stored in different format and not displayed c…

…orrectly

Co-authored-by: Lukas Varga <[email protected]>
Signed-off-by: eloinoel <[email protected]>

src/frontend/components/RenderChat/index.tsx

@@ -7,21 +7,30 @@ import { useActiveChatId, useGetChat } from 'src/frontend/hooks';
 import type { conversationMessage } from 'src/frontend/types';
 import { Style } from './style';
 
+
+
 export function RenderChat() {
   const { activeChatId } = useActiveChatId();
   const { chat, status } = useGetChat(activeChatId);
   const { colors } = useTheme();
   const { data: user } = useUser();
 
   if (status === 'loading') return <ActivityIndicator />;
-
+  let id = 0;
   return (
     <>
+      {/* biome-ignore lint/suspicious/noExplicitAny: <explanation> */}
       {chat?.conversation.map((item: conversationMessage) => {
         const { type, message } = item;
+        let AIResponse = ""
+        if(type === 'AI') {
+          const entries = Object.entries(message)
+          if(entries.length > 0)
+            AIResponse = entries[0][1];
+        }
         return (
           <View
-            key={message}
+            key={type === 'USER' ? (message as string) + id++ : AIResponse + id++}
             style={[Style.bubble, { backgroundColor: colors.surfaceVariant, marginBottom: 16 }]}
           >
             <View style={{ flexDirection: 'row', alignItems: 'center' }}>
@@ -34,7 +43,7 @@ export function RenderChat() {
                 {type === 'AI' ? 'AiLixir' : user?.displayName || 'User'}
               </Text>
             </View>
-            <Markdown>{message}</Markdown>
+            <Markdown>{type === 'AI' ? AIResponse : message as string}</Markdown>
           </View>
         );
       })}

src/frontend/screens/ChatUI/index.tsx

@@ -112,11 +112,12 @@ export function ChatUI() {
 
   useEffect(() => {
     const create = async () => {
-      const { id } = await createChat({
+      const result = await createChat({
         title: 'NewChat',
         model: ['gpt-4'],
         conversation: []
       });
+      const id = result?.id
       setActiveChatId(id || '');
     };
     if (activeChatId === 'default') create();

src/frontend/types/index.d.ts

@@ -5,7 +5,10 @@ export type LLM = {
   active: boolean;
 };
 
-export type conversationMessage = { [key: string]: string };
+export type conversationMessage = {
+  type: 'USER' | 'AI';
+  message: string | { [key: string]: string };
+};
 
 export type Chat = {
   id?: string;