cleanup

Author: retroryan

Lab_6_Context_Providers/03_fulltext_context_provider.ipynb

@@ -55,16 +55,7 @@
    "id": "4",
    "metadata": {},
    "outputs": [],
-   "source": [
-    "# Get configuration from environment\n",
-    "config = get_agent_config()\n",
-    "\n",
-    "# Load Neo4j settings\n",
-    "neo4j_settings = Neo4jSettings()\n",
-    "\n",
-    "print(f\"Neo4j URI: {neo4j_settings.uri}\")\n",
-    "print(f\"Fulltext Index: {neo4j_settings.fulltext_index_name}\")"
-   ]
+   "source": "# Get configuration from environment\nconfig = get_agent_config()\n\n# Load Neo4j settings\nneo4j_settings = Neo4jSettings()\n\n# Fulltext index name (must match the index created during data loading)\nFULLTEXT_INDEX = \"search_chunks\"\n\nprint(f\"Neo4j URI: {neo4j_settings.uri}\")\nprint(f\"Fulltext Index: {FULLTEXT_INDEX}\")"
   },
   {
    "cell_type": "markdown",
@@ -87,22 +78,7 @@
    "id": "6",
    "metadata": {},
    "outputs": [],
-   "source": [
-    "# Create context provider with fulltext search\n",
-    "provider = Neo4jContextProvider(\n",
-    "    uri=neo4j_settings.uri,\n",
-    "    username=neo4j_settings.username,\n",
-    "    password=neo4j_settings.get_password(),\n",
-    "    index_name=neo4j_settings.fulltext_index_name,\n",
-    "    index_type=\"fulltext\",\n",
-    "    top_k=3,\n",
-    "    context_prompt=(\n",
-    "        \"## Knowledge Graph Context\\n\"\n",
-    "        \"Use the following information from the knowledge graph \"\n",
-    "        \"to answer questions about companies, products, and financials:\"\n",
-    "    ),\n",
-    ")"
-   ]
+   "source": "# Create context provider with fulltext search\nprovider = Neo4jContextProvider(\n    uri=neo4j_settings.uri,\n    username=neo4j_settings.username,\n    password=neo4j_settings.get_password(),\n    index_name=FULLTEXT_INDEX,\n    index_type=\"fulltext\",\n    top_k=3,\n    context_prompt=(\n        \"## Knowledge Graph Context\\n\"\n        \"Use the following information from the knowledge graph \"\n        \"to answer questions about companies, products, and financials:\"\n    ),\n)"
   },
   {
    "cell_type": "markdown",
@@ -247,4 +223,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}