mindroot 9.3.0__py3-none-any.whl → 9.5.0__py3-none-any.whl

mindroot/coreplugins/chat/__init__.py

@@ -1,4 +1,7 @@
- # This file is required to make Python treat the directory as a package.
+# This file is required to make Python treat the directory as a package.
 from .mod import *
 from .services import *
 from .commands import *
+
+# Import widget manager to ensure it's loaded
+from .widget_manager import widget_manager

mindroot/coreplugins/chat/router.py

@@ -1,5 +1,5 @@
-from fastapi import APIRouter, HTTPException, Request, Response, Depends
-from fastapi.responses import HTMLResponse, RedirectResponse
+from fastapi import APIRouter, HTTPException, Request, Response, Depends, Query
+from fastapi.responses import HTMLResponse, RedirectResponse, JSONResponse
 from fastapi import File, UploadFile, Form
 from sse_starlette.sse import EventSourceResponse
 from .models import MessageParts
@@ -22,6 +22,7 @@ import json
 from lib.chatcontext import ChatContext
 import shutil
 from pydantic import BaseModel
+from lib.auth.api_key import verify_api_key
 
 router = APIRouter()
 
@@ -52,20 +53,38 @@ async def context1(request: Request, log_id: str):
     print(context)
     return "ok"
 
-@router.get("/context2/{log_id}")
-async def context2(request: Request, log_id: str):
-    user = request.state.user.username
-    context = await get_context(log_id, user)
-    print(context)
-    return "ok"
 
-
-# need to serve persona images from ./personas/local/[persona_name]/avatar.png
-@router.get("/chat/personas/{persona_name}/avatar.png")
-async def get_persona_avatar(persona_name: str):
-    file_path = f"personas/local/{persona_name}/avatar.png"
+# need to serve persona images from ./personas/local/[persona_path]/avatar.png
+@router.get("/chat/personas/{persona_path:path}/avatar.png")
+async def get_persona_avatar(persona_path: str):
+    # Check if this is a registry persona with deduplicated assets
+    if persona_path.startswith("registry/"):
+        persona_json_path = f"personas/{persona_path}/persona.json"
+        if os.path.exists(persona_json_path):
+            try:
+                with open(persona_json_path, "r") as f:
+                    persona_data = json.load(f)
+                
+                # Check if persona has asset hashes (deduplicated storage)
+                asset_hashes = persona_data.get("asset_hashes", {})
+                if "avatar" in asset_hashes:
+                    # Redirect to deduplicated asset endpoint
+                    return RedirectResponse(f"/assets/{asset_hashes['avatar']}")
+            except Exception as e:
+                print(f"Error checking for deduplicated assets: {e}")
+    
+    # Handle registry personas: registry/owner/name
+    if persona_path.startswith('registry/'):
+        file_path = f"personas/{persona_path}/avatar.png"
+    else:
+        # Legacy support: check local first, then shared
+        file_path = f"personas/local/{persona_path}/avatar.png"
+        if not os.path.exists(file_path):
+            file_path = f"personas/registry/{persona_path}/avatar.png"
+    
     if not os.path.exists(file_path):
-        return {"error": "File not found"}
+        resolved =  os.path.realpath(file_path)
+        return {"error": "File not found: " + resolved}
         
     with open(file_path, "rb") as f:
         image_bytes = f.read()
@@ -79,6 +98,48 @@ async def get_persona_avatar(persona_name: str):
         }
     )
 
+@router.get("/chat/personas/{persona_path:path}/faceref.png")
+async def get_persona_faceref(persona_path: str):
+    # Check if this is a registry persona with deduplicated assets
+    if persona_path.startswith("registry/"):
+        persona_json_path = f"personas/{persona_path}/persona.json"
+        if os.path.exists(persona_json_path):
+            try:
+                with open(persona_json_path, "r") as f:
+                    persona_data = json.load(f)
+                
+                # Check if persona has asset hashes (deduplicated storage)
+                asset_hashes = persona_data.get("asset_hashes", {})
+                if "faceref" in asset_hashes:
+                    # Redirect to deduplicated asset endpoint
+                    return RedirectResponse(f"/assets/{asset_hashes['faceref']}")
+            except Exception as e:
+                print(f"Error checking for deduplicated assets: {e}")
+    
+    # Handle registry personas: registry/owner/name
+    if persona_path.startswith('registry/'):
+        file_path = f"personas/{persona_path}/faceref.png"
+    else:
+        # Legacy support: check local first, then shared
+        file_path = f"personas/local/{persona_path}/faceref.png"
+        if not os.path.exists(file_path):
+            file_path = f"personas/registry/{persona_path}/faceref.png"
+    
+    if not os.path.exists(file_path):
+        # Fallback to avatar if faceref doesn't exist
+        return RedirectResponse(f"/chat/personas/{persona_path}/avatar.png")
+        
+    with open(file_path, "rb") as f:
+        image_bytes = f.read()
+    
+    return Response(
+        content=image_bytes, 
+        media_type="image/png",
+        headers={
+            "Cache-Control": "max-age=3600",
+            "Content-Disposition": "inline; filename=faceref.png"
+        }
+    )
 
 @router.get("/chat/{log_id}/events")
 async def chat_events(log_id: str):
@@ -103,8 +164,26 @@ async def send_message(request: Request, log_id: str, message_parts: List[Messag
     return {"status": "ok", "task_id": task_id}
 
 @router.get("/agent/{agent_name}", response_class=HTMLResponse)
-async def get_chat_html(request: Request, agent_name: str):
-    user = request.state.user
+async def get_chat_html(request: Request, agent_name: str, api_key: str = Query(None), embed: bool = Query(False)):
+    # Handle API key authentication if provided
+    if api_key:
+        try:
+            user_data = await verify_api_key(api_key)
+            if not user_data:
+                raise HTTPException(status_code=401, detail="Invalid API key")
+            # Create a mock user object for API key users
+            class MockUser:
+                def __init__(self, username):
+                    self.username = username
+            user = MockUser(user_data['username'])
+        except Exception as e:
+            raise HTTPException(status_code=401, detail="Invalid API key")
+    else:
+        # Use regular authentication
+        if not hasattr(request.state, "user"):
+            return RedirectResponse("/login")
+        user = request.state.user
+    
     log_id = nanoid.generate()
     plugins = list_enabled()
     print("Init chat with user", user)
@@ -119,9 +198,28 @@ async def get_chat_html(request: Request, agent_name: str):
         debug_box("Access token saved to session file")
     else:
         debug_box("No access token found in request state")
-        
+    
+    # If embed mode is requested, redirect to embed session
+    if embed:
+        return RedirectResponse(f"/session/{agent_name}/{log_id}?embed=true")
+    
+    # Regular redirect
     return RedirectResponse(f"/session/{agent_name}/{log_id}")
 
+@router.get("/makesession/{agent_name}")
+async def make_session(request: Request, agent_name: str):
+    """
+    Create a new chat session for the specified agent.
+    Returns a redirect to the chat session page.
+    """
+    if not hasattr(request.state, "user"):
+        return RedirectResponse("/login")
+    user = request.state.user
+    log_id = nanoid.generate()
+    
+    await init_chat_session(user, agent_name, log_id)
+    return JSONResponse({ "log_id": log_id })
+
 @router.get("/history/{agent_name}/{log_id}")
 async def chat_history(request: Request, agent_name: str, log_id: str):
     user = request.state.user.username
@@ -137,7 +235,8 @@ async def chat_history(request: Request, agent_name: str, log_id: str):
     return history
 
 @router.get("/session/{agent_name}/{log_id}")
-async def chat_history(request: Request, agent_name: str, log_id: str):
+async def chat_session(request: Request, agent_name: str, log_id: str, embed: bool = Query(False)):
+    # Check authentication (API key or regular user)
     plugins = list_enabled()
     if not hasattr(request.state, "user"):
         return RedirectResponse("/login")
@@ -155,6 +254,10 @@ async def chat_history(request: Request, agent_name: str, log_id: str):
 
     if auth_token is not None:
         chat_data["access_token"] = auth_token
+    
+    # Add embed mode flag
+    if embed:
+        chat_data["embed_mode"] = True
 
     html = await render('chat', chat_data)
     return HTMLResponse(html)
@@ -241,6 +344,7 @@ async def get_token_count(request: Request, log_id: str):
     
     if token_counts is None:
         return {"status": "error", "message": f"Chat log with ID {log_id} not found"}
+
     
     return {"status": "ok", "token_counts": token_counts}
 
@@ -290,9 +394,9 @@ async def delete_chat_session(request: Request, log_id: str, user=Depends(requir
 
 
 @router.get("/chat/{log_id}/tokens")
-async def get_token_count(request: Request, log_id: str):
+async def get_token_count_alt(request: Request, log_id: str):
     """
-    Get token counts for a chat log identified by log_id, including any delegated tasks.
+    Alternative token count endpoint using token_counter module.
     
     Parameters:
     - log_id: The log ID to count tokens for
@@ -306,5 +410,13 @@ async def get_token_count(request: Request, log_id: str):
     
     if token_counts is None:
         return {"status": "error", "message": f"Chat log with ID {log_id} not found"}
+
     
     return {"status": "ok", "token_counts": token_counts}
+
+# Include widget routes
+try:
+    from .widget_routes import router as widget_router
+    router.include_router(widget_router)
+except ImportError as e:
+    print(f"Warning: Could not load widget routes: {e}")

mindroot/coreplugins/chat/router_dedup_patch.py

@@ -0,0 +1,20 @@
+# Patch for chat router to add deduplication support
+# This should be integrated into the main router.py file
+
+# Add this code to the get_persona_avatar function after line 59:
+
+# Check if this is a registry persona with deduplicated assets
+if persona_path.startswith('registry/'):
+    persona_json_path = f"personas/{persona_path}/persona.json"
+    if os.path.exists(persona_json_path):
+        try:
+            with open(persona_json_path, 'r') as f:
+                persona_data = json.load(f)
+            
+            # Check if persona has asset hashes (deduplicated storage)
+            asset_hashes = persona_data.get('asset_hashes', {})
+            if 'avatar' in asset_hashes:
+                # Redirect to deduplicated asset endpoint
+                return RedirectResponse(f"/assets/{asset_hashes['avatar']}")
+        except Exception as e:
+            print(f"Error checking for deduplicated assets: {e}")

mindroot/coreplugins/chat/services.py

@@ -23,6 +23,36 @@ import base64
 import nanoid
 sse_clients = {}
 
+@service()
+async def prompt(model: str, instructions: str, temperature=0, max_tokens=400, json=False, context=None):
+    messages = [
+        { "role": "system",
+          "content": "Respond to prompt with no extraneous commentary."
+        },
+        { "role": "user",
+          "content": instructions
+        }]
+    
+    stream = await context.stream_chat(model, temperature=temperature,
+                                       max_tokens=max_tokens,
+                                       messages=messages,
+                                       json=False,
+                                       context=context)
+    text = ""
+    if os.environ.get("AH_DEBUG") == "True":
+        print("Prompting, instructions ", instructions)
+    async for chunk in stream:
+        #print("Chunk received: ", chunk)
+        if chunk is None or chunk == "":
+            continue
+        else:
+            text += chunk
+            if os.environ.get("AH_DEBUG") == "True":
+                print(chunk, end='', flush=True)
+
+    return text
+
+
 def results_text(results):
     text = ""
     for result in results:
@@ -304,7 +334,7 @@ async def send_message_to_agent(session_id: str, message: str | List[MessagePart
                 actual_results = False
                 await asyncio.sleep(0.001)
                 for result in results:
-                    if result['result'] is not None:
+                    if 'result' in result and result['result'] is not None:
                         if result['result'] == 'continue':
                             out_results.append(result)
                             continue_processing = True

mindroot/coreplugins/chat/static/css/action-fix.css

@@ -0,0 +1,32 @@
+/* Action component full width fix - ONLY target action components */
+action-component {
+  display: block !important;
+  width: 100% !important;
+}
+
+action-component .action-details {
+  width: 100% !important;
+  display: block !important;
+}
+
+action-component .action-details summary {
+  width: 100% !important;
+  display: block !important;
+  box-sizing: border-box !important;
+}
+
+/* Action summary styling - brighter text */
+action-component .action-summary {
+  background: #333 !important;
+  color: #f0f0f0 !important;
+}
+
+action-component .param-preview {
+  color: #ddd !important;
+}
+
+action-component .fn_name {
+  color: #f0f0f0 !important;
+}
+
+/* DO NOT TOUCH CHAT MESSAGE ALIGNMENT - Let default.css handle it */

mindroot/coreplugins/chat/static/css/admin-custom.css

@@ -15,11 +15,12 @@ body {
 }
 
 details {
-  background: rgb(10, 10, 25);
+  background: #fff;
   border: 1px solid rgba(255, 255, 255, 0.1);
   border-radius: 8px;
   margin-bottom: 20px;
   overflow: hidden;
+  width: 100%;
   transition: all 0.3s ease;
 }
 
@@ -29,7 +30,7 @@ details[open] {
 
 summary {
   padding: 15px 20px;
-  background: rgb(15, 15, 30);
+  background: #222;
   color: #f0f0f0;
   cursor: pointer;
   transition: background 0.3s ease;
@@ -42,7 +43,8 @@ summary:hover {
 }
 
 details[open] summary {
-  background: rgb(25, 25, 50);
+  /* background: rgb(25, 25, 50); */
+  background: #222;
 }
 
 .details-content {

mindroot/coreplugins/chat/static/css/dark.css

@@ -3,7 +3,7 @@ html, body {
   color: #f0f0f0;
   padding: 5px 5px;
   font-family: ui-sans-serif, -apple-system, system-ui, "Segoe UI", Roboto, Ubuntu, Cantarell, "Noto Sans", sans-serif, Helvetica, "Apple Color Emoji", Arial, "Segoe UI Emoji", "Segoe UI Symbol";
-  /* font-size: 16px; */
+  font-size: 18px;
   display: flex;
   line-height: 1.5;
   flex-direction: column;
@@ -89,8 +89,10 @@ chat-form {
   width: 100%;
 }
 
+
 chat-message {
-  max-width: 69%;
+  /* max-width: 69%; */
+  width: 100%;
   padding: 4px 8px;
   margin: 2px 0;
   display: block;
@@ -127,6 +129,15 @@ chat-message:first-of-type .outer-msg.without-avatar {
   background-color:  transparent;
 }
 
+.outer-msg.user {
+  flex-direction: column;
+  align-items: flex-end;
+}
+
+.message:has(.action-summary) {
+  width: 100%;
+}
+
 .user {
   align-self: flex-end;
 }
@@ -146,6 +157,7 @@ chat-message:first-of-type .outer-msg.without-avatar {
   /* background-color: #080808; */
   border: 1px solid #181818;
   border-radius: 30px;
+  width: 80%;
 }
 
 .msg-ai:first-child {
@@ -745,7 +757,16 @@ button:hover {
 }
 
 .param_name { font-weight: bold; }
-.param_value { color: yellow; }
+.param_value { color: #ccc; }
+
+summary {
+  border-radius: 8px;
+  color: white;
+}
+
+details {
+  width: 100%;
+}
 
 .fn_result, .code-result { 
   white-space: normal !important;

mindroot/coreplugins/chat/static/css/default.css

@@ -3,7 +3,7 @@ html, body {
   color: #f0f0f0;
   padding: 5px 5px;
   font-family: ui-sans-serif, -apple-system, system-ui, "Segoe UI", Roboto, Ubuntu, Cantarell, "Noto Sans", sans-serif, Helvetica, "Apple Color Emoji", Arial, "Segoe UI Emoji", "Segoe UI Symbol";
-  /* font-size: 16px; */
+  font-size: 18px;
   display: flex;
   line-height: 1.5;
   flex-direction: column;
@@ -89,8 +89,10 @@ chat-form {
   width: 100%;
 }
 
+
 chat-message {
-  max-width: 69%;
+  /* max-width: 69%; */
+  width: 100%;
   padding: 4px 8px;
   margin: 2px 0;
   display: block;
@@ -127,6 +129,15 @@ chat-message:first-of-type .outer-msg.without-avatar {
   background-color:  transparent;
 }
 
+.outer-msg.user {
+  flex-direction: column;
+  align-items: flex-end;
+}
+
+.message:has(.action-summary) {
+  width: 100%;
+}
+
 .user {
   align-self: flex-end;
 }
@@ -146,6 +157,7 @@ chat-message:first-of-type .outer-msg.without-avatar {
   /* background-color: #080808; */
   border: 1px solid #181818;
   border-radius: 30px;
+  width: 80%;
 }
 
 .msg-ai:first-child {
@@ -745,7 +757,16 @@ button:hover {
 }
 
 .param_name { font-weight: bold; }
-.param_value { color: yellow; }
+.param_value { color: #ccc; }
+
+summary {
+  border-radius: 8px;
+  color: white;
+}
+
+details {
+  width: 100%;
+}
 
 .fn_result, .code-result { 
   white-space: normal !important;

mindroot/coreplugins/chat/static/css/main.css

@@ -2,3 +2,4 @@
 @import url('reset.css');
 @import url('default.css');
 @import url('mobile.css');
+@import url('action-fix.css');