aboutsummaryrefslogtreecommitdiff
path: root/packages/openai-sdk-python/src/supermemory_openai/tools.py
diff options
context:
space:
mode:
Diffstat (limited to 'packages/openai-sdk-python/src/supermemory_openai/tools.py')
-rw-r--r--packages/openai-sdk-python/src/supermemory_openai/tools.py20
1 files changed, 18 insertions, 2 deletions
diff --git a/packages/openai-sdk-python/src/supermemory_openai/tools.py b/packages/openai-sdk-python/src/supermemory_openai/tools.py
index ec87db10..111253cd 100644
--- a/packages/openai-sdk-python/src/supermemory_openai/tools.py
+++ b/packages/openai-sdk-python/src/supermemory_openai/tools.py
@@ -16,6 +16,12 @@ from supermemory.types import (
)
from supermemory.types.search_execute_response import Result
+from .exceptions import (
+ SupermemoryConfigurationError,
+ SupermemoryMemoryOperationError,
+ SupermemoryNetworkError,
+)
+
class SupermemoryToolsConfig(TypedDict, total=False):
"""Configuration for Supermemory tools.
@@ -194,10 +200,15 @@ class SupermemoryTools:
results=response.results,
count=len(response.results),
)
+ except (OSError, ConnectionError) as network_error:
+ return MemorySearchResult(
+ success=False,
+ error=f"Network error: {network_error}",
+ )
except Exception as error:
return MemorySearchResult(
success=False,
- error=str(error),
+ error=f"Memory search failed: {error}",
)
async def add_memory(self, memory: str) -> MemoryAddResult:
@@ -225,10 +236,15 @@ class SupermemoryTools:
success=True,
memory=response,
)
+ except (OSError, ConnectionError) as network_error:
+ return MemoryAddResult(
+ success=False,
+ error=f"Network error: {network_error}",
+ )
except Exception as error:
return MemoryAddResult(
success=False,
- error=str(error),
+ error=f"Memory add failed: {error}",
)