fix: Remove dict and use string for search query

Remove dictionary and use string for query in notebook and simple example Fix COG-337
topoteretes · Nov 14, 2024 · cc53693 · cc53693
1 parent f13146d
commit cc53693
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/examples/python/simple_example.py b/examples/python/simple_example.py
@@ -27,8 +27,7 @@ async def main():
 
     # Query cognee for insights on the added text
     search_results = await cognee.search(
-        SearchType.INSIGHTS,
-        {'query': 'Tell me about NLP'}
+        SearchType.INSIGHTS, query='Tell me about NLP'
     )
 
     # Display search results

diff --git a/notebooks/cognee_llama_index.ipynb b/notebooks/cognee_llama_index.ipynb
@@ -149,8 +149,7 @@
     "\n",
     "# Query cognee for summaries\n",
     "search_results = await cognee.search(\n",
-    "    SearchType.SUMMARIES,\n",
-    "    {'query': \"What are the main news discussed in the document?\"}\n",
+    "    SearchType.SUMMARIES, query=\"What are the main news discussed in the document?\"\n",
     ")\n",
     "# Display search results\n",
     "print(\"\\n Summary of main news discussed:\\n\")\n",