From 7edcb16c163ecddfe899921648cc530709c1dc20 Mon Sep 17 00:00:00 2001
From: Shoubhit Dash <shoubhit2005@gmail.com>
Date: Mon, 27 Oct 2025 23:48:15 +0530
Subject: [PATCH] export withsupermemory wrapper from package

---
 .../openai-sdk-python/src/supermemory_openai/__init__.py   | 7 +++++++
 packages/openai-sdk-python/src/supermemory_openai/tools.py | 1 +
 2 files changed, 8 insertions(+)

diff --git a/packages/openai-sdk-python/src/supermemory_openai/__init__.py b/packages/openai-sdk-python/src/supermemory_openai/__init__.py
index f05816e0..8207c66d 100644
--- a/packages/openai-sdk-python/src/supermemory_openai/__init__.py
+++ b/packages/openai-sdk-python/src/supermemory_openai/__init__.py
@@ -15,6 +15,10 @@ from .tools import (
     create_search_memories_tool,
     create_add_memory_tool,
 )
+from .wrapper import (
+    with_supermemory,
+    WithSupermemoryOptions,
+)
 
 __all__ = [
     # Tools
@@ -31,4 +35,7 @@ __all__ = [
     "execute_memory_tool_calls",
     "create_search_memories_tool",
     "create_add_memory_tool",
+    # Wrapper
+    "with_supermemory",
+    "WithSupermemoryOptions",
 ]
diff --git a/packages/openai-sdk-python/src/supermemory_openai/tools.py b/packages/openai-sdk-python/src/supermemory_openai/tools.py
index ec87db10..29074858 100644
--- a/packages/openai-sdk-python/src/supermemory_openai/tools.py
+++ b/packages/openai-sdk-python/src/supermemory_openai/tools.py
@@ -7,6 +7,7 @@ from openai.types.chat import (
     ChatCompletionMessageToolCall,
     ChatCompletionToolMessageParam,
     ChatCompletionFunctionToolParam,
+    ChatCompletionToolParam,
 )
 import supermemory
 from supermemory.types import (