gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Add other search types to notebook

Browse source 
Added other search types and explanations to notebook

Refactor #COG-387
This commit is contained in:
Igor Ilic 2024-10-09 20:58:16 +02:00
parent 9ec292aac5
commit bd8f5284a9
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
notebooks/cognee_demo.ipynb
View file

@ -735,7 +735,7 @@
"node = (await vector_engine.search(\"entities\", \"sarah.nguyen@example.com\"))[0]\n", "node = (await vector_engine.search(\"entities\", \"sarah.nguyen@example.com\"))[0]\n",
"node_name = node.payload[\"name\"]\n", "node_name = node.payload[\"name\"]\n",
"\n", "\n",
"search_results = await cognee.search(SearchType.SUMMARIES, query = random_node_name)\n", "search_results = await cognee.search(SearchType.SUMMARIES, query = node_name)\n",
"print(\"\\n\\Extracted summaries are:\\n\")\n", "print(\"\\n\\Extracted summaries are:\\n\")\n",
"for result in search_results:\n", "for result in search_results:\n",
" print(f\"{result}\\n\")" " print(f\"{result}\\n\")"
@ -756,7 +756,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"search_results = await cognee.search(SearchType.CHUNKS, query = random_node_name)\n", "search_results = await cognee.search(SearchType.CHUNKS, query = node_name)\n",
"print(\"\\n\\nExtracted chunks are:\\n\")\n", "print(\"\\n\\nExtracted chunks are:\\n\")\n",
"for result in search_results:\n", "for result in search_results:\n",
" print(f\"{result}\\n\")" " print(f\"{result}\\n\")"
@ -777,7 +777,7 @@
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
"source": [ "source": [
"search_results = await cognee.search(SearchType.INSIGHTS, query = random_node_name)\n", "search_results = await cognee.search(SearchType.INSIGHTS, query = node_name)\n",
"print(\"\\n\\nExtracted sentences are:\\n\")\n", "print(\"\\n\\nExtracted sentences are:\\n\")\n",
"for result in search_results:\n", "for result in search_results:\n",
" print(f\"{result}\\n\")" " print(f\"{result}\\n\")"