django-agent-studio 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

django_agent_studio/agents/builder.py

@@ -39,6 +39,7 @@ You have access to tools that allow you to:
 6. **Discover and add dynamic tools** from the Django project codebase
 7. **Create multi-agent systems** by adding sub-agent tools
 8. **Switch between agents and systems** in the UI
+9. **Configure agent memory** (enable/disable the remember tool)
 
 ## IMPORTANT: Tool Usage
 
@@ -95,6 +96,26 @@ You can control the builder UI to switch between different agents and systems:
 When the user asks to work on a different agent or system, use these tools to switch context.
 The UI will automatically update to show the selected agent/system.
 
+## Agent Memory
+
+Agents have a built-in memory system that allows them to remember facts about users across conversations:
+- Memory is **enabled by default** for all agents
+- When enabled, the agent has a `remember` tool it can use to store key-value facts
+- Memories are scoped per-user and per-conversation
+- Memory only works for **authenticated users** (not anonymous visitors)
+- Use `get_memory_status` to check if memory is enabled
+- Use `set_memory_enabled` to enable or disable memory
+
+**When to disable memory:**
+- Public-facing agents where you don't want user data stored
+- Simple Q&A agents that don't need personalization
+- Agents handling sensitive information that shouldn't be persisted
+
+**When to keep memory enabled (default):**
+- Personal assistants that should remember user preferences
+- Support agents that benefit from knowing user history
+- Any agent where personalization improves the experience
+
 When helping users:
 - Ask clarifying questions to understand what they want their agent to do
 - Suggest appropriate system prompts based on the agent's purpose
@@ -231,6 +252,34 @@ BUILDER_TOOLS = [
             },
         },
     },
+    {
+        "type": "function",
+        "function": {
+            "name": "set_memory_enabled",
+            "description": "Enable or disable conversation memory for the agent. When enabled (default), the agent can remember facts about users across messages using the 'remember' tool. Memory only works for authenticated users.",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "enabled": {
+                        "type": "boolean",
+                        "description": "Whether to enable memory. True = agent can remember facts, False = no memory.",
+                    },
+                },
+                "required": ["enabled"],
+            },
+        },
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_memory_status",
+            "description": "Check if memory is enabled for the current agent.",
+            "parameters": {
+                "type": "object",
+                "properties": {},
+            },
+        },
+    },
     {
         "type": "function",
         "function": {
@@ -908,6 +957,33 @@ Knowledge: {len(config.get('knowledge', []))} sources
                     return {"success": True, "message": "Model settings updated"}
                 return {"error": "No active version found"}
 
+            elif tool_name == "set_memory_enabled":
+                if version:
+                    enabled = args.get("enabled", True)
+                    if version.extra_config is None:
+                        version.extra_config = {}
+                    version.extra_config["memory_enabled"] = enabled
+                    await sync_to_async(version.save)()
+                    status = "enabled" if enabled else "disabled"
+                    await create_revision(agent, comment=f"Memory {status}")
+                    return {
+                        "success": True,
+                        "message": f"Memory {status} for this agent",
+                        "memory_enabled": enabled,
+                    }
+                return {"error": "No active version found"}
+
+            elif tool_name == "get_memory_status":
+                if version:
+                    extra = version.extra_config or {}
+                    enabled = extra.get("memory_enabled", True)  # Default is True
+                    return {
+                        "memory_enabled": enabled,
+                        "message": f"Memory is {'enabled' if enabled else 'disabled'} for this agent",
+                        "note": "When enabled, the agent has a 'remember' tool to store facts about users. Memory only works for authenticated users.",
+                    }
+                return {"error": "No active version found"}
+
             elif tool_name == "add_knowledge":
                 inclusion_mode = args.get("inclusion_mode", "always")
                 knowledge = await sync_to_async(AgentKnowledge.objects.create)(

django_agent_studio/agents/dynamic.py

@@ -92,6 +92,10 @@ class DynamicAgentRuntime(AgentRuntime):
         """Execute the agent with the dynamic configuration and agentic loop."""
         config = self.config
 
+        # Check if memory is enabled (default: True)
+        extra_config = config.get("extra", {})
+        memory_enabled = extra_config.get("memory_enabled", True)
+
         # Build the messages list
         messages = []
 
@@ -108,12 +112,14 @@ class DynamicAgentRuntime(AgentRuntime):
         if rag_context:
             system_prompt = f"{system_prompt}\n\n{rag_context}"
 
-        # Add conversation memories (if we have a conversation_id and user)
-        memory_store = await self._get_memory_store(ctx)
-        if memory_store:
-            memory_context = await self._recall_memories(memory_store)
-            if memory_context:
-                system_prompt = f"{system_prompt}\n\n{memory_context}"
+        # Add conversation memories (if memory is enabled and we have context)
+        memory_store = None
+        if memory_enabled:
+            memory_store = await self._get_memory_store(ctx)
+            if memory_store:
+                memory_context = await self._recall_memories(memory_store)
+                if memory_context:
+                    system_prompt = f"{system_prompt}\n\n{memory_context}"
 
         if system_prompt:
             messages.append({"role": "system", "content": system_prompt})
@@ -121,9 +127,10 @@ class DynamicAgentRuntime(AgentRuntime):
         # Add conversation history
         messages.extend(ctx.input_messages)
 
-        # Build tool schemas - include memory tool
+        # Build tool schemas - include memory tool only if memory is enabled
         tools = self._build_tool_schemas(config)
-        tools.append(MEMORY_TOOL_SCHEMA)  # Add remember tool to all designed agents
+        if memory_enabled:
+            tools.append(MEMORY_TOOL_SCHEMA)
 
         tool_map = self._build_tool_map(config)  # Maps tool name to execution info
 

{django_agent_studio-0.1.4.dist-info → django_agent_studio-0.1.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: django-agent-studio
-Version: 0.1.4
+Version: 0.1.5
 Summary: Visual agent builder and management studio for Django - build custom GPTs with a two-pane interface
 Author: Chris Barry
 License-Expression: MIT

{django_agent_studio-0.1.4.dist-info → django_agent_studio-0.1.5.dist-info}/RECORD

@@ -3,8 +3,8 @@ django_agent_studio/apps.py,sha256=L89QWn4XOvPBs2z6qHAhaE4uZQpasJntYD75aSd2p_k,6
 django_agent_studio/urls.py,sha256=O_rrobaxpmg8gR0JmeUzQ0TI3E8mGdcz27p4OJlcI8s,743
 django_agent_studio/views.py,sha256=bQxSIeL9-D4MzKwWkeHEFXrzo1CD911qicrt_A8t_6M,3153
 django_agent_studio/agents/__init__.py,sha256=VYL_ato0DtggIo4BGRkyiz9cm1ARPXhhTQFzoG__NVM,800
-django_agent_studio/agents/builder.py,sha256=aUH5swcBYKmVByB2C9cMoTDmoGhvxmXDM0BXY6_ut1k,77147
-django_agent_studio/agents/dynamic.py,sha256=M1pSfwfujKKikdDZPcEcd7DCsM-0ZTWGRvEC9HuIWMs,13681
+django_agent_studio/agents/builder.py,sha256=Kpn16CK_5JuPJT02UO0VsTPhNusY2v4LrWA2Dcvs7WM,80594
+django_agent_studio/agents/dynamic.py,sha256=KhmdAHkYUdfuuBgsKeY-Sf9sjDA-QiBVWHpw0NAgPX0,13939
 django_agent_studio/api/__init__.py,sha256=vtBwuvBENyFFhFqCWyFsI6cYu4N9ZGqSMmHIRhr9a_U,45
 django_agent_studio/api/permissions.py,sha256=MutmA8TxZb4ZwGfeEoolK-QI04Gbcxs7DPNzkXe_Bss,5302
 django_agent_studio/api/serializers.py,sha256=rkn9xtACbJZlCBr6TLD5r-HsE1AWlaX39WYegtwEIig,18268
@@ -26,7 +26,7 @@ django_agent_studio/templates/django_agent_studio/base.html,sha256=rpHmr7CAWGwRk
 django_agent_studio/templates/django_agent_studio/builder.html,sha256=L2KX-RLVpnC2mmYsIl_aSeJIc6b63knTYVrDg8aaU1g,61425
 django_agent_studio/templates/django_agent_studio/home.html,sha256=pgwKPjiQe9UxYYLGSsKT-6erEPDUdW9MZ5D-MOjIxK4,4385
 django_agent_studio/templates/django_agent_studio/test.html,sha256=h9aTtTD1eYcgG-n410qMSryHdpXyKny0hjiKYAoGsic,3779
-django_agent_studio-0.1.4.dist-info/METADATA,sha256=WltdK6F9mB4hg3Koj_ydk-I6rEuzqPzqiBksIM0xiR0,11294
-django_agent_studio-0.1.4.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-django_agent_studio-0.1.4.dist-info/top_level.txt,sha256=O1kqZzXPOsJlqnPSAcB2fH5WpJNY8ZNfHEJzX9_SZ0A,20
-django_agent_studio-0.1.4.dist-info/RECORD,,
+django_agent_studio-0.1.5.dist-info/METADATA,sha256=_2hWCRBbGlZNCwlNNzIxeaT2jvjPiOhQI9krgm-zoOw,11294
+django_agent_studio-0.1.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+django_agent_studio-0.1.5.dist-info/top_level.txt,sha256=O1kqZzXPOsJlqnPSAcB2fH5WpJNY8ZNfHEJzX9_SZ0A,20
+django_agent_studio-0.1.5.dist-info/RECORD,,