Llm integration

pages/_app.tsx

@@ -78,6 +78,32 @@ function CustomApp({
                 />
             ),
         },
+        {
+            name: 'Search',
+            id: '10',
+            isSelected: router.pathname === '/search',
+            href: '/search',
+            renderItem: () => (
+                <WfoMenuItemLink
+                    path={'/search'}
+                    translationString="Search"
+                    isSelected={router.pathname === '/search'}
+                />
+            ),
+        },
+        {
+            name: 'Agent',
+            id: '10',
+            isSelected: router.pathname === '/agent',
+            href: '/agent',
+            renderItem: () => (
+                <WfoMenuItemLink
+                    path={'/agent'}
+                    translationString="Agent"
+                    isSelected={router.pathname === '/agent'}
+                />
+            ),
+        },
     ];
 
     const errorMonitoringHandler: WfoErrorMonitoring | undefined = {

pages/agent.tsx

@@ -0,0 +1,11 @@
+import { CopilotKit } from '@copilotkit/react-core';
+import '@copilotkit/react-ui/styles.css';
+import { WfoAgent } from '@orchestrator-ui/orchestrator-ui-components';
+
+export default function SearchPage() {
+    return (
+        <CopilotKit runtimeUrl="/api/copilotkit" agent="query_agent">
+            <WfoAgent />
+        </CopilotKit>
+    );
+}

pages/api/copilotkit.ts

@@ -0,0 +1,47 @@
+import { NextApiRequest, NextApiResponse } from 'next';
+import { getServerSession } from 'next-auth';
+
+import { HttpAgent } from '@ag-ui/client';
+import {
+    CopilotRuntime,
+    ExperimentalEmptyAdapter,
+    copilotRuntimeNextJSPagesRouterEndpoint,
+} from '@copilotkit/runtime';
+import { WfoSession } from '@orchestrator-ui/orchestrator-ui-components';
+
+import { authOptions } from './auth/[...nextauth]';
+
+const serviceAdapter = new ExperimentalEmptyAdapter();
+
+export default async function copilotHandler(
+    req: NextApiRequest,
+    res: NextApiResponse,
+) {
+    const session = (await getServerSession(
+        req,
+        res,
+        authOptions,
+    )) as WfoSession;
+
+    const headers: Record<string, string> = {};
+    if (session?.accessToken) {
+        headers['Authorization'] = `Bearer ${session.accessToken}`;
+    }
+
+    const runtime = new CopilotRuntime({
+        agents: {
+            query_agent: new HttpAgent({
+                url: process.env.AGENT_URL || 'http://localhost:8080/agent/',
+                headers,
+            }),
+        },
+    });
+
+    const handler = copilotRuntimeNextJSPagesRouterEndpoint({
+        runtime,
+        serviceAdapter,
+        endpoint: '/api/copilotkit',
+    });
+
+    return handler(req, res);
+}

pages/search.tsx

@@ -0,0 +1,7 @@
+import React from 'react';
+
+import { WfoSearch } from '@orchestrator-ui/orchestrator-ui-components';
+
+export default function SearchPage() {
+    return <WfoSearch />;
+}