changed llm case

kartikpersistent · kartikpersistent · commit 94cf7287fccc · 2024-06-24T15:35:21.000Z
diff --git a/backend/src/generate_graphDocuments_from_llm.py b/backend/src/generate_graphDocuments_from_llm.py
@@ -24,7 +24,7 @@ def generate_graphDocuments(model: str, graph: Neo4jGraph, chunkId_chunkDoc_list
     logging.info(f"allowedNodes: {allowedNodes}, allowedRelationship: {allowedRelationship}")
 
     graph_documents = []
-    if model == "Diffbot":
+    if model == "diffbot":
         graph_documents = get_graph_from_diffbot(graph, chunkId_chunkDoc_list)
 
     elif model in OPENAI_MODELS:
