fix(platform): align PlatformWorkspaceContext with WorkspaceContextProtocol

Agent-Logs-Url: https://github.com/MervinPraison/PraisonAI/sessions/4f65e6a4-b164-458e-ad64-6141c6d529c0

Co-authored-by: MervinPraison <454862+MervinPraison@users.noreply.github.com>

Author: Copilot

src/praisonai-platform/praisonai_platform/services/workspace_context.py

@@ -7,15 +7,18 @@
 
 from __future__ import annotations
 
+import json
 from typing import Any, Dict, Optional
 
 from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
+from praisonaiagents.auth.protocols import WorkspaceContextProtocol
+
 from ..db.models import Agent, Workspace
 
 
-class PlatformWorkspaceContext:
+class PlatformWorkspaceContext(WorkspaceContextProtocol):
     """
     Platform implementation of WorkspaceContextProtocol.
     
@@ -33,27 +36,34 @@ def __init__(self, workspace_id: str, session: AsyncSession):
         self.workspace_id = workspace_id
         self._session = session
 
-    async def get_workspace_context(self) -> Optional[Dict[str, Any]]:
+    async def get_workspace_context(self, workspace_id: str) -> Optional[str]:
         """
         Get workspace-level context for agents.
         
         Returns:
-            Dict containing workspace data if found, None otherwise.
-            Keys: id, name, slug, description, settings
+            Workspace context string if found, None otherwise.
         """
-        workspace = await self._session.get(Workspace, self.workspace_id)
+        if workspace_id != self.workspace_id:
+            return None
+
+        workspace = await self._session.get(Workspace, workspace_id)
         if workspace is None:
             return None
 
-        return {
+        payload = {
             "id": workspace.id,
             "name": workspace.name,
             "slug": workspace.slug,
             "description": workspace.description,
             "settings": workspace.settings or {},
         }
+        return json.dumps(payload, ensure_ascii=False, default=str)
 
-    async def get_agent_config(self, agent_id: str) -> Optional[Dict[str, Any]]:
+    async def get_agent_config(
+        self,
+        workspace_id: str,
+        agent_id: str,
+    ) -> Optional[Dict[str, Any]]:
         """
         Get agent configuration from the platform.
         
@@ -64,11 +74,14 @@ async def get_agent_config(self, agent_id: str) -> Optional[Dict[str, Any]]:
             Agent configuration dict if found, None otherwise.
             Keys: id, name, runtime_mode, instructions, config, max_concurrent_tasks
         """
+        if workspace_id != self.workspace_id:
+            return None
+
         # Query for agent scoped to the workspace
         stmt = (
             select(Agent)
             .where(Agent.id == agent_id)
-            .where(Agent.workspace_id == self.workspace_id)
+            .where(Agent.workspace_id == workspace_id)
         )
         result = await self._session.execute(stmt)
         agent = result.scalar_one_or_none()
@@ -83,4 +96,4 @@ async def get_agent_config(self, agent_id: str) -> Optional[Dict[str, Any]]:
             "instructions": agent.instructions,
             "config": agent.runtime_config or {},
             "max_concurrent_tasks": agent.max_concurrent_tasks,
-        }
+        }