graphrag

graphrag.py

@@ -44,9 +44,9 @@ Revenue: How do you make money?
 question=f"Create marketing campaign that can improve customer acquisition, activation, retention and referral for this persona: {customer}"
 def knowledge_graph(text):
     documents = [Document(page_content=text)]
-    llm_transformer_filtered = LLMGraphTransformer(llm=llm,
+    llm_transformer_filtered = LLMGraphTransformer(llm=llm)
     # allowed_nodes=["Need", "Issue", "Product"],
-    # allowed_relationships=["WANT", "WITH", "USING", "RECOMMEND"],)
+    # allowed_relationships=["WANT", "WITH", "USING", "RECOMMEND"]
     graph_documents_filtered = llm_transformer_filtered.convert_to_graph_documents(documents)
     graph = NetworkxEntityGraph()