Add followups endpoint, and handle followup suggestions

Author: akucharska

backend/companion/companionRouter.js

@@ -162,7 +162,66 @@ async function handleChatMessage(req, res) {
   }
 }
 
+async function handleFollowUpSuggestions(req, res) {
+  console.log('handleFollowUpSuggestions', req, res);
+
+  const clusterUrl = req.headers['x-cluster-url'];
+  const certificateAuthorityData =
+    req.headers['x-cluster-certificate-authority-data'];
+  const clusterToken = req.headers['x-k8s-authorization']?.replace(
+    /^Bearer\s+/i,
+    '',
+  );
+  const clientCertificateData = req.headers['x-client-certificate-data'];
+  const clientKeyData = req.headers['x-client-key-data'];
+  const sessionId = req.headers['session-id'];
+  const conversationId = sessionId;
+
+  try {
+    const baseUrl =
+      'https://companion.cp.dev.kyma.cloud.sap/api/conversations/';
+    const targetUrl = new URL(
+      `${encodeURIComponent(conversationId)}/questions`,
+      baseUrl,
+    );
+
+    const AUTH_TOKEN = await tokenManager.getToken();
+
+    const headers = {
+      Accept: 'application/json',
+      'Content-Type': 'application/json',
+      Authorization: `Bearer ${AUTH_TOKEN}`,
+      'X-Cluster-Certificate-Authority-Data': certificateAuthorityData,
+      'X-Cluster-Url': clusterUrl,
+      'Session-Id': sessionId,
+    };
+
+    if (clusterToken) {
+      headers['X-K8s-Authorization'] = clusterToken;
+    } else if (clientCertificateData && clientKeyData) {
+      headers['X-Client-Certificate-Data'] = clientCertificateData;
+      headers['X-Client-Key-Data'] = clientKeyData;
+    } else {
+      throw new Error('Missing authentication credentials');
+    }
+
+    const response = await fetch(targetUrl, {
+      method: 'GET',
+      headers,
+    });
+
+    const data = await response.json();
+    res.json({
+      promptSuggestions: data?.questions,
+      conversationId: data?.conversation_id,
+    });
+  } catch (error) {
+    res.status(500).json({ error: 'Failed to fetch AI chat data' });
+  }
+}
+
 router.post('/suggestions', handlePromptSuggestions);
 router.post('/messages', handleChatMessage);
+router.post('/followup', handleFollowUpSuggestions);
 
 export default router;

src/components/KymaCompanion/api/getFollowUpQuestions.ts

@@ -1,8 +1,10 @@
 import { getClusterConfig } from 'state/utils/getBackendInfo';
+import { PromptSuggestionsResponse } from './getPromptSuggestions';
 
 interface GetFollowUpQuestionsParams {
   sessionID?: string;
   handleFollowUpQuestions: (results: any) => void;
+  handleError: (error?: Error) => void;
   clusterUrl: string;
   token: string;
   certificateAuthorityData: string;
@@ -11,30 +13,33 @@ interface GetFollowUpQuestionsParams {
 export default async function getFollowUpQuestions({
   sessionID = '',
   handleFollowUpQuestions,
+  handleError,
   clusterUrl,
   token,
   certificateAuthorityData,
 }: GetFollowUpQuestionsParams): Promise<void> {
   try {
     const { backendAddress } = getClusterConfig();
-    const url = `${backendAddress}/api/v1/namespaces/ai-core/services/http:ai-backend-clusterip:5000/proxy/api/v1/llm/followup`;
-    const payload = JSON.parse(`{"session_id":"${sessionID}"}`);
+    const url = `${backendAddress}/ai-chat/followup`;
     const k8sAuthorization = `Bearer ${token}`;
 
-    let { results } = await fetch(url, {
+    const response = await fetch(url, {
       headers: {
         accept: 'application/json',
         'content-type': 'application/json',
         'X-Cluster-Certificate-Authority-Data': certificateAuthorityData,
         'X-Cluster-Url': clusterUrl,
         'X-K8s-Authorization': k8sAuthorization,
-        'X-User': sessionID,
+        'Session-Id': sessionID,
       },
-      body: JSON.stringify(payload),
       method: 'POST',
-    }).then(result => result.json());
-    handleFollowUpQuestions(results);
+    });
+
+    const promptSuggestions = (await response.json()).promptSuggestions;
+
+    handleFollowUpQuestions(promptSuggestions);
   } catch (error) {
+    handleError();
     console.error('Error fetching data:', error);
   }
 }

src/components/KymaCompanion/api/getPromptSuggestions.ts

@@ -8,7 +8,7 @@ interface GetPromptSuggestionsParams {
   resourceName?: string;
 }
 
-interface PromptSuggestionsResponse {
+export interface PromptSuggestionsResponse {
   promptSuggestions: string[];
   conversationId: string;
 }

src/components/KymaCompanion/components/Chat/Chat.tsx

@@ -79,6 +79,7 @@ export default function Chat() {
       getFollowUpQuestions({
         sessionID,
         handleFollowUpQuestions,
+        handleError,
         clusterUrl: cluster.currentContext.cluster.cluster.server,
         token: authData.token,
         certificateAuthorityData: