From 896d5a10764c82ce3fb590bb173eefa2a9b3adbe Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Fri, 21 Nov 2025 05:24:36 +0000
Subject: [PATCH 1/2] Initial plan


From 6483c73d735833e7387e94fd7256586970955c0f Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Fri, 21 Nov 2025 05:34:53 +0000
Subject: [PATCH 2/2] Fix: Add legacy Wrapper class aliases for backward
 compatibility

Co-authored-by: divagr18 <114850579+divagr18@users.noreply.github.com>
---
 memlayer/wrappers/claude.py | 4 ++++
 memlayer/wrappers/gemini.py | 3 +++
 memlayer/wrappers/ollama.py | 4 ++++
 memlayer/wrappers/openai.py | 4 ++++
 4 files changed, 15 insertions(+)

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