test

audit_page/knowledge_graph.py

@@ -93,7 +93,9 @@ def convert_neo4j_to_agraph(neo4j_graph, node_colors):
         shape = "circle"  # Default shape, can be customized
         
         # For example purposes, no images are added, but you can set 'image' if needed.
-        nodes.append(Node(id=node_id,title=type, label=label, size=size, shape=shape,color=node_colors[type]))
+        new_node = Node(id=node_id,title=type, label=label, size=size, shape=shape,color=node_colors[type])
+        if not if_node_exists(nodes, new_node.id):
+            nodes.append(new_node)
 
     # Creating Agraph edges
     for relationship in neo4j_graph.relationships:

utils/kg/construct_kg.py

@@ -5,7 +5,7 @@ from langchain_core.documents import Document
 
 def get_graph(text,allowed_nodes=None,prompt=None):
     
-    llm = ChatOpenAI(temperature=0, model_name="gpt-4o")
+    llm = ChatOpenAI(temperature=0, model_name="gpt-4o-2024-08-06")
 
     if allowed_nodes:
         llm_transformer =   LLMGraphTransformer(llm=llm,allowed_nodes=allowed_nodes)