scriptoriumadmin/supermemory
1
0
Fork 0
mirror of https://github.com/supermemoryai/supermemory.git synced 2026-09-12 23:01:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add input validation for options and container tag

Browse source
This commit is contained in:
Shoubhit Dash 2025-10-27 23:56:43 +05:30
parent 69e94b83ee
commit 9f1d545543
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
packages/openai-sdk-python/src/supermemory_openai/wrapper.py
View file

@ -32,7 +32,15 @@ class WithSupermemoryOptions:
verbose: Enable detailed logging
mode: Memory search mode ("profile", "query", or "full")
add_memory: Memory addition mode ("always" or "never")
Raises:
ValueError: If mode or add_memory values are invalid
"""
if mode not in ("profile", "query", "full"):
raise ValueError(f"Invalid mode: {mode}. Must be 'profile', 'query', or 'full'")
if add_memory not in ("always", "never"):
raise ValueError(f"Invalid add_memory: {add_memory}. Must be 'always' or 'never'")
self.conversation_id = conversation_id
self.verbose = verbose
self.mode = mode
@ -54,7 +62,13 @@ class SupermemoryOpenAIWrapper:
client: OpenAI client instance
container_tag: Container tag for memory operations
options: Configuration options
Raises:
ValueError: If container_tag is empty or None
"""
if not container_tag or not container_tag.strip():
raise ValueError("container_tag cannot be empty or None")
self.client = client
self.container_tag = container_tag
self.options = options or WithSupermemoryOptions()