mirror of
https://github.com/mlabonne/llm-course.git
synced 2024-11-10 19:10:54 +00:00
378 lines
1.9 MiB
Plaintext
378 lines
1.9 MiB
Plaintext
|
{
|
||
|
|
"cells": [
|
||
|
|
{
|
||
|
|
"cell_type": "markdown",
|
||
|
|
"metadata": {
|
||
|
|
"id": "view-in-github",
|
||
|
|
"colab_type": "text"
|
||
|
|
},
|
||
|
|
"source": [
|
||
|
|
"<a href=\"https://colab.research.google.com/github/mlabonne/how-to-data-science/blob/main/Improve_ChatGPT_with_Knowledge_Graphs.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "markdown",
|
||
|
|
"source": [
|
||
|
|
"# Improve ChatGPT with Knowledge Graphs\n",
|
||
|
|
"\n",
|
||
|
|
"❤️ Created by [@maximelabonne](https://twitter.com/maximelabonne)."
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"id": "vM5VDCJyAWnC"
|
||
|
|
},
|
||
|
|
"id": "vM5VDCJyAWnC"
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"!pip install -q openai langchain"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"id": "uWz0SMuX5FFp"
|
||
|
|
},
|
||
|
|
"id": "uWz0SMuX5FFp",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": []
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"import os\n",
|
||
|
|
"import openai\n",
|
||
|
|
"\n",
|
||
|
|
"os.environ['OPENAI_API_KEY'] = \"your OpenAI key\"\n",
|
||
|
|
"openai.api_key = os.environ['OPENAI_API_KEY']\n",
|
||
|
|
"\n",
|
||
|
|
"question = \"When did apple announced the Vision Pro?\"\n",
|
||
|
|
"completion = openai.ChatCompletion.create(model=\"gpt-3.5-turbo\",\n",
|
||
|
|
" temperature=0,\n",
|
||
|
|
" messages=[{\"role\": \"user\",\n",
|
||
|
|
" \"content\": question}])\n",
|
||
|
|
"print(completion[\"choices\"][0][\"message\"][\"content\"])"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"id": "XpvHDkzNnFLj",
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/"
|
||
|
|
},
|
||
|
|
"outputId": "3d634edc-dbba-4294-8dcc-2ea998e7928e"
|
||
|
|
},
|
||
|
|
"id": "XpvHDkzNnFLj",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "stream",
|
||
|
|
"name": "stdout",
|
||
|
|
"text": [
|
||
|
|
"As an AI language model, I do not have access to current events or real-time information. However, as of my last training data, Apple has not announced any product called \"Vision Pro.\" It is possible that this product does not exist or has not been announced yet.\n"
|
||
|
|
]
|
||
|
|
}
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"execution_count": null,
|
||
|
|
"id": "88a9f2e3-c729-455a-a338-2f83776c1d4c",
|
||
|
|
"metadata": {
|
||
|
|
"id": "88a9f2e3-c729-455a-a338-2f83776c1d4c",
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/"
|
||
|
|
},
|
||
|
|
"outputId": "f6c9f4d8-387b-4ffc-e200-0c35ed449d8a"
|
||
|
|
},
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "execute_result",
|
||
|
|
"data": {
|
||
|
|
"text/plain": [
|
||
|
|
"[('Apple', 'Vision Pro', 'announced'),\n",
|
||
|
|
" ('Vision Pro', '2023', 'was announced in')]"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
"metadata": {},
|
||
|
|
"execution_count": 3
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"source": [
|
||
|
|
"from langchain.llms import OpenAI\n",
|
||
|
|
"from langchain.indexes import GraphIndexCreator\n",
|
||
|
|
"from langchain.chains import GraphQAChain\n",
|
||
|
|
"from langchain.prompts import PromptTemplate\n",
|
||
|
|
"\n",
|
||
|
|
"text = \"Apple announced the Vision Pro in 2023.\"\n",
|
||
|
|
"\n",
|
||
|
|
"index_creator = GraphIndexCreator(llm=OpenAI(temperature=0))\n",
|
||
|
|
"graph = index_creator.from_text(text)\n",
|
||
|
|
"graph.get_triples()"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"execution_count": null,
|
||
|
|
"id": "eecf2d57-3efa-4b0d-941a-95438d42893c",
|
||
|
|
"metadata": {
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/",
|
||
|
|
"height": 1000
|
||
|
|
},
|
||
|
|
"id": "eecf2d57-3efa-4b0d-941a-95438d42893c",
|
||
|
|
"outputId": "1db33465-8f43-4294-84e0-0ad481b005aa"
|
||
|
|
},
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "display_data",
|
||
|
|
"data": {
|
||
|
|
"text/plain": [
|
||
|
|
"<Figure size 2400x1500 with 1 Axes>"
|
||
|
|
],
|
||
|
|
"image/png": "iVBORw0KGgoAAAANSUhEUgAAB38AAAS/CAYAAAAKITI3AAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjcuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/bCgiHAAAACXBIWXMAAC4jAAAuIwF4pT92AAEAAElEQVR4nOzdd3hcZ53+/3tm1LtkWe69O7bj3u24SAZSCCGBACkkIcuy1KXlF9h8F5aSBRI2LATIQnqhpBBSCZbcYlvuvXdJtmxLVm8jTTu/PxwrPhqVmVE5mtH7dV25wnx0ym1piHXp1vMcm2EYhgAAAAAAAAAAAAAAYc1udQAAAAAAAAAAAAAAQOdR/gIAAAAAAAAAAABABKD8BQAAAAAAAAAAAIAIQPkLAAAAAAAAAAAAABGA8hcAAAAAAAAAAAAAIgDlLwAAAAAAAAAAAABEAMpfAAAAAAAAAAAAAIgAlL8AAAAAAAAAAAAAEAEofwEAAAAAAAAAAAAgAlD+AgAAAAAAAAAAAEAEoPwFAAAAAAAAAAAAgAhA+QsAAAAAAAAAAAAAEYDyFwAAAAAAAAAAAAAiAOUvAAAAAAAAAAAAAEQAyl8AAAAAAAAAAAAAiACUvwAAAAAAAAAAAAAQASh/AQAAAAAAAAAAACACUP4CAAAAAAAAAAAAQASg/AUAAAAAAAAAAACACED5CwAAAAAAAAAAAAARgPIXAAAAAAAAAAAAACIA5S8AAAAAAAAAAAAARADKXwAAAAAAAAAAAACIAJS/AAAAAAAAAAAAABABKH8BAAAAAAAAAAAAIAJQ/gIAAAAAAAAAAABABKD8BQAAAAAAAAAAAIAIQPkLAAAAAAAAAAAAABGA8hcAAAAAAAAAAAAAIgDlLwAAAAAAAAAAAABEAMpfAAAAAAAAAAAAAIgAlL8AAAAAAAAAAAAAEAEofwEAAAAAAAAAAAAgAlD+AgAAAAAAAAAAAEAEiLI6AAAAQE8wDEN1TR6V1DSptLZRpR/8+/LrJjndXnl9Pnl8hrwf/OOw2+Sw2xRlt8lhtys+2qGs5FgNSIlVVnKcsj7494CUWCXFRslms1n9xwQAAAAAAADQh1H+AgCAiOL1GTp9qU77i6t1oLhaRy7U6GJ1Y3PB212uFMMDU+M0aVCKpg1J1dQhqRrdP0kOO6UwAAAAAAAAgO5nMwzDsDoEAABAKLw+Q6cu1enAB0XvweJqHb5QowZX95W8wUqIcWjyoBRN+aAMnkYhDAAAAAAAAKCbUP4CAICwUlbXpLVHS5V3pESbT5apvhcVvYFKjHFo0dhMZU8aoBUTs5SZFGt1JAAAAAAAAAARgPIXAAD0aoZh6ERpnfKOlCjvSIn2nK1SJH33YrNJM4alKXvSAOVMGqCxWUk8OxgAAAAAAABASCh/AQBAr+PzGdpeUKHVhy8XvkUVDVZH6jHDMxKUPWmAVk0eoLkjM2Rne2gAAAAAAAAAAaL8BQAAvUa1061Xd53TS9sKdbqs3uo4lhudmag75o3QbbOGKjU+2uo4AAAAAAAAAHo5yl8AAGC5A8XVenFrod7YV6xGt8/qOL1OXLRdN187RHfNH6EpQ1KtjgMAAAAAAACgl6L8BQAAlmh0e/X2/gt6YWuh9p2rsjpO2Jg+LE13zR+hG6YOUly0w+o4AAAAAAAAAHoRyl8AANCjahvd+uPGM3p+S4GqnG6r44St9IRo3b1gpO5fPErJcWwJDQAAAAAAAIDyFwAA9JAmj1cvbi3Sb9efVEW9y+o4ESMjMUZfWTZWd84frtgoVgIDAAAAAAAAfRnlLwAA6FZen6HX9xTrsbzjKq5yWh0nYg1Ji9c3s8frlhlD5LDbrI4DAAAAAAAAwAKUvwAAoFsYhqHcIyV65J/HdKK0zuo4fcb4AUn67qqJyp6UJZuNEhgAAAAAAADoSyh/AQBAl9tVWKmH/3FEuworrY7SZ80aka7vf2ySZo1ItzoKAAAAAAAAgB5C+QsAALqM0+XVI6uP6Zn8M+I7DOvZbNK9C0fpu6smKD6G5wEDAAAAAAAAkY7yFwAAdIkdBRX67qv7VFDeYHUUtDCqX6J+cds0zRmZYXUUAAAAAAAAAN2I8hcAAHQKq33DA6uAAQAAAAAAgMhH+QsAAELGat/wwypgAAAAAAAAIHJR/gIAgKCx2je8sQoYAAAAAAAAiEyUvwAAICjFVU79y/M7dfhCjdVR0EnXDE7RH+6arSFp8VZHAQAAAAAAANAFKH8BAEDAdhRU6Esv7lJ5vcvqKOgimUkxeuKOWZrNNtAAAAAAAABA2KP8BQAAAfnrjiI99MZBub186xBpoh02/fQTU/Xp2cOsjgIAAAAAAACgEyh/AQBAuzxen37y7hE9m19gdRR0s3sXjtR/XD9JUQ671VEAAAAAAAAAhIDyFwAAtKmqwaWv/nmPNp0sszoKesjisZl6/LMzlJYQY3UUAAAAAAAAAEGi/AUAAK06WVqrLzy3U4UVDVZHQQ8bkZGgpz4/W2Ozkq2OAgAAAAAAACAIlL8AAMDPgeJq3fXUNlU53VZHgUXS4qP14hfmacqQVKujAAAAAAAAAAgQ5S8AADDZVVipe57drtpGj9VRYLHkuCg9d+9czRyebnUUAAAAAAAAAAGg/AUAAM22nSnXfc/uUL3La3UU9BKJMQ49fc8czRvVz+ooAAAAAAAAADpA+QsAACRdLn4//8x2Nbp9VkdBLxMXbddz986lAAYAAAAAAAB6ObvVAQAAgPV2FVbqvmd3UPyiVY1un+57dod2F1VaHQUAAAAAAABAOyh/AQDo4w4WV+ueZ7ez1TPaVe/y6vPPbNfB4mqrowAAAAAAAABoA+UvAAB92MnSWt351DbVNnqsjoIwUNvo0Z1PbdPJ0lqrowAAAAAAAABoBeUvAAB9VFWDS194bqeqnG6royCMVDnduv/5napqcFkdBQAAAAAAAEALlL8AAPRBHq9PX/nTbhVWNFgdBWGooLxBX/3zHnm8PCMaAAAAAAAA6E0ofwEA6IN+8u4RbT5VbnUMhLFNJ8v003ePWB0DAAAAAAAAwFUofwEA6GP+sqNIz+YXWB0DEeCZ/AL9dUeR1TEAAAAAAAAAfIDyFwCAPmRHQYX+3xsHrY6BCPLQGwe1s6DC6hgAAAAAAAAARPkLAECfUVzl1L+9tEtur2F1FEQQt9fQl17apeIqp9VRAAAAAAAAgD6P8hcAgD7A6fLqiy/sVFmdy+ooiEBldS598YWdcrq8VkcBAAAAAAAA+jTKXwAA+oBHVh/TofM1VsdABDt0vkaPrj5mdQwAAAAAAACgT6P8BQAgwm0/U6Fn8s9YHQN9wNP5Z7SD5/8CAAAAAAAAlqH8BQAggjldXj3w2j4ZPOYXPcAwpO++uo/tnwEAAAAAAACLUP4CABDBHll9TAXlDVbHQB9SUN7A9s8AAAAAAACARSh/AQCIUGz3DKuw/TMAAAAAAABgDcpfAAAiENs9w0ps/wwAAAAAAABYg/IXAIAIxHbPsBrbPwMAAAAAAAA9j/IXAIAIs6uwku2e0Ss8nX9GuworrY4BAAAAAAAA9BmUvwAARBDDMPTTdw+z3TN6BcOQHv7HERm8IQEAAAAAAIAeQfkLAEAEyTtSqt1FVVbHAJrtKqzUmqOlVscAAAAAAAAA+gTKXwAAIoTXZ+iR1UetjgH4+cU/j8rrY/UvAAAAAAAA0N0ofwEAiBCv7ynW8ZI6q2MAfo6X1Onve4utjgEAAAAAAABEPMpfAAAiQKPbq8fyjlsdA2jT/+QeV5PHa3UMAAAAAAAAIKJR/gIAEAFe2lak4iqn1TGANhVXOfXStiKrYwAAAAAAAAARjfIXAIAwV9vo1m/Xn7Q6BtChx9edVG2j2+oYAAAAAAAAQMSi/AUAIMz9ceMZVdS7rI4BdKii3qUnN52xOgYAAAAAAAAQsSh/AQAIY41ur57fUmB1DCBgz28pUKObZ/8CAAAAAAAA3YHyFwCAMPbOgQuqcrKNLsJHZYNb7x64YHUMAAAAAAAAICJR/gIAEMZe2FpodQQgaLxvAQAAAAAAgO5B+QsAQJg6UFytvWerr
|
||
|
|
},
|
||
|
|
"metadata": {}
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"source": [
|
||
|
|
"import networkx as nx\n",
|
||
|
|
"import matplotlib.pyplot as plt\n",
|
||
|
|
"\n",
|
||
|
|
"# Create graph\n",
|
||
|
|
"G = nx.DiGraph()\n",
|
||
|
|
"G.add_edges_from((source, target, {'relation': relation}) for source, relation, target in graph.get_triples())\n",
|
||
|
|
"\n",
|
||
|
|
"# Plot the graph\n",
|
||
|
|
"plt.figure(figsize=(8,5), dpi=300)\n",
|
||
|
|
"pos = nx.spring_layout(G, k=3, seed=0)\n",
|
||
|
|
"\n",
|
||
|
|
"nx.draw_networkx_nodes(G, pos, node_size=2000)\n",
|
||
|
|
"nx.draw_networkx_edges(G, pos, edge_color='gray')\n",
|
||
|
|
"nx.draw_networkx_labels(G, pos, font_size=12)\n",
|
||
|
|
"edge_labels = nx.get_edge_attributes(G, 'relation')\n",
|
||
|
|
"nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_size=10)\n",
|
||
|
|
"\n",
|
||
|
|
"# Display the plot\n",
|
||
|
|
"plt.axis('off')\n",
|
||
|
|
"plt.show()"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"chain = GraphQAChain.from_llm(OpenAI(temperature=0), graph=graph, verbose=True)\n",
|
||
|
|
"chain.run(question)"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/",
|
||
|
|
"height": 199
|
||
|
|
},
|
||
|
|
"id": "KZIs4N5S8e8S",
|
||
|
|
"outputId": "af6ba50c-e5f6-40be-adaf-07911dab948f"
|
||
|
|
},
|
||
|
|
"id": "KZIs4N5S8e8S",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "stream",
|
||
|
|
"name": "stdout",
|
||
|
|
"text": [
|
||
|
|
"\n",
|
||
|
|
"\n",
|
||
|
|
"\u001b[1m> Entering new GraphQAChain chain...\u001b[0m\n",
|
||
|
|
"Entities Extracted:\n",
|
||
|
|
"\u001b[32;1m\u001b[1;3m Apple, Vision Pro\u001b[0m\n",
|
||
|
|
"Full Context:\n",
|
||
|
|
"\u001b[32;1m\u001b[1;3mApple announced Vision ProVision Pro was announced in 2023\u001b[0m\n",
|
||
|
|
"\n",
|
||
|
|
"\u001b[1m> Finished chain.\u001b[0m\n"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"output_type": "execute_result",
|
||
|
|
"data": {
|
||
|
|
"text/plain": [
|
||
|
|
"' Apple announced Vision Pro in 2023.'"
|
||
|
|
],
|
||
|
|
"application/vnd.google.colaboratory.intrinsic+json": {
|
||
|
|
"type": "string"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"metadata": {},
|
||
|
|
"execution_count": 5
|
||
|
|
}
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"from langchain.graphs.networkx_graph import KnowledgeTriple\n",
|
||
|
|
"\n",
|
||
|
|
"# Knowledge graph\n",
|
||
|
|
"kg = [\n",
|
||
|
|
" ('Apple', 'is', 'Company'),\n",
|
||
|
|
" ('Apple', 'created', 'iMac'),\n",
|
||
|
|
" ('Apple', 'created', 'iPhone'),\n",
|
||
|
|
" ('Apple', 'created', 'Apple Watch'),\n",
|
||
|
|
" ('Apple', 'created', 'Vision Pro'),\n",
|
||
|
|
"\n",
|
||
|
|
" ('Apple', 'developed', 'macOS'),\n",
|
||
|
|
" ('Apple', 'developed', 'iOS'),\n",
|
||
|
|
" ('Apple', 'developed', 'watchOS'),\n",
|
||
|
|
"\n",
|
||
|
|
" ('Apple', 'is located in', 'USA'),\n",
|
||
|
|
" ('Steve Jobs', 'co-founded', 'Apple'),\n",
|
||
|
|
" ('Steve Wozniak', 'co-founded', 'Apple'),\n",
|
||
|
|
" ('Tim Cook', 'is the CEO of', 'Apple'),\n",
|
||
|
|
"\n",
|
||
|
|
" ('iOS', 'runs on', 'iPhone'),\n",
|
||
|
|
" ('macOS', 'runs on', 'iMac'),\n",
|
||
|
|
" ('watchOS', 'runs on', 'Apple Watch'),\n",
|
||
|
|
"\n",
|
||
|
|
" ('Apple', 'was founded in', '1976'),\n",
|
||
|
|
" ('Apple', 'owns', 'App Store'),\n",
|
||
|
|
" ('App Store', 'sells', 'iOS apps'),\n",
|
||
|
|
"\n",
|
||
|
|
" ('iPhone', 'announced in', '2007'),\n",
|
||
|
|
" ('iMac', 'announced in', '1998'),\n",
|
||
|
|
" ('Apple Watch', 'announced in', '2014'),\n",
|
||
|
|
" ('Vision Pro', 'announced in', '2023'),\n",
|
||
|
|
"]\n",
|
||
|
|
"\n",
|
||
|
|
"graph = index_creator.from_text('')\n",
|
||
|
|
"for (node1, relation, node2) in kg:\n",
|
||
|
|
" graph.add_triple(KnowledgeTriple(node1, relation, node2))"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"id": "aa1R_DL-vTgv"
|
||
|
|
},
|
||
|
|
"id": "aa1R_DL-vTgv",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": []
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"# Create directed graph\n",
|
||
|
|
"G = nx.DiGraph()\n",
|
||
|
|
"for node1, relation, node2 in kg:\n",
|
||
|
|
" G.add_edge(node1, node2, label=relation)\n",
|
||
|
|
"\n",
|
||
|
|
"# Plot the graph\n",
|
||
|
|
"plt.figure(figsize=(25, 25), dpi=300)\n",
|
||
|
|
"pos = nx.spring_layout(G, k=2, iterations=50, seed=0)\n",
|
||
|
|
"\n",
|
||
|
|
"nx.draw_networkx_nodes(G, pos, node_size=5000)\n",
|
||
|
|
"nx.draw_networkx_edges(G, pos, edge_color='gray', edgelist=G.edges(), width=2)\n",
|
||
|
|
"nx.draw_networkx_labels(G, pos, font_size=12)\n",
|
||
|
|
"edge_labels = nx.get_edge_attributes(G, 'label')\n",
|
||
|
|
"nx.draw_networkx_edge_labels(G, pos, edge_labels=edge_labels, font_size=12)\n",
|
||
|
|
"\n",
|
||
|
|
"# Display the plot\n",
|
||
|
|
"plt.axis('off')\n",
|
||
|
|
"plt.show()"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/",
|
||
|
|
"height": 1000
|
||
|
|
},
|
||
|
|
"id": "GhVuIxbUwJ1a",
|
||
|
|
"outputId": "a3c0e3a8-b4a3-4baa-93ea-2daf03e352a2"
|
||
|
|
},
|
||
|
|
"id": "GhVuIxbUwJ1a",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "display_data",
|
||
|
|
"data": {
|
||
|
|
"text/plain": [
|
||
|
|
"<Figure size 7500x7500 with 1 Axes>"
|
||
|
|
],
|
||
|
|
"image/png": "iVBORw0KGgoAAAANSUhEUgAAFvAAABbLCAYAAAASpbmAAAAAOXRFWHRTb2Z0d2FyZQBNYXRwbG90bGliIHZlcnNpb24zLjcuMSwgaHR0cHM6Ly9tYXRwbG90bGliLm9yZy/bCgiHAAAACXBIWXMAAC4jAAAuIwF4pT92AAEAAElEQVR4nOzcMQHDABDEsGv4c/6uZpAMEgIj8O/ubgAAAAAAAAAAAAAAAAAAAAAAAAAAAADAtu15OwAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAvsTAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAAAAAAAAAAMLAGwAAAAAAAAAAAAAAAAAAAAAAAAAAAADCwBsAAAAAAAAAAAAAAAAAAAAAAAAAAAAAwsAbAAAAAAAAAAAAAAAAAAAAA
|
||
|
|
},
|
||
|
|
"metadata": {}
|
||
|
|
}
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"cell_type": "code",
|
||
|
|
"source": [
|
||
|
|
"chain = GraphQAChain.from_llm(OpenAI(temperature=0), graph=graph, verbose=True)\n",
|
||
|
|
"chain.run(question)"
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"colab": {
|
||
|
|
"base_uri": "https://localhost:8080/",
|
||
|
|
"height": 380
|
||
|
|
},
|
||
|
|
"id": "7oGBfjuSuRNm",
|
||
|
|
"outputId": "f4b03c71-c05e-492a-c9ba-21accfb994c8"
|
||
|
|
},
|
||
|
|
"id": "7oGBfjuSuRNm",
|
||
|
|
"execution_count": null,
|
||
|
|
"outputs": [
|
||
|
|
{
|
||
|
|
"output_type": "stream",
|
||
|
|
"name": "stdout",
|
||
|
|
"text": [
|
||
|
|
"\n",
|
||
|
|
"\n",
|
||
|
|
"\u001b[1m> Entering new GraphQAChain chain...\u001b[0m\n",
|
||
|
|
"Entities Extracted:\n",
|
||
|
|
"\u001b[32;1m\u001b[1;3m Apple, Vision Pro\u001b[0m\n",
|
||
|
|
"Full Context:\n",
|
||
|
|
"\u001b[32;1m\u001b[1;3mApple is Company\n",
|
||
|
|
"Apple created iMac\n",
|
||
|
|
"Apple created iPhone\n",
|
||
|
|
"Apple created Apple Watch\n",
|
||
|
|
"Apple created Vision Pro\n",
|
||
|
|
"Apple developed macOS\n",
|
||
|
|
"Apple developed iOS\n",
|
||
|
|
"Apple developed watchOS\n",
|
||
|
|
"Apple is located in USA\n",
|
||
|
|
"Apple was founded in 1976\n",
|
||
|
|
"Apple owns App StoreVision Pro announced in 2023\u001b[0m\n",
|
||
|
|
"\n",
|
||
|
|
"\u001b[1m> Finished chain.\u001b[0m\n"
|
||
|
|
]
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"output_type": "execute_result",
|
||
|
|
"data": {
|
||
|
|
"text/plain": [
|
||
|
|
"' Apple announced the Vision Pro in 2023.'"
|
||
|
|
],
|
||
|
|
"application/vnd.google.colaboratory.intrinsic+json": {
|
||
|
|
"type": "string"
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"metadata": {},
|
||
|
|
"execution_count": 8
|
||
|
|
}
|
||
|
|
]
|
||
|
|
}
|
||
|
|
],
|
||
|
|
"metadata": {
|
||
|
|
"kernelspec": {
|
||
|
|
"display_name": "Python 3 (ipykernel)",
|
||
|
|
"language": "python",
|
||
|
|
"name": "python3"
|
||
|
|
},
|
||
|
|
"language_info": {
|
||
|
|
"codemirror_mode": {
|
||
|
|
"name": "ipython",
|
||
|
|
"version": 3
|
||
|
|
},
|
||
|
|
"file_extension": ".py",
|
||
|
|
"mimetype": "text/x-python",
|
||
|
|
"name": "python",
|
||
|
|
"nbconvert_exporter": "python",
|
||
|
|
"pygments_lexer": "ipython3",
|
||
|
|
"version": "3.9.16"
|
||
|
|
},
|
||
|
|
"colab": {
|
||
|
|
"provenance": [],
|
||
|
|
"include_colab_link": true
|
||
|
|
}
|
||
|
|
},
|
||
|
|
"nbformat": 4,
|
||
|
|
"nbformat_minor": 5
|
||
|
|
}
|