From 0f7db86342a893ae7ff0c58fec3b30f337f7c064 Mon Sep 17 00:00:00 2001
From: Babar Ali <148423037+Babarali2k21@users.noreply.github.com>
Date: Thu, 8 Jan 2026 10:17:03 +0100
Subject: [PATCH] fix: prevent dataset_name leaking into OpenAI requests

Signed-off-by: Babar Ali <148423037+Babarali2k21@users.noreply.github.com>
---
 cognee-mcp/src/cognee_client.py |  4 +---
 run_cognee.py                   | 24 ++++++++++++++++++++++++
 2 files changed, 25 insertions(+), 3 deletions(-)
 create mode 100644 run_cognee.py

diff --git a/cognee-mcp/src/cognee_client.py b/cognee-mcp/src/cognee_client.py
index 9d98cb0b5..3ffbca8d8 100644
--- a/cognee-mcp/src/cognee_client.py
+++ b/cognee-mcp/src/cognee_client.py
@@ -192,9 +192,7 @@ class CogneeClient:
 
             with redirect_stdout(sys.stderr):
                 results = await self.cognee.search(
-                    query_type=SearchType[query_type.upper()],
-                    query_text=query_text,
-                    top_k=top_k
+                    query_type=SearchType[query_type.upper()], query_text=query_text, top_k=top_k
                 )
                 return results
 
diff --git a/run_cognee.py b/run_cognee.py
new file mode 100644
index 000000000..206c5fc4d
--- /dev/null
+++ b/run_cognee.py
@@ -0,0 +1,24 @@
+import asyncio
+import cognee
+
+DATASET = "sheet6_demo"
+
+
+async def main():
+    # optional clean slate
+    await cognee.prune.prune_data()
+    await cognee.prune.prune_system(metadata=True)
+
+    await cognee.add("Cognee turns documents into AI memory.", dataset_name=DATASET)
+
+    # IMPORTANT: use datasets=[...] (not dataset_name=...)
+    await cognee.cognify(datasets=[DATASET])
+
+    results = await cognee.search(query_text="What does Cognee do?", datasets=[DATASET])
+
+    for r in results:
+        print(r.get("search_result"))
+
+
+if __name__ == "__main__":
+    asyncio.run(main())