from tooluniverse import ToolUniverse as BaseToolUniverse

class PatchedToolUniverse(BaseToolUniverse):
    def infer_tool_embeddings(self, *args, **kwargs):
        # Call original method but DO NOT exit
        super().infer_tool_embeddings(*args, **kwargs)
        print("✅ Tool embedding complete — no exit() called.")