diff --git a/memlayer/wrappers/claude.py b/memlayer/wrappers/claude.py index 464e930..563effe 100644 --- a/memlayer/wrappers/claude.py +++ b/memlayer/wrappers/claude.py @@ -805,3 +805,7 @@ def close(self): except Exception as e: print(f"Warning: Error during cleanup: {e}") + +# Legacy alias for backward compatibility +ClaudeWrapper = Claude + diff --git a/memlayer/wrappers/gemini.py b/memlayer/wrappers/gemini.py index da0b70f..2c4c890 100644 --- a/memlayer/wrappers/gemini.py +++ b/memlayer/wrappers/gemini.py @@ -758,3 +758,6 @@ def close(self): print(f"Warning: Error during cleanup: {e}") +# Legacy alias for backward compatibility +GeminiWrapper = Gemini + diff --git a/memlayer/wrappers/ollama.py b/memlayer/wrappers/ollama.py index 05f26a6..3b2b573 100644 --- a/memlayer/wrappers/ollama.py +++ b/memlayer/wrappers/ollama.py @@ -700,3 +700,7 @@ def close(self): self._graph_storage.close() except Exception as e: print(f"Warning: Error during cleanup: {e}") + + +# Legacy alias for backward compatibility +OllamaWrapper = Ollama diff --git a/memlayer/wrappers/openai.py b/memlayer/wrappers/openai.py index 689af7b..4b1b297 100644 --- a/memlayer/wrappers/openai.py +++ b/memlayer/wrappers/openai.py @@ -817,3 +817,7 @@ def close(self): self._graph_storage.close() except Exception as e: print(f"Warning: Error during cleanup: {e}") + + +# Legacy alias for backward compatibility +OpenAIWrapper = OpenAI