Gcs response changes (#332)

* gcs bucket scan api response updated

* gpt 4o addition

Author: aashipandya

backend/src/QA_integration.py

@@ -82,7 +82,8 @@ def get_llm(model: str,max_tokens=1000) -> Any:
         "Gemini Pro": "gemini-1.0-pro-001",
         "Gemini 1.5 Pro": "gemini-1.5-pro-preview-0409",
         "OpenAI GPT 4": "gpt-4-0125-preview",
-        "Diffbot" : "gpt-4-0125-preview"
+        "Diffbot" : "gpt-4-0125-preview",
+        "OpenAI GPT 4o":"gpt-4o"
          }
 
     if model in model_versions:

backend/src/generate_graphDocuments_from_llm.py

@@ -29,16 +29,16 @@ def generate_graphDocuments(model: str, graph: Neo4jGraph, chunkId_chunkDoc_list
         model_version = "gpt-3.5-turbo-16k"
         graph_documents = get_graph_from_OpenAI(model_version, graph, chunkId_chunkDoc_list, allowedNodes, allowedRelationship)
 
-    elif model == "OpenAI GPT 4":
-        model_version = "gpt-4-0125-preview"
+    elif model == "OpenAI GPT 4o":
+        model_version = "gpt-4o"
         graph_documents = get_graph_from_OpenAI(model_version, graph, chunkId_chunkDoc_list, allowedNodes, allowedRelationship)
 
     elif model == "Gemini 1.0 Pro" :
         model_version = "gemini-1.0-pro-001"
         graph_documents = get_graph_from_Gemini(model_version, graph, chunkId_chunkDoc_list, allowedNodes, allowedRelationship)
 
     elif model == "Gemini 1.5 Pro" :
-        model_version = "gemini-1.5-pro-preview-0409"
+        model_version = "gemini-1.5-pro-preview-0514"
         graph_documents = get_graph_from_Gemini(model_version, graph, chunkId_chunkDoc_list, allowedNodes, allowedRelationship)
 
     logging.info(f"graph_documents = {len(graph_documents)}")

frontend/src/utils/Constants.ts

@@ -28,7 +28,7 @@ export const docChunkEntities = `+[chunks]
 export const llms =
   process.env?.LLM_MODELS?.trim() != ''
     ? process.env.LLM_MODELS?.split(',')
-    : ['Diffbot', 'Gemini 1.0 Pro', 'OpenAI GPT 3.5', 'OpenAI GPT 4', 'Gemini 1.5 Pro'];
+    : ['Diffbot', 'Gemini 1.0 Pro', 'OpenAI GPT 3.5', 'OpenAI GPT 4o', 'Gemini 1.5 Pro'];
 export const defaultLLM = llms?.includes('OpenAI GPT 3.5')
   ? 'OpenAI GPT 3.5'
   : llms?.includes('Gemini 1.0 Pro')