Skip to content

Commit 73aa5e5

Browse files
antejavorantejavor
committed
Update.
1 parent d9c25f6 commit 73aa5e5

File tree

1 file changed

+70
-38
lines changed

1 file changed

+70
-38
lines changed

‎memgraph-graphRAG/graphRAG.ipynb‎

Lines changed: 70 additions & 38 deletions
Original file line numberDiff line numberDiff line change
@@ -30,10 +30,29 @@
3030
"this by running the following command: \n",
3131
"\n",
3232
"TODO: Updated the command when the vector search is available in the official\n",
33-
"Memgraph docker image\n",
34-
"```bash\n",
35-
"docker run -p 7687:7687 -p 7444:7444 memgraph/memgraph-mage:exp-vector-1 --log-level=TRACE --also-log-to-stderr --telemetry-enabled=False --experimental-vector-indexes='tag__Entity__embedding__{\"dimension\":384,\"limit\":3000}'\n",
36-
"```\n",
33+
"Memgraph docker image\n"
34+
]
35+
},
36+
{
37+
"cell_type": "code",
38+
"execution_count": null,
39+
"metadata": {
40+
"vscode": {
41+
"languageId": "shellscript"
42+
}
43+
},
44+
"outputs": [],
45+
"source": [
46+
"\n",
47+
"\n",
48+
"docker run -p 7687:7687 -p 7444:7444 memgraph/memgraph-mage:exp-vector-1 --log-level=TRACE --also-log-to-stderr --telemetry-enabled=False --experimental-vector-indexes='tag__Entity__embedding__{\"dimension\":384,\"limit\":3000}'\n"
49+
]
50+
},
51+
{
52+
"cell_type": "markdown",
53+
"metadata": {},
54+
"source": [
55+
"\n",
3756
"\n",
3857
"You can run this command outside of this notebook. \n",
3958
"\n",
@@ -46,14 +65,13 @@
4665
"execution_count": null,
4766
"metadata": {
4867
"vscode": {
49-
"languageId": "markdown"
68+
"languageId": "shellscript"
5069
}
5170
},
5271
"outputs": [],
5372
"source": [
54-
"```bash\n",
55-
"cat ./data/memgraph-export-got.cypherl | docker run -i memgraph/mgconsole --host=localhost\n",
56-
"``` "
73+
"\n",
74+
"cat ./data/memgraph-export-got.cypherl | docker run -i memgraph/mgconsole --host=localhost\n"
5775
]
5876
},
5977
{
@@ -355,13 +373,6 @@
355373
"7. Ask LLM the question with the relevant data "
356374
]
357375
},
358-
{
359-
"cell_type": "code",
360-
"execution_count": null,
361-
"metadata": {},
362-
"outputs": [],
363-
"source": []
364-
},
365376
{
366377
"cell_type": "code",
367378
"execution_count": null,
@@ -433,35 +444,36 @@
433444
"\n",
434445
"Here are a few examples of questions and answers: \n",
435446
"\n",
436-
"`To whom was Viserys Targaryen loyal to?`\n",
447+
"**To whom was Viserys Targaryen loyal to?**\n",
437448
"\n",
438449
"The response is:\n",
439-
"**Based on the provided data, Viserys Targaryen was loyal to House Targaryen. This information is derived from the relationships indicating that Viserys Targaryen was loyal to House Targaryen and the connections between them**\n",
450+
"Based on the provided data, Viserys Targaryen was loyal to House Targaryen. This information is derived from the relationships indicating that Viserys Targaryen was loyal to House Targaryen and the connections between them\n",
440451
"\n",
441-
"`Who killed Viserys Targaryen in Game of thrones?`\n",
452+
"**Who killed Viserys Targaryen in Game of thrones?**\n",
442453
"\n",
443454
"The response is:\n",
444455
"\n",
445456
"**Based on the provided relevance expansion data, Khal Drogo killed Viserys Targaryen in \"Game of Thrones.\" This information is inferred from the relationship where Khal Drogo is linked to Viserys Targaryen with the action of being \"KILLED\" by Khal Drogo. The data does not show any other character directly killing Viserys Targaryen.\n",
446457
"**\n",
447458
"\n",
448-
"`\"What was the weapon used to kill Viserys Targaryen in Game of Thrones?\"`\n",
459+
"**\"What was the weapon used to kill Viserys Targaryen in Game of Thrones?\"**\n",
449460
"\n",
450461
"The response is: \n",
451-
"**\n",
452-
"Based on the provided data, the weapon used to kill Viserys Targaryen in Game of Thrones was not explicitly mentioned. The data only shows that Khal Drogo was involved in the killing of Viserys Targaryen. There is no specific mention of the weapon used in the relevance expansion data. Therefore, I do not have enough information to answer the question about the weapon used to kill Viserys Targaryen.**\n",
462+
"\n",
463+
"Based on the provided data, the weapon used to kill Viserys Targaryen in Game of Thrones was not explicitly mentioned. The data only shows that Khal Drogo was involved in the killing of Viserys Targaryen. There is no specific mention of the weapon used in the relevance expansion data. Therefore, I do not have enough information to answer the question about the weapon used to kill Viserys Targaryen.\n",
453464
"\n",
454465
"This response is wrong, there is a method mentioned, not weapon, but LLM didn't catch the context due to different naming. \n",
455466
"\n",
456-
"`\"Who betrayed Viserys Targaryen in Game of Thrones?\"`\n",
467+
"**\"Who betrayed Viserys Targaryen in Game of Thrones?\"**\n",
457468
"\n",
458469
"Based on the provided data, Khal Drogo betrayed Viserys Targaryen in Game of Thrones by killing him. This conclusion is drawn from the relationship between Khal Drogo and Viserys Targaryen where it is stated that Khal Drogo killed Viserys Targaryen.\n",
459470
"\n",
460471
"This response is based on the killing relationship, but betrayal could have different consequences. \n",
461472
"\n",
462473
"Let's expand this knowledge. \n",
463474
"\n",
464-
"`\"Who betrayed Viserys Targaryen in Game of Thrones?\"`\n",
475+
"**\"Who betrayed Viserys Targaryen in Game of Thrones?\"**\n",
476+
"\n",
465477
"Based on the provided data, Khal Drogo betrayed Viserys Targaryen in Game of Thrones. This conclusion is derived from the relationship between Viserys Targaryen and Khal Drogo, where Khal Drogo is connected to Viserys Targaryen through the 'BETRAYED_BY' relationship, indicating that Khal Drogo betrayed Viserys Targaryen."
466478
]
467479
},
@@ -714,21 +726,41 @@
714726
"cell_type": "markdown",
715727
"metadata": {},
716728
"source": [
717-
"The dataset now has additional knowledge: \n",
718-
"`\n",
719-
"Executed query: MATCH (a {id: 1}), (b {id: 2}) CREATE (a)-[:PARENT_OF]->(b)\n",
720-
"Executed query: MATCH (a {id: 1}), (b {id: 3}) CREATE (a)-[:SIBLING_OF]->(b)\n",
721-
"Executed query: MATCH (a {id: 1}), (b {id: 4}) CREATE (a)-[:LOYAL_TO]->(b)\n",
722-
"Executed query: MATCH (a {id: 1}), (b {id: 5}) CREATE (a)-[:HAPPENED_IN]->(b)\n",
723-
"Executed query: MATCH (a {id: 1}), (b {id: 6}) CREATE (a)-[:HAPPENED_IN]->(b)\n",
724-
"Executed query: MATCH (a {id: 1}), (b {id: 7}) CREATE (a)-[:SIBLING_OF]->(b)\n",
725-
"Executed query: MATCH (a {id: 1}), (b {id: 8}) CREATE (a)-[:MARRIED_TO]->(b)\n",
726-
"Executed query: MATCH (a {id: 1}), (b {id: 9}) CREATE (a)-[:HEALED_BY]->(b)\n",
727-
"Executed query: MATCH (a {id: 8}), (b {id: 9}) CREATE (a)-[:RULES]->(b)\n",
728-
"Executed query: MATCH (a {id: 8}), (b {id: 1}) CREATE (a)-[:BETRAYED_BY]->(b)\n",
729-
"` \n",
730-
"\n",
731-
"Now again asking the same question yields a correct answer: `\"Who betrayed Viserys Targaryen in Game of Thrones?\"`\n",
729+
"The dataset now has additional knowledge: "
730+
]
731+
},
732+
{
733+
"cell_type": "code",
734+
"execution_count": null,
735+
"metadata": {
736+
"vscode": {
737+
"languageId": "sql"
738+
}
739+
},
740+
"outputs": [],
741+
"source": [
742+
"\n",
743+
"\n",
744+
"MATCH (a {id: 1}), (b {id: 2}) CREATE (a)-[:PARENT_OF]->(b)\n",
745+
"MATCH (a {id: 1}), (b {id: 3}) CREATE (a)-[:SIBLING_OF]->(b)\n",
746+
"MATCH (a {id: 1}), (b {id: 4}) CREATE (a)-[:LOYAL_TO]->(b)\n",
747+
"MATCH (a {id: 1}), (b {id: 5}) CREATE (a)-[:HAPPENED_IN]->(b)\n",
748+
"MATCH (a {id: 1}), (b {id: 6}) CREATE (a)-[:HAPPENED_IN]->(b)\n",
749+
"MATCH (a {id: 1}), (b {id: 7}) CREATE (a)-[:SIBLING_OF]->(b)\n",
750+
"MATCH (a {id: 1}), (b {id: 8}) CREATE (a)-[:MARRIED_TO]->(b)\n",
751+
"MATCH (a {id: 1}), (b {id: 9}) CREATE (a)-[:HEALED_BY]->(b)\n",
752+
"MATCH (a {id: 8}), (b {id: 9}) CREATE (a)-[:RULES]->(b)\n",
753+
"MATCH (a {id: 8}), (b {id: 1}) CREATE (a)-[:BETRAYED_BY]->(b)\n",
754+
"\n",
755+
"\n"
756+
]
757+
},
758+
{
759+
"cell_type": "markdown",
760+
"metadata": {},
761+
"source": [
762+
"\n",
763+
"Now again asking the same question yields a correct answer: **\"Who betrayed Viserys Targaryen in Game of Thrones?\"**\n",
732764
"\n",
733765
"Based on the provided data, Khal Drogo betrayed Viserys Targaryen in Game of Thrones. This conclusion is derived from the relationship between Viserys Targaryen and Khal Drogo, where Khal Drogo is connected to Viserys Targaryen through the 'BETRAYED_BY' relationship, indicating that Khal Drogo betrayed Viserys Targaryen.\n",
734766
"\n"

0 commit comments

Comments
 (0)