+
{activeTab === 'mcp' && }
{activeTab === 'conversations' && }
{activeTab === 'inference' && }
diff --git a/src/contexts/MCPContext.tsx b/src/contexts/MCPContext.tsx
index 71bc612..4c3e42b 100644
--- a/src/contexts/MCPContext.tsx
+++ b/src/contexts/MCPContext.tsx
@@ -8,6 +8,7 @@ import type {
MCPServerConfig,
MCPResource,
MCPContextValue,
+ MCPMessageCallback,
} from '@/types/mcp';
import type { Tool } from '@/types/inference';
import { MCPConnectionManager } from '@/mcp/connection';
@@ -24,6 +25,33 @@ export function MCPProvider({ children }: MCPProviderProps) {
const [isLoading, setIsLoading] = useState(false);
const [error, setError] = useState(null);
const hasLoadedPersisted = useRef(false);
+
+ // Message callback management
+ const [messageCallbacks, setMessageCallbacks] = useState