feat: adjust kb custom resource

Author: CasLubbers

src/ai/AkamaiAgentCR.test.ts

@@ -49,7 +49,12 @@ describe('AkamaiAgentCR', () => {
       expect(agentCR.metadata.labels?.['apl.io/teamId']).toBe('team-123')
       expect(agentCR.spec.foundationModel).toBe('gpt-4')
       expect(agentCR.spec.systemPrompt).toBe('You are a helpful assistant')
-      expect(agentCR.spec.knowledgeBase).toBe('test-kb')
+      expect(agentCR.spec.tools).toEqual([
+        {
+          type: 'knowledgeBase',
+          name: 'test-kb',
+        },
+      ])
     })
 
     test('should set teamId label and not merge custom labels', () => {
@@ -73,7 +78,7 @@ describe('AkamaiAgentCR', () => {
 
       const agentCR = new AkamaiAgentCR('team-123', 'test-agent', requestWithoutKB)
 
-      expect(agentCR.spec.knowledgeBase).toBeUndefined()
+      expect(agentCR.spec.tools).toBeUndefined()
     })
   })
 

src/ai/AkamaiAgentCR.ts

@@ -19,7 +19,12 @@ export class AkamaiAgentCR {
   public spec: {
     foundationModel: string
     systemPrompt: string
-    knowledgeBase?: string
+    tools?: Array<{
+      type: string
+      name: string
+      description?: string
+      endpoint?: string
+    }>
   }
 
   constructor(teamId: string, agentName: string, request: AplAgentRequest) {
@@ -38,7 +43,15 @@ export class AkamaiAgentCR {
     this.spec = {
       foundationModel: request.spec.foundationModel,
       systemPrompt: request.spec.agentInstructions,
-      knowledgeBase: request.spec.knowledgeBase,
+      tools: request.spec.knowledgeBase
+        ? [
+            {
+              type: 'knowledgeBase',
+              name: request.spec.knowledgeBase,
+              description: `Search the ${request.spec.knowledgeBase} knowledge base for relevant information. Use this when you need factual information, documentation, or specific details stored in the knowledge base.`,
+            },
+          ]
+        : undefined,
     }
   }
 
@@ -54,6 +67,9 @@ export class AkamaiAgentCR {
 
   // Transform to API response format
   toApiResponse(teamId: string): AplAgentResponse {
+    // Extract knowledgeBase from tools array (find first knowledgeBase tool)
+    const knowledgeBaseTool = this.spec.tools?.find((tool) => tool.type === 'knowledgeBase')
+
     return {
       kind: 'AkamaiAgent',
       metadata: {
@@ -66,7 +82,7 @@ export class AkamaiAgentCR {
       spec: {
         foundationModel: this.spec.foundationModel,
         agentInstructions: this.spec.systemPrompt,
-        knowledgeBase: this.spec.knowledgeBase || '',
+        knowledgeBase: knowledgeBaseTool?.name || '',
       },
       status: {
         conditions: [

src/ai/aiModelHandler.ts

@@ -14,7 +14,7 @@ function getConditions(deployment: V1Deployment) {
 
 export function transformK8sDeploymentToAplAIModel(deployment: V1Deployment): AplAIModelResponse {
   const labels = deployment.metadata?.labels || {}
-  const modelName = deployment.metadata?.name || labels.modelName
+  const modelName = labels.modelName || deployment.metadata?.name || ''
 
   // Convert K8s deployment conditions to schema format
   const conditions = getConditions(deployment)