bohr-agent-sdk 0.1.100__py3-none-any.whl → 0.1.102__py3-none-any.whl

dp/agent/cli/templates/ui/api/messages.py

@@ -0,0 +1,161 @@
+"""
+User-facing messages for API modules
+Supports internationalization - Chinese and English messages
+"""
+
+# Language detection helper
+def get_message(messages_dict, lang='zh'):
+    """Get message in specified language, fallback to Chinese"""
+    return messages_dict.get(lang, messages_dict.get('zh'))
+
+# Error messages
+ERROR_MESSAGES = {
+    'access_denied': {
+        'zh': '访问被拒绝',
+        'en': 'Access denied'
+    },
+    'invalid_file_path': {
+        'zh': '无效的文件路径',
+        'en': 'Invalid file path'
+    },
+    'file_not_found': {
+        'zh': '文件未找到',
+        'en': 'File not found'
+    },
+    'folder_not_found': {
+        'zh': '文件夹未找到',
+        'en': 'Folder not found'
+    },
+    'file_or_folder_not_found': {
+        'zh': '文件或文件夹不存在',
+        'en': 'File or folder does not exist'
+    },
+    'decode_error': {
+        'zh': '无法解码文件内容',
+        'en': 'Unable to decode file content'
+    },
+    'session_not_exist': {
+        'zh': '会话不存在',
+        'en': 'Session does not exist'
+    },
+    'delete_session_failed': {
+        'zh': '删除会话失败',
+        'en': 'Failed to delete session'
+    },
+    'invalid_project_id': {
+        'zh': '无效的 Project ID: {project_id}，必须是整数',
+        'en': 'Invalid Project ID: {project_id}, must be an integer'
+    },
+    'project_id_required': {
+        'zh': '请先设置项目 ID 后再上传文件。',
+        'en': 'Please set project ID before uploading files.'
+    },
+    'unsupported_file_type': {
+        'zh': '不支持的文件类型: {file_ext}',
+        'en': 'Unsupported file type: {file_ext}'
+    },
+    'file_too_large': {
+        'zh': '文件 {filename} 超过大小限制 (10MB)',
+        'en': 'File {filename} exceeds size limit (10MB)'
+    },
+    'no_permission_project': {
+        'zh': '您没有权限使用项目 ID: {project_id}。请从项目列表中选择您有权限的项目。',
+        'en': 'You do not have permission to use project ID: {project_id}. Please select a project you have permission for.'
+    },
+    'accesskey_not_found': {
+        'zh': '未找到 AccessKey',
+        'en': 'AccessKey not found'
+    },
+    'appkey_not_found': {
+        'zh': '未找到 AppKey',
+        'en': 'AppKey not found'
+    },
+    'accesskey_or_appkey_not_found': {
+        'zh': '未找到 AccessKey 或 AppKey',
+        'en': 'AccessKey or AppKey not found'
+    },
+    'get_project_list_failed': {
+        'zh': '获取项目列表失败',
+        'en': 'Failed to get project list'
+    },
+    'project_not_belong_to_user': {
+        'zh': '该项目不属于当前用户',
+        'en': 'This project does not belong to current user'
+    },
+    'temp_user_no_session': {
+        'zh': '临时用户没有历史会话',
+        'en': 'Temporary users have no session history'
+    },
+    'temp_user_cannot_export': {
+        'zh': '临时用户没有会话可导出',
+        'en': 'Temporary users have no sessions to export'
+    },
+    'no_session_found': {
+        'zh': '没有找到会话',
+        'en': 'No sessions found'
+    },
+    'clear_failed': {
+        'zh': '清除失败: {error}',
+        'en': 'Clear failed: {error}'
+    },
+    'export_failed': {
+        'zh': '导出失败: {error}',
+        'en': 'Export failed: {error}'
+    },
+    'no_active_session': {
+        'zh': '没有活动的会话',
+        'en': 'No active session'
+    },
+    'session_service_not_initialized': {
+        'zh': '会话服务未初始化',
+        'en': 'Session service not initialized'
+    },
+    'get_session_list_failed': {
+        'zh': '获取会话列表失败',
+        'en': 'Failed to get session list'
+    },
+    'get_session_messages_failed': {
+        'zh': '获取会话消息失败',
+        'en': 'Failed to get session messages'
+    },
+    'please_set_project_id': {
+        'zh': '🔒 请先设置项目 ID',
+        'en': '🔒 Please set project ID first'
+    }
+}
+
+# Success messages
+SUCCESS_MESSAGES = {
+    'project_id_set': {
+        'zh': 'Project ID 已设置为: {project_id}',
+        'en': 'Project ID set to: {project_id}'
+    },
+    'project_id_set_from_env': {
+        'zh': 'Project ID 已从环境变量设置为: {project_id} (开发模式)',
+        'en': 'Project ID set from environment variable: {project_id} (development mode)'
+    },
+    'delete_success': {
+        'zh': '成功删除: {filename}',
+        'en': 'Successfully deleted: {filename}'
+    },
+    'session_cleared': {
+        'zh': '历史会话已清除',
+        'en': 'Session history cleared'
+    },
+    'no_session_to_clear': {
+        'zh': '没有找到历史会话',
+        'en': 'No session history found'
+    }
+}
+
+# UI labels
+UI_LABELS = {
+    'workspace': {
+        'zh': '工作空间',
+        'en': 'Workspace'
+    },
+    'websocket_server_running': {
+        'zh': '{agent_name} WebSocket 服务器正在运行',
+        'en': '{agent_name} WebSocket server is running'
+    }
+}

dp/agent/cli/templates/ui/api/projects.py

@@ -0,0 +1,146 @@
+# Projects API using bohrium-open-sdk
+from typing import Dict, Any, List, Optional
+
+from fastapi import APIRouter, Request
+from bohrium_open_sdk import OpenSDK
+
+from server.utils import get_ak_info_from_request
+
+router = APIRouter()
+
+
+def make_client_from_request(request: Request) -> OpenSDK:
+    """Parse AccessKey/AppKey from request headers and create SDK client"""
+    access_key, app_key = get_ak_info_from_request(request.headers)
+    if not access_key:
+        raise ValueError("未找到 AccessKey")
+    if not app_key:
+        raise ValueError("未找到 AppKey")
+    return OpenSDK(access_key=access_key, app_key=app_key)
+
+
+def normalize_project(item: Dict[str, Any]) -> Dict[str, Any]:
+    """
+    Normalize project structure for frontend:
+    - id: item['project_id'] or item['id']
+    - name: item['project_name'] or item['name']
+    Set default values for missing fields
+    """
+    pid = item.get("project_id", item.get("id"))
+    name = item.get("project_name", item.get("name"))
+    return {
+        "id": pid,
+        "name": name,
+        "creatorName": item.get("creatorName", ""),
+        "createTime": item.get("createTime", ""),
+        "projectRole": item.get("projectRole", 0),
+    }
+
+
+async def verify_user_project(access_key: str, app_key: str, project_id: int) -> bool:
+    """
+    Use bohrium-open-sdk to verify if project_id belongs to current user
+    """
+    if not access_key or not app_key or not project_id:
+        return False
+
+    try:
+        client = OpenSDK(access_key=access_key, app_key=app_key)
+        res = client.user.list_project()
+        if res.get("code") != 0:
+            return False
+
+        items: List[Dict[str, Any]] = res.get("data", {}).get("items", []) or []
+        user_project_ids = [it.get("project_id", it.get("id")) for it in items]
+
+        # Compatible with string/integer ID
+        user_project_ids_int: List[int] = []
+        for pid in user_project_ids:
+            try:
+                user_project_ids_int.append(int(pid))
+            except Exception:
+                pass
+
+        return int(project_id) in user_project_ids_int
+
+    except Exception as e:
+        return False
+
+
+@router.get("/api/projects")
+async def get_projects(request: Request) -> Dict[str, Any]:
+    """Get user's project list using SDK"""
+    try:
+        client = make_client_from_request(request)
+        user_info = client.user.get_info()
+        # Response data contains: user_id, name, org_id
+        response = f"Get userid success,the user_id is: {user_info['data']['user_id']}"
+        print(response)
+        res = client.user.list_project()
+        if res.get("code") != 0:
+            return {
+                "success": False,
+                "error": res.get("error", "获取项目列表失败"),
+                "projects": [],
+            }
+
+        items: List[Dict[str, Any]] = res.get("data", {}).get("items", []) or []
+        projects = [normalize_project(it) for it in items]
+
+        return {"success": True, "projects": projects}
+
+    except ValueError as ve:
+        return {"success": False, "error": str(ve), "projects": []}
+    except Exception as e:
+        return {"success": False, "error": str(e), "projects": []}
+
+
+@router.get("/api/projects/{project_id}/verify")
+async def verify_project(project_id: int, request: Request) -> Dict[str, Any]:
+    """Verify if specific project_id belongs to current user"""
+    try:
+        access_key, app_key = get_ak_info_from_request(request.headers)
+
+        if not access_key or not app_key:
+            return {
+                "success": False,
+                "error": "未找到 AccessKey 或 AppKey",
+                "valid": False,
+            }
+
+        is_valid = await verify_user_project(access_key, app_key, project_id)
+
+        if not is_valid:
+            return {
+                "success": True,
+                "valid": False,
+                "error": "该项目不属于当前用户",
+            }
+
+        # Find specific project info and return
+        client = OpenSDK(access_key=access_key, app_key=app_key)
+        res = client.user.list_project()
+        if res.get("code") != 0:
+            return {
+                "success": False,
+                "error": res.get("error", "获取项目列表失败"),
+                "valid": False,
+            }
+
+        items: List[Dict[str, Any]] = res.get("data", {}).get("items", []) or []
+        project_info: Optional[Dict[str, Any]] = None
+        for it in items:
+            raw_id = it.get("project_id", it.get("id"))
+            try:
+                if int(raw_id) == int(project_id):
+                    project_info = normalize_project(it)
+                    break
+            except Exception:
+                continue
+
+        return {"success": True, "valid": True, "project": project_info or {"id": project_id}}
+
+    except ValueError as ve:
+        return {"success": False, "error": str(ve), "valid": False}
+    except Exception as e:
+        return {"success": False, "error": str(e), "valid": False}

dp/agent/cli/templates/ui/api/sessions.py

@@ -0,0 +1,93 @@
+# Session management API
+import json
+import shutil
+from datetime import datetime
+from fastapi import Request
+from fastapi.responses import JSONResponse, Response
+
+from server.utils import get_ak_info_from_request
+from api.websocket import manager
+
+
+async def clear_user_sessions(request: Request):
+    """Clear all historical sessions for current user"""
+    access_key, _ = get_ak_info_from_request(request.headers)
+    
+    if not access_key:
+        return JSONResponse(
+            content={"error": "临时用户没有历史会话"},
+            status_code=400
+        )
+    
+    try:
+        # Get user's session directory
+        ak_hash = manager.persistent_manager._get_ak_hash(access_key)
+        user_sessions_dir = manager.persistent_manager.ak_sessions_dir / ak_hash / "sessions"
+        
+        if user_sessions_dir.exists():
+            # Delete all session files
+            shutil.rmtree(user_sessions_dir)
+            user_sessions_dir.mkdir(parents=True, exist_ok=True)
+            
+            return JSONResponse(content={
+                "message": "历史会话已清除",
+                "status": "success"
+            })
+        else:
+            return JSONResponse(content={
+                "message": "没有找到历史会话",
+                "status": "success"
+            })
+            
+    except Exception as e:
+        return JSONResponse(
+            content={"error": f"清除失败: {str(e)}"},
+            status_code=500
+        )
+
+
+async def export_user_sessions(request: Request):
+    """Export all sessions for current user"""
+    access_key, _ = get_ak_info_from_request(request.headers)
+    
+    if not access_key:
+        return JSONResponse(
+            content={"error": "临时用户没有会话可导出"},
+            status_code=400
+        )
+    
+    try:
+        # Load all user sessions
+        sessions = await manager.persistent_manager.load_user_sessions(access_key)
+        
+        if not sessions:
+            return JSONResponse(
+                content={"error": "没有找到会话"},
+                status_code=404
+            )
+        
+        # Build export data
+        export_data = {
+            "export_time": datetime.now().isoformat(),
+            "user_type": "registered",
+            "sessions": []
+        }
+        
+        for session in sessions.values():
+            session_data = manager.persistent_manager._serialize_session(session)
+            export_data["sessions"].append(session_data)
+        
+        # Return JSON file
+        return Response(
+            content=json.dumps(export_data, indent=2, ensure_ascii=False),
+            media_type="application/json",
+            headers={
+                "Content-Disposition": f"attachment; filename=sessions_export_{datetime.now().strftime('%Y%m%d_%H%M%S')}.json"
+            }
+        )
+        
+    except Exception as e:
+        return JSONResponse(
+            content={"error": f"导出失败: {str(e)}"},
+            status_code=500
+        )

dp/agent/cli/templates/ui/api/utils.py

@@ -0,0 +1,161 @@
+"""
+Common utilities for API modules
+"""
+import os
+from pathlib import Path
+from typing import Optional, Tuple
+from fastapi import Request
+from fastapi.responses import JSONResponse
+
+from server.utils import get_ak_info_from_request
+from server.user_files import UserFileManager
+from api.messages import ERROR_MESSAGES, get_message
+
+
+def get_user_identifier(access_key: Optional[str], app_key: Optional[str], session_id: Optional[str]) -> str:
+    """
+    Get unique user identifier based on access_key or session_id
+    
+    Args:
+        access_key: User's access key
+        app_key: User's app key
+        session_id: Session ID for temporary users
+        
+    Returns:
+        Unique user identifier string
+    """
+    if access_key:
+        return access_key
+    return session_id or "anonymous"
+
+
+def extract_session_id_from_request(request: Request) -> Optional[str]:
+    """
+    Extract session_id from request cookies
+    
+    Args:
+        request: FastAPI request object
+        
+    Returns:
+        Session ID if found, None otherwise
+    """
+    cookie_header = request.headers.get("cookie", "")
+    if not cookie_header:
+        return None
+        
+    from http.cookies import SimpleCookie
+    simple_cookie = SimpleCookie()
+    simple_cookie.load(cookie_header)
+    
+    if "session_id" in simple_cookie:
+        return simple_cookie["session_id"].value
+    return None
+
+
+def get_user_context_from_request(request: Request) -> Tuple[str, Optional[str], Optional[str]]:
+    """
+    Extract user context from request
+    
+    Returns:
+        Tuple of (user_identifier, access_key, app_key)
+    """
+    access_key, app_key = get_ak_info_from_request(request.headers)
+    session_id = extract_session_id_from_request(request)
+    user_identifier = get_user_identifier(access_key, app_key, session_id)
+    
+    return user_identifier, access_key, app_key
+
+
+def validate_file_access(file_path: Path, user_files_dir: Path, lang='zh') -> Optional[JSONResponse]:
+    """
+    Validate if file path is within user directory and accessible
+    
+    Args:
+        file_path: Path to validate
+        user_files_dir: User's files directory
+        lang: Language for error messages
+        
+    Returns:
+        JSONResponse with error if validation fails, None if valid
+    """
+    try:
+        # Resolve paths to absolute
+        file_resolved = file_path.resolve()
+        user_files_dir_resolved = user_files_dir.resolve()
+        
+        # Check if file is within user directory
+        if not str(file_resolved).startswith(str(user_files_dir_resolved)):
+            return JSONResponse(
+                content={"error": get_message(ERROR_MESSAGES['access_denied'], lang)},
+                status_code=403
+            )
+            
+        # Check if file exists
+        if not file_resolved.exists():
+            error_key = 'folder_not_found' if file_path.is_dir() else 'file_not_found'
+            return JSONResponse(
+                content={"error": get_message(ERROR_MESSAGES[error_key], lang)},
+                status_code=404
+            )
+            
+        return None
+        
+    except Exception:
+        return JSONResponse(
+            content={"error": get_message(ERROR_MESSAGES['invalid_file_path'], lang)},
+            status_code=400
+        )
+
+
+def process_file_path(file_path_str: str, user_files_dir: Path) -> Path:
+    """
+    Process file path string and convert to Path object
+    
+    Args:
+        file_path_str: File path as string
+        user_files_dir: User's files directory
+        
+    Returns:
+        Processed Path object
+    """
+    if file_path_str.startswith('/'):
+        return Path(file_path_str)
+    else:
+        # Relative path, based on user directory
+        return user_files_dir / file_path_str
+
+
+def safe_filename(filename: str) -> str:
+    """
+    Generate safe filename by replacing dangerous characters
+    
+    Args:
+        filename: Original filename
+        
+    Returns:
+        Safe filename
+    """
+    return filename.replace('/', '_').replace('\\', '_').replace('..', '_')
+
+
+async def check_project_id_required(context_manager, user_identifier: str) -> bool:
+    """
+    Check if project_id is required and set
+    
+    Args:
+        context_manager: WebSocket manager instance
+        user_identifier: User identifier
+        
+    Returns:
+        True if project_id is set, False otherwise
+    """
+    # First check environment variable
+    if os.environ.get('BOHR_PROJECT_ID'):
+        return True
+        
+    # Check user's connection context
+    for context in context_manager.active_connections.values():
+        if context.get_user_identifier() == user_identifier:
+            return bool(context.project_id)
+            
+    return False