add membase plugin

Author: mossv2

plugins/membase/.python-version

@@ -0,0 +1 @@
+3.10

plugins/membase/README.md

@@ -0,0 +1,55 @@
+# Membase Plugin for GAME SDK
+
+A plugin for interacting with membase protocol through the GAME SDK.
+
+## Description
+
+Membase is a core component of the Unibase ecosystem. It stores historical information, interaction records, and persistent data of Agents, ensuring their continuity and traceability.
+
+The membase plugin enables seamless integration with the membase protocol for decentralized storage. It provides functionality to upload memory to and reload it from the Unibase DA network.
+
+- support conversation switch
+- support conversation pesistence, upload if auto_upload_to_hub is set, conversation content can be visit at: https://testnet.hub.membase.io/
+- support conversation preload from membase hub: https://testnet.hub.membase.io/
+
+## Installation
+
+```bash
+pip install -e plugins/membase
+```
+
+## Configuration
+
+The plugin requires the following environment variables to be set:
+
+```shell
+MEMBASE_HUB=<Membase hub endpoint, default is 'https://testnet.hub.membase.io' >
+MEMBASE_ACCOUNT=<Membase account address>
+MEMBASE_ID=<your agent name>
+```
+
+## Usage
+
+```python
+import time
+import uuid
+from membase_plugin_gamesdk.membase_plugin_gamesdk import MembasePlugin
+
+# set your own account and agent name
+# or set environment variables
+default_account = "game_sdk_test"
+default_agent_name = "game_sdk_test_agent"
+membase_plugin = MembasePlugin(
+  account=default_account,
+  agent_name=default_agent_name,
+  auto_upload_to_hub=True,
+  preload_from_hub=True,
+)
+
+membase_plugin.add_memory("Hello, world!")
+new_conversation_id = str(uuid.uuid4())
+membase_plugin.switch_conversation_id(new_conversation_id)
+membase_plugin.add_memory("Hello, world! 2")
+```
+
+more in `test_membase.py` file

plugins/membase/membase_plugin_gamesdk/membase_plugin_gamesdk.py

@@ -0,0 +1,45 @@
+import os
+from typing import Literal, Optional
+from membase.memory.multi_memory import MultiMemory
+from membase.memory.message import Message
+
+class MembasePlugin:
+    def __init__(self, account: Optional[str] = None, agent_name: Optional[str] = None, auto_upload_to_hub: bool = False, preload_from_hub: bool = False):
+        if not account:
+            self.account = os.getenv("MEMBASE_ACCOUNT")
+        else:
+            self.account = account
+        if not self.account:
+            raise ValueError("MEMBASE_ACCOUNT is not set and provided account is None")
+        
+        if not agent_name:
+            self.id = os.getenv("MEMBASE_ID")
+        else:
+            self.id = agent_name
+        if not self.id:
+            self.id = self.account
+
+        self._multi_memory = MultiMemory(
+            membase_account=self.account,
+            auto_upload_to_hub=auto_upload_to_hub,
+            preload_from_hub=preload_from_hub,
+        )
+
+    # memory_type: user,system,assistant | default: user
+    def add_memory(self, memory: str, memory_type: Literal["user", "system", "assistant"] = "user", conversation_id: Optional[str] = None):
+        msg = Message(
+            name=self.id,
+            content=memory,
+            role=memory_type,
+        )
+        self._multi_memory.add(msg, conversation_id)
+
+    def get_memory(self, conversation_id: Optional[str] = None, recent_n: Optional[int] = None):
+        return self._multi_memory.get(conversation_id, recent_n)
+    
+    def switch_conversation_id(self, conversation_id: Optional[str] = None):
+        self._multi_memory.update_conversation_id(conversation_id)
+    
+    def reload_memory(self, conversation_id: str):
+        self._multi_memory.load_from_hub(conversation_id)
+    

plugins/membase/pyproject.toml

@@ -0,0 +1,12 @@
+[project]
+name = "membase_plugin_gamesdk"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "membase>=0.1.5",
+]
+
+[tool.uv.sources]
+membase = { git = "https://github.com/unibaseio/membase.git" }

plugins/membase/test_membase.py

@@ -0,0 +1,41 @@
+import time
+import uuid
+from membase_plugin_gamesdk.membase_plugin_gamesdk import MembasePlugin
+
+# set your own account and agent name
+# or set environment variables
+default_account = "game_sdk_test"
+default_agent_name = "game_sdk_test_agent"
+membase_plugin = MembasePlugin(
+  account=default_account,
+  agent_name=default_agent_name,
+  auto_upload_to_hub=True,
+  preload_from_hub=True,
+)
+
+membase_plugin.add_memory("Hello, world!")
+new_conversation_id = str(uuid.uuid4())
+membase_plugin.switch_conversation_id(new_conversation_id)
+membase_plugin.add_memory("Hello, world! 2")
+
+time.sleep(3)
+
+new_agent_name = "game_sdk_test_agent_new"
+new_membase_plugin = MembasePlugin(
+  account=default_account,
+  agent_name=new_agent_name,
+  auto_upload_to_hub=True,
+  preload_from_hub=True,
+)
+
+res = new_membase_plugin.get_memory(new_conversation_id, 1)
+if res is None:
+    raise Exception("Failed to get memory none")
+if len(res) != 1:
+    raise Exception("Failed to get memory")
+if res[0].content != "Hello, world! 2":
+    raise Exception("Content is not correct")
+
+new_membase_plugin.add_memory("Hello, world! 3")
+
+print("Test passed")

pyproject.toml

@@ -31,4 +31,7 @@ dependencies = [
 
 [project.urls]
 "Homepage" = "https://github.com/game-by-virtuals/game-python"
-"Bug Tracker" = "https://github.com/game-by-virtuals/game-python"
+"Bug Tracker" = "https://github.com/game-by-virtuals/game-python"
+
+[tool.uv.workspace]
+members = ["plugins/membase"]