add sanitize inputs

gradio_mcp_space.py

@@ -30,6 +30,47 @@ except Exception as e:
             return func
         return decorator
 
+
+def _sanitize_value(v):
+    if isinstance(v, str):
+        return v.strip()
+    if isinstance(v, dict):
+        return {k: _sanitize_value(val) for k, val in v.items()}
+    if isinstance(v, (list, tuple)):
+        t = type(v)
+        return t(_sanitize_value(x) for x in v)
+    return v
+
+
+def sanitize_inputs(func):
+    """Decorator that trims whitespace from all string args/kwargs before calling func."""
+    def wrapper(*args, **kwargs):
+        new_args = tuple(_sanitize_value(a) for a in args)
+        new_kwargs = {k: _sanitize_value(v) for k, v in kwargs.items()}
+        return func(*new_args, **new_kwargs)
+    # preserve original attributes
+    try:
+        wrapper.__name__ = func.__name__
+        wrapper.__doc__ = func.__doc__
+    except Exception:
+        pass
+    return wrapper
+
+
+# Wrap the existing `observe` decorator (from langfuse or fallback) so that
+# all observed tools receive sanitized inputs automatically. This avoids
+# having to manually add `@sanitize_inputs` above every `@observe`.
+try:
+    _original_observe = observe
+    def _observe_with_sanitize(*o_args, **o_kwargs):
+        def decorator(f):
+            return _original_observe(*o_args, **o_kwargs)(sanitize_inputs(f))
+        return decorator
+    observe = _observe_with_sanitize
+except Exception:
+    # If anything goes wrong, keep the existing observe as-is.
+    pass
+
 # Add parent directory to path
 sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'RepoKnowledgeGraphLib'))