all models

Author: prakriti-solankey

backend/example.env

@@ -48,4 +48,4 @@ CHUNKS_TO_BE_CREATED="50"
 BEDROCK_EMBEDDING_MODEL="model_name,aws_access_key,aws_secret_key,region_name"                       #model_name="amazon.titan-embed-text-v1"
 LLM_MODEL_CONFIG_bedrock_nova_micro_v1="model_name,aws_access_key,aws_secret_key,region_name"        #model_name="amazon.nova-micro-v1:0"
 LLM_MODEL_CONFIG_bedrock_nova_lite_v1="model_name,aws_access_key,aws_secret_key,region_name"         #model_name="amazon.nova-lite-v1:0"
-LLM_MODEL_CONFIG_bedrock_nova_pro_v1="model_name,aws_access_key,aws_secret_key,region_name"          #model_name="amazon.nova-pro-v1:0"
+LLM_MODEL_CONFIG_bedrock_nova_pro_v1="model_name,aws_access_key,aws_secret_key,region_name"          #model_name="amazon.nova-pro-v1:0"

backend/score.py

@@ -561,4 +561,4 @@ def get_nodelabels_relationships(self):
             return node_labels,relationship_types
         except Exception as e:
             print(f"Error in getting node labels/relationship types from db: {e}")
-            return []
+            return []

backend/src/graphDB_dataAccess.py

@@ -361,7 +361,6 @@ async def processing_source(uri, userName, password, database, model, file_name,
 
       logging.info('Update the status as Processing')
       update_graph_chunk_processed = int(os.environ.get('UPDATE_GRAPH_CHUNKS_PROCESSED'))
-      
       # selected_chunks = []
       is_cancelled_status = False
       job_status = "Completed"

backend/src/main.py

@@ -11,7 +11,6 @@
 from src.graphDB_dataAccess import graphDBdataAccess
 import time 
 
-
 DROP_INDEX_QUERY = "DROP INDEX entities IF EXISTS;"
 LABELS_QUERY = "CALL db.labels()"
 FULL_TEXT_QUERY = "CREATE FULLTEXT INDEX entities FOR (n{labels_str}) ON EACH [n.id, n.description];"

backend/src/post_processing.py

@@ -133,7 +133,7 @@ def create_gcs_bucket_folder_name_hashed(uri, file_name):
 
 def formatted_time(current_time):
   formatted_time = current_time.strftime('%Y-%m-%d %H:%M:%S %Z')
-  return formatted_time
+  return str(formatted_time)
 
 def last_url_segment(url):
   parsed_url = urlparse(url)
@@ -173,4 +173,4 @@ def get_bedrock_embeddings():
        return bedrock_embeddings
    except Exception as e:
        print(f"An unexpected error occurred: {e}")
-       raise
+       raise

backend/src/shared/common_fn.py

@@ -1,7 +1,45 @@
 import axios from 'axios';
 import { url } from '../utils/Utils';
+import { UserCredentials } from '../types';
 
 const api = axios.create({
   baseURL: url(),
+  data: {},
 });
+
+export const createDefaultFormData = (userCredentials: UserCredentials) => {
+  const formData = new FormData();
+  formData.append('uri', userCredentials?.uri ?? '');
+  formData.append('database', userCredentials?.database ?? '');
+  formData.append('userName', userCredentials?.userName ?? '');
+  formData.append('password', userCredentials?.password ?? '');
+  formData.append('email', userCredentials?.email ?? '');
+  api.interceptors.request.use(
+    (config) => {
+      if (config.data instanceof FormData) {
+        for (const [key, value] of formData.entries()) {
+          if (!config.data.has(key)) {
+            config.data.append(key, value);
+          }
+        }
+      } else {
+        const formData = new FormData();
+        for (const [key, value] of formData.entries()) {
+          formData.append(key, value);
+        }
+        for (const [key, value] of Object.entries(config.data || {})) {
+          formData.append(key, value as any);
+        }
+        config.data = formData;
+      }
+
+      return config;
+    },
+    (error) => {
+      return Promise.reject(error);
+    }
+  );
+  return formData;
+};
+
 export default api;

frontend/src/API/Index.ts

@@ -32,6 +32,7 @@ const ChatContent: React.FC<ChatProps> = ({ chatMessages }) => {
     const encodedPassword = urlParams.get('password');
     const database = urlParams.get('database');
     const port = urlParams.get('port');
+    const email = urlParams.get('email');
     const openModal = urlParams.get('open') === 'true';
     if (openModal || !(uri && user && encodedPassword && database && port)) {
       setOpenConnection((prev) => ({ ...prev, openPopUp: true }));
@@ -42,6 +43,7 @@ const ChatContent: React.FC<ChatProps> = ({ chatMessages }) => {
         password: atob(atob(encodedPassword)),
         database,
         port,
+        email: email ?? '',
       };
       setShowBackButton();
       setUserCredentials(credentialsForAPI);

frontend/src/components/ChatBot/ChatOnlyComponent.tsx

@@ -23,12 +23,10 @@ import {
   ExtendedRelationship,
   Messages,
   ResponseMode,
-  UserCredentials,
   metricstate,
   multimodelmetric,
   nodeDetailsProps,
 } from '../../types';
-import { useCredentials } from '../../context/UserCredentials';
 import { chatBotAPI } from '../../services/QnaAPI';
 import { v4 as uuidv4 } from 'uuid';
 import { useFileContext } from '../../context/UsersFiles';
@@ -63,7 +61,6 @@ const Chatbot: FC<ChatbotProps> = (props) => {
   } = props;
   const [inputMessage, setInputMessage] = useState('');
   const [loading, setLoading] = useState<boolean>(isLoading);
-  const { userCredentials } = useCredentials();
   const { model, chatModes, selectedRows, filesData } = useFileContext();
   const messagesEndRef = useRef<HTMLDivElement>(null);
   const [showInfoModal, setShowInfoModal] = useState<boolean>(false);
@@ -215,7 +212,6 @@ const Chatbot: FC<ChatbotProps> = (props) => {
     try {
       const apiCalls = chatModes.map((mode) =>
         chatBotAPI(
-          userCredentials as UserCredentials,
           inputMessage,
           sessionId,
           model,

frontend/src/components/ChatBot/Chatbot.tsx

@@ -1,5 +1,5 @@
 import { FC, useContext, useState } from 'react';
-import { ChunkProps, UserCredentials } from '../../types';
+import { ChunkProps } from '../../types';
 import { LoadingSpinner, TextLink, Typography } from '@neo4j-ndl/react';
 import { DocumentTextIconOutline, GlobeAltIconOutline } from '@neo4j-ndl/react/icons';
 import wikipedialogo from '../../assets/images/wikipedia.svg';
@@ -10,13 +10,11 @@ import ReactMarkdown from 'react-markdown';
 import { generateYouTubeLink, getLogo, isAllowedHost } from '../../utils/Utils';
 import { ThemeWrapperContext } from '../../context/ThemeWrapper';
 import { chatModeLables } from '../../utils/Constants';
-import { useCredentials } from '../../context/UserCredentials';
 import GraphViewModal from '../Graph/GraphViewModal';
 import { handleGraphNodeClick } from './chatInfo';
 
 const ChunkInfo: FC<ChunkProps> = ({ loading, chunks, mode }) => {
   const themeUtils = useContext(ThemeWrapperContext);
-  const { userCredentials } = useCredentials();
   const [neoNodes, setNeoNodes] = useState<any[]>([]);
   const [neoRels, setNeoRels] = useState<any[]>([]);
   const [openGraphView, setOpenGraphView] = useState(false);
@@ -25,7 +23,6 @@ const ChunkInfo: FC<ChunkProps> = ({ loading, chunks, mode }) => {
 
   const handleChunkClick = (elementId: string, viewMode: string) => {
     handleGraphNodeClick(
-      userCredentials as UserCredentials,
       elementId,
       viewMode,
       setNeoNodes,