feat: Add get_or_create_user tool

Author: ACAne0320

tools/user/get_or_create_user.py

@@ -0,0 +1,31 @@
+from collections.abc import Generator
+from typing import Any
+import json
+
+from dify_plugin import Tool
+from dify_plugin.entities.tool import ToolInvokeMessage
+from memobase import MemoBaseClient
+
+class GetOrCreateUserTool(Tool):
+    def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:
+        user_id = tool_parameters.get("user_id", "")
+
+        try:
+            credentials = self.runtime.credentials
+            memobase_url = credentials.get("memobase_url", "")
+            memobase_api_key = credentials.get("memobase_api_key", "")
+            
+            client = MemoBaseClient(
+                project_url=memobase_url,
+                api_key=memobase_api_key,
+            )
+            
+            user_id = client.get_or_create_user(user_id)
+            
+            yield self.create_json_message({"status": "success", "user_id": user_id})
+            
+        except Exception as e:
+            yield self.create_json_message({
+                "status": "error",
+                "message": str(e)
+            })

tools/user/get_or_create_user.yaml

@@ -0,0 +1,30 @@
+identity:
+  name: get_or_create_user
+  author: acane0320
+  label:
+    en_US: Get or Create a User
+    zh_Hans: 获取或创建用户
+    pt_BR: Obter ou Criar Usuário
+description:
+  human:
+    en_US: Get or create a user in Memobase
+    zh_Hans: 在 Memobase 中获取或创建用户
+    pt_BR: Obter ou Criar Usuário no Memobase
+  llm: Use this tool to get or create a user in Memobase. You need to provide the user ID.
+parameters:
+  - name: user_id
+    type: string
+    required: true
+    label:
+      en_US: User ID
+      zh_Hans: 用户 ID
+      pt_BR: ID do Usuário
+    human_description:
+      en_US: Unique identifier for the user
+      zh_Hans: 用户的唯一标识符
+      pt_BR: Identificador exclusivo para o usuário
+    llm_description: Unique identifier for the user.
+    form: llm
+extra:
+  python:
+    source: tools/user/get_or_create_user.py