From f105fd1b2c03359884e698b47ef07d82953be3a9 Mon Sep 17 00:00:00 2001
From: Alex <a@tushynski.me>
Date: Sun, 8 Sep 2024 23:19:10 +0100
Subject: [PATCH] lint: final fix

---
 frontend/src/components/SourceDropdown.tsx        | 2 +-
 frontend/src/conversation/conversationHandlers.ts | 2 +-
 frontend/src/preferences/SelectDocsModal.tsx      | 4 +---
 frontend/src/settings/Documents.tsx               | 6 ++----
 frontend/src/settings/index.tsx                   | 2 +-
 frontend/src/store.ts                             | 9 +++------
 frontend/src/upload/Upload.tsx                    | 4 ++--
 7 files changed, 11 insertions(+), 18 deletions(-)

diff --git a/frontend/src/components/SourceDropdown.tsx b/frontend/src/components/SourceDropdown.tsx
index e76b9664..d5146da5 100644
--- a/frontend/src/components/SourceDropdown.tsx
+++ b/frontend/src/components/SourceDropdown.tsx
@@ -1,7 +1,7 @@
 import React from 'react';
 import Trash from '../assets/trash.svg';
 import Arrow2 from '../assets/dropdown-arrow.svg';
-import { Doc } from '../preferences/preferenceApi';
+import { Doc } from '../models/misc';
 import { useDispatch } from 'react-redux';
 import { useTranslation } from 'react-i18next';
 type Props = {
diff --git a/frontend/src/conversation/conversationHandlers.ts b/frontend/src/conversation/conversationHandlers.ts
index eaea1b4f..335f07e9 100644
--- a/frontend/src/conversation/conversationHandlers.ts
+++ b/frontend/src/conversation/conversationHandlers.ts
@@ -1,5 +1,5 @@
 import conversationService from '../api/services/conversationService';
-import { Doc } from '../preferences/preferenceApi';
+import { Doc } from '../models/misc';
 import { Answer, FEEDBACK, RetrievalPayload } from './conversationModels';
 
 export function handleFetchAnswer(
diff --git a/frontend/src/preferences/SelectDocsModal.tsx b/frontend/src/preferences/SelectDocsModal.tsx
index f2aa4754..97d7626d 100644
--- a/frontend/src/preferences/SelectDocsModal.tsx
+++ b/frontend/src/preferences/SelectDocsModal.tsx
@@ -93,9 +93,7 @@ export default function APIKeyModal({
                             }}
                             className="h-10 w-full cursor-pointer border-x-2 border-b-2 hover:bg-gray-100"
                           >
-                            <p className="ml-5 py-3">
-                              {doc.name} {doc.version}
-                            </p>
+                            <p className="ml-5 py-3">{doc.name}</p>
                           </div>
                         );
                       }
diff --git a/frontend/src/settings/Documents.tsx b/frontend/src/settings/Documents.tsx
index 19bde1fa..342e2f52 100644
--- a/frontend/src/settings/Documents.tsx
+++ b/frontend/src/settings/Documents.tsx
@@ -61,12 +61,10 @@ const Documents: React.FC<DocumentsProps> = ({
                       {document.tokens ? formatTokens(+document.tokens) : ''}
                     </td>
                     <td className="border-r border-t px-4 py-2">
-                      {document.location === 'remote'
-                        ? 'Pre-loaded'
-                        : 'Private'}
+                      {document.type === 'remote' ? 'Pre-loaded' : 'Private'}
                     </td>
                     <td className="border-t px-4 py-2">
-                      {document.location !== 'remote' && (
+                      {document.type !== 'remote' && (
                         <img
                           src={Trash}
                           alt="Delete"
diff --git a/frontend/src/settings/index.tsx b/frontend/src/settings/index.tsx
index 141bd227..5bef3b43 100644
--- a/frontend/src/settings/index.tsx
+++ b/frontend/src/settings/index.tsx
@@ -6,7 +6,7 @@ import userService from '../api/services/userService';
 import ArrowLeft from '../assets/arrow-left.svg';
 import ArrowRight from '../assets/arrow-right.svg';
 import i18n from '../locale/i18n';
-import { Doc } from '../preferences/preferenceApi';
+import { Doc } from '../models/misc';
 import {
   selectSourceDocs,
   setSourceDocs,
diff --git a/frontend/src/store.ts b/frontend/src/store.ts
index 3d1408b3..a085dad3 100644
--- a/frontend/src/store.ts
+++ b/frontend/src/store.ts
@@ -26,15 +26,12 @@ const store = configureStore({
       conversations: null,
       sourceDocs: [
         {
-          location: '',
-          language: '',
           name: 'default',
-          version: '',
           date: '',
-          description: '',
-          docLink: '',
-          fullName: '',
           model: '1.0',
+          type: 'remote',
+          id: 'default',
+          retriever: 'clasic',
         },
       ],
       modalState: 'INACTIVE',
diff --git a/frontend/src/upload/Upload.tsx b/frontend/src/upload/Upload.tsx
index 5d5d1ac5..7681ce34 100644
--- a/frontend/src/upload/Upload.tsx
+++ b/frontend/src/upload/Upload.tsx
@@ -120,7 +120,7 @@ function Upload({
                     dispatch(setSourceDocs(data));
                     dispatch(
                       setSelectedDocs(
-                        data?.find((d) => d.location.toLowerCase() === 'local'),
+                        data?.find((d) => d.type?.toLowerCase() === 'local'),
                       ),
                     );
                   });
@@ -137,7 +137,7 @@ function Upload({
                     dispatch(setSourceDocs(data));
                     dispatch(
                       setSelectedDocs(
-                        data?.find((d) => d.location.toLowerCase() === 'local'),
+                        data?.find((d) => d.type?.toLowerCase() === 'local'),
                       ),
                     );
                   });