style: fix prettier formatting across all changed files

Author: tejaskash

src/cli/aws/__tests__/bedrock-import.test.ts

@@ -1,4 +1,5 @@
-import { describe, expect, it, vi, beforeEach } from 'vitest';
+import { getBedrockAgentConfig, listBedrockAgentAliases, listBedrockAgents } from '../bedrock-import';
+import { beforeEach, describe, expect, it, vi } from 'vitest';
 
 const mockAgentSend = vi.fn();
 const mockBedrockSend = vi.fn();
@@ -7,7 +8,9 @@ const mockS3Send = vi.fn();
 // Mock the AWS SDK clients using class syntax for proper `new` support
 vi.mock('@aws-sdk/client-bedrock-agent', () => {
   return {
-    BedrockAgentClient: class { send = mockAgentSend; },
+    BedrockAgentClient: class {
+      send = mockAgentSend;
+    },
     ListAgentsCommand: class {},
     ListAgentAliasesCommand: class {},
     GetAgentCommand: class {},
@@ -22,15 +25,19 @@ vi.mock('@aws-sdk/client-bedrock-agent', () => {
 
 vi.mock('@aws-sdk/client-bedrock', () => {
   return {
-    BedrockClient: class { send = mockBedrockSend; },
+    BedrockClient: class {
+      send = mockBedrockSend;
+    },
     GetFoundationModelCommand: class {},
     GetGuardrailCommand: class {},
   };
 });
 
 vi.mock('@aws-sdk/client-s3', () => {
   return {
-    S3Client: class { send = mockS3Send; },
+    S3Client: class {
+      send = mockS3Send;
+    },
     GetObjectCommand: class {},
   };
 });
@@ -43,8 +50,6 @@ vi.mock('js-yaml', () => ({
   default: { load: vi.fn((s: string) => JSON.parse(s)) },
 }));
 
-import { listBedrockAgents, listBedrockAgentAliases, getBedrockAgentConfig } from '../bedrock-import';
-
 describe('bedrock-import', () => {
   beforeEach(() => {
     vi.clearAllMocks();
@@ -77,15 +82,11 @@ describe('bedrock-import', () => {
   describe('listBedrockAgentAliases', () => {
     it('returns mapped alias summaries', async () => {
       mockAgentSend.mockResolvedValueOnce({
-        agentAliasSummaries: [
-          { agentAliasId: 'alias-1', agentAliasName: 'prod', description: 'Production' },
-        ],
+        agentAliasSummaries: [{ agentAliasId: 'alias-1', agentAliasName: 'prod', description: 'Production' }],
       });
 
       const result = await listBedrockAgentAliases('us-east-1', 'agent-1');
-      expect(result).toEqual([
-        { aliasId: 'alias-1', aliasName: 'prod', description: 'Production' },
-      ]);
+      expect(result).toEqual([{ aliasId: 'alias-1', aliasName: 'prod', description: 'Production' }]);
     });
   });
 

src/cli/aws/bedrock-import.ts

@@ -2,7 +2,7 @@
  * AWS SDK v3 wrapper for fetching Bedrock Agent configurations.
  * Port of the starter toolkit's agent_info.py.
  */
-
+import { getCredentialProvider } from './account';
 import type {
   ActionGroupInfo,
   BedrockAgentConfig,
@@ -12,21 +12,20 @@ import type {
   CollaboratorInfo,
   KnowledgeBaseInfo,
 } from './bedrock-import-types';
+import { BedrockClient, GetFoundationModelCommand, GetGuardrailCommand } from '@aws-sdk/client-bedrock';
 import {
   BedrockAgentClient,
+  GetAgentActionGroupCommand,
   GetAgentAliasCommand,
   GetAgentCommand,
-  GetAgentActionGroupCommand,
   GetKnowledgeBaseCommand,
   ListAgentActionGroupsCommand,
   ListAgentAliasesCommand,
   ListAgentCollaboratorsCommand,
   ListAgentKnowledgeBasesCommand,
   ListAgentsCommand,
 } from '@aws-sdk/client-bedrock-agent';
-import { BedrockClient, GetFoundationModelCommand, GetGuardrailCommand } from '@aws-sdk/client-bedrock';
 import { GetObjectCommand, S3Client } from '@aws-sdk/client-s3';
-import { getCredentialProvider } from './account';
 import yaml from 'js-yaml';
 
 function createBedrockAgentClient(region: string): BedrockAgentClient {
@@ -172,9 +171,7 @@ async function fetchActionGroups(
   agentId: string,
   agentVersion: string
 ): Promise<ActionGroupInfo[]> {
-  const listResponse = await client.send(
-    new ListAgentActionGroupsCommand({ agentId, agentVersion })
-  );
+  const listResponse = await client.send(new ListAgentActionGroupsCommand({ agentId, agentVersion }));
   const summaries = listResponse.actionGroupSummaries ?? [];
   const actionGroups: ActionGroupInfo[] = [];
 
@@ -230,17 +227,13 @@ async function fetchKnowledgeBases(
   agentId: string,
   agentVersion: string
 ): Promise<KnowledgeBaseInfo[]> {
-  const listResponse = await client.send(
-    new ListAgentKnowledgeBasesCommand({ agentId, agentVersion })
-  );
+  const listResponse = await client.send(new ListAgentKnowledgeBasesCommand({ agentId, agentVersion }));
   const summaries = listResponse.agentKnowledgeBaseSummaries ?? [];
   const knowledgeBases: KnowledgeBaseInfo[] = [];
 
   for (const summary of summaries) {
     try {
-      const kbDetail = await client.send(
-        new GetKnowledgeBaseCommand({ knowledgeBaseId: summary.knowledgeBaseId! })
-      );
+      const kbDetail = await client.send(new GetKnowledgeBaseCommand({ knowledgeBaseId: summary.knowledgeBaseId! }));
       const kb = kbDetail.knowledgeBase;
       knowledgeBases.push({
         knowledgeBaseId: summary.knowledgeBaseId!,
@@ -276,9 +269,7 @@ async function fetchCollaborators(
   }
 
   try {
-    const listResponse = await agentClient.send(
-      new ListAgentCollaboratorsCommand({ agentId, agentVersion })
-    );
+    const listResponse = await agentClient.send(new ListAgentCollaboratorsCommand({ agentId, agentVersion }));
     const summaries = listResponse.agentCollaboratorSummaries ?? [];
     const collaborators: CollaboratorInfo[] = [];
 
@@ -295,9 +286,13 @@ async function fetchCollaborators(
       const collabConfig = await getBedrockAgentConfig(region, collabAgentId, collabAliasId);
       const collabInfo: CollaboratorInfo = {
         ...collabConfig,
-        collaboratorName: cleanVariableName((summary as unknown as { collaboratorName?: string }).collaboratorName ?? ''),
-        collaborationInstruction: (summary as unknown as { collaborationInstruction?: string }).collaborationInstruction ?? '',
-        relayConversationHistory: (summary as unknown as { relayConversationHistory?: string }).relayConversationHistory ?? 'DISABLED',
+        collaboratorName: cleanVariableName(
+          (summary as unknown as { collaboratorName?: string }).collaboratorName ?? ''
+        ),
+        collaborationInstruction:
+          (summary as unknown as { collaborationInstruction?: string }).collaborationInstruction ?? '',
+        relayConversationHistory:
+          (summary as unknown as { relayConversationHistory?: string }).relayConversationHistory ?? 'DISABLED',
       };
       collaborators.push(collabInfo);
     }

src/cli/commands/create/action.ts

@@ -1,5 +1,4 @@
 import { APP_DIR, CONFIG_DIR, ConfigIO, setEnvVar, setSessionProjectRoot } from '../../../lib';
-import { executeImportAgent } from '../../operations/agent/import';
 import type {
   AgentCoreProjectSpec,
   BuildType,
@@ -18,6 +17,7 @@ import {
   mapModelProviderToIdentityProviders,
   writeAgentToProject,
 } from '../../operations/agent/generate';
+import { executeImportAgent } from '../../operations/agent/import';
 import { credentialPrimitive } from '../../primitives/registry';
 import { CDKRenderer, createRenderer } from '../../templates';
 import type { CreateResult } from './types';

src/cli/commands/create/validate.ts

@@ -60,15 +60,19 @@ export function validateCreateOptions(options: CreateOptions, cwd?: string): Val
     if (!options.agentId) return { valid: false, error: '--agent-id is required for import' };
     if (!options.agentAliasId) return { valid: false, error: '--agent-alias-id is required for import' };
     if (!options.region) return { valid: false, error: '--region is required for import' };
-    if (!options.framework) return { valid: false, error: '--framework is required for import (Strands or LangChain_LangGraph)' };
+    if (!options.framework)
+      return { valid: false, error: '--framework is required for import (Strands or LangChain_LangGraph)' };
     const fw = matchEnumValue(SDKFrameworkSchema, options.framework) ?? options.framework;
     options.framework = fw;
     if (fw !== 'Strands' && fw !== 'LangChain_LangGraph') {
       return { valid: false, error: `Import only supports Strands or LangChain_LangGraph, got: ${options.framework}` };
     }
     options.memory ??= 'none';
     if (!MEMORY_OPTIONS.includes(options.memory as (typeof MEMORY_OPTIONS)[number])) {
-      return { valid: false, error: `Invalid memory option: ${options.memory}. Use none, shortTerm, or longAndShortTerm` };
+      return {
+        valid: false,
+        error: `Invalid memory option: ${options.memory}. Use none, shortTerm, or longAndShortTerm`,
+      };
     }
     return { valid: true };
   }

src/cli/operations/agent/import/__tests__/translator.test.ts

@@ -1,8 +1,8 @@
-import { describe, expect, it } from 'vitest';
 import type { BedrockAgentConfig } from '../../../../aws/bedrock-import-types';
-import { StrandsTranslator } from '../strands-translator';
 import { LangGraphTranslator } from '../langgraph-translator';
 import { generatePyprojectToml } from '../pyproject-generator';
+import { StrandsTranslator } from '../strands-translator';
+import { describe, expect, it } from 'vitest';
 
 function makeSimpleAgentConfig(overrides: Partial<BedrockAgentConfig> = {}): BedrockAgentConfig {
   return {

src/cli/operations/agent/import/base-translator.ts

@@ -5,7 +5,6 @@
  * Generates Python code strings for action groups, KBs, multi-agent collaboration,
  * code interpreter, user input, guardrails, prompts, and memory wiring.
  */
-
 import type {
   ActionGroupInfo,
   BedrockAgentConfig,
@@ -76,18 +75,15 @@ export abstract class BaseBedrockTranslator {
     protected readonly collaboratorContext?: { name: string; instruction: string; relayHistory: string }
   ) {
     this.agentInfo = config.agent;
-    this.actionGroups = (config.action_groups ?? []).filter(
-      ag => (ag.actionGroupState ?? 'ENABLED') === 'ENABLED'
-    );
+    this.actionGroups = (config.action_groups ?? []).filter(ag => (ag.actionGroupState ?? 'ENABLED') === 'ENABLED');
     this.customActionGroups = this.actionGroups.filter(ag => !ag.parentActionSignature);
     this.knowledgeBases = config.knowledge_bases ?? [];
     this.collaborators = config.collaborators ?? [];
 
     this.modelId = this.agentInfo.foundationModel ?? '';
     this.agentRegion = this.agentInfo.agentArn?.split(':')[3] ?? 'us-east-1';
     this.instruction = this.agentInfo.instruction ?? '';
-    this.promptConfigs =
-      this.agentInfo.promptOverrideConfiguration?.promptConfigurations ?? [];
+    this.promptConfigs = this.agentInfo.promptOverrideConfiguration?.promptConfigurations ?? [];
     this.enabledPrompts = [];
 
     // Memory
@@ -111,7 +107,8 @@ export abstract class BaseBedrockTranslator {
     this.guardrailConfig = {};
     const gc = this.agentInfo.guardrailConfiguration;
     if (gc) {
-      const gId = (gc as Record<string, string>).guardrailId ?? (gc as Record<string, string>).guardrailIdentifier ?? '';
+      const gId =
+        (gc as Record<string, string>).guardrailId ?? (gc as Record<string, string>).guardrailIdentifier ?? '';
       const gVersion = (gc as Record<string, string>).version ?? (gc as Record<string, string>).guardrailVersion ?? '';
       if (gId) {
         this.guardrailConfig = { guardrailIdentifier: gId, guardrailVersion: gVersion };
@@ -221,7 +218,16 @@ load_dotenv()
     const params = fn.parameters ?? {};
     const paramList = Object.entries(params)
       .map(([name, info]) => {
-        const pyType = info.type === 'string' ? 'str' : info.type === 'integer' ? 'int' : info.type === 'number' ? 'float' : info.type === 'boolean' ? 'bool' : 'str';
+        const pyType =
+          info.type === 'string'
+            ? 'str'
+            : info.type === 'integer'
+              ? 'int'
+              : info.type === 'number'
+                ? 'float'
+                : info.type === 'boolean'
+                  ? 'bool'
+                  : 'str';
         return `${name}: ${pyType}`;
       })
       .join(', ');
@@ -286,13 +292,9 @@ memory_id = os.environ.get("MEMORY_ID", "")
       platform === 'strands'
         ? 'tools_used.update(list(agent_result.metrics.tool_metrics.keys()))'
         : 'tools_used.update([msg.name for msg in agent_result if isinstance(msg, ToolMessage)])';
-    const responseContent =
-      platform === 'strands' ? 'str(agent_result)' : 'agent_result[-1].content';
+    const responseContent = platform === 'strands' ? 'str(agent_result)' : 'agent_result[-1].content';
 
-    const lines = [
-      'def endpoint(payload, context):',
-      '    try:',
-    ];
+    const lines = ['def endpoint(payload, context):', '    try:'];
     if (this.agentcoreMemoryEnabled) {
       lines.push('        global user_id');
       lines.push('        user_id = user_id or payload.get("userId", uuid.uuid4().hex[:8])');
@@ -303,30 +305,30 @@ memory_id = os.environ.get("MEMORY_ID", "")
       '        tools_used.clear()',
       '        agent_query = payload.get("prompt", "")',
       '        if not agent_query:',
-      '            return {\'error\': "No query provided, please provide a \'prompt\' field in the payload."}',
+      "            return {'error': \"No query provided, please provide a 'prompt' field in the payload.\"}",
       '',
       '        agent_result = invoke_agent(agent_query)',
       '',
       '        ' + toolsUsedUpdate,
       '        response_content = ' + responseContent,
       '',
       '        sources = []',
-      '        urls = re.findall(r\'https?://[^\\s<>"{}|\\\\^`\\[\\]]+\', response_content)',
+      "        urls = re.findall(r'https?://[^\\s<>\"{}|\\\\^`\\[\\]]+', response_content)",
       '        source_tags = re.findall(r"<source>(.*?)</source>", response_content)',
       '        sources.extend(urls)',
       '        sources.extend(source_tags)',
       '        sources = list(set(sources))',
       '',
-      '        formatted_messages = [(agent_query, "USER"), (response_content if response_content else "No Response.", "ASSISTANT")]',
+      '        formatted_messages = [(agent_query, "USER"), (response_content if response_content else "No Response.", "ASSISTANT")]'
     );
     if (memoryEventCode) {
       lines.push(memoryEventCode);
     }
     lines.push(
       '',
-      '        return {\'result\': {\'response\': response_content, \'sources\': sources, \'tools_used\': list(tools_used), \'sessionId\': session_id, \'messages\': formatted_messages}}',
+      "        return {'result': {'response': response_content, 'sources': sources, 'tools_used': list(tools_used), 'sessionId': session_id, 'messages': formatted_messages}}",
       '    except Exception as e:',
-      '        return {\'error\': str(e)}',
+      "        return {'error': str(e)}"
     );
     code += lines.join('\n');
     return code;

src/cli/operations/agent/import/index.ts

@@ -3,18 +3,17 @@
  * Provides executeImportAgent() as the shared handler called by both
  * the TUI hook (useAddAgent) and the non-interactive CLI (AgentPrimitive).
  */
-
 import { APP_DIR } from '../../../../lib';
 import type { SDKFramework } from '../../../../schema';
 import { getBedrockAgentConfig } from '../../../aws/bedrock-import';
+import { getErrorMessage } from '../../../errors';
+import type { AddResult } from '../../../primitives/types';
 import type { MemoryOption } from '../../../tui/screens/generate/types';
-import { writeAgentToProject } from '../generate/write-agent-to-project';
 import { setupPythonProject } from '../../python';
+import { writeAgentToProject } from '../generate/write-agent-to-project';
 import { LangGraphTranslator } from './langgraph-translator';
 import { generatePyprojectToml } from './pyproject-generator';
 import { StrandsTranslator } from './strands-translator';
-import type { AddResult } from '../../../primitives/types';
-import { getErrorMessage } from '../../../errors';
 import { mkdirSync, writeFileSync } from 'fs';
 import { dirname, join } from 'path';
 

src/cli/operations/agent/import/langgraph-translator.ts

@@ -2,10 +2,9 @@
  * LangChain/LangGraph-specific translator for Bedrock Agent import.
  * Port of the starter toolkit's bedrock_to_langchain.py.
  */
-
+import type { BedrockAgentConfig } from '../../../aws/bedrock-import-types';
 import type { TranslationResult, TranslatorOptions } from './base-translator';
 import { BaseBedrockTranslator } from './base-translator';
-import type { BedrockAgentConfig } from '../../../aws/bedrock-import-types';
 
 export class LangGraphTranslator extends BaseBedrockTranslator {
   constructor(
@@ -164,15 +163,11 @@ retriever_tool_${kbName} = retriever_${kbName}.as_tool(name="kb_${kbName}", desc
       const fileName = `langchain_collaborator_${collabName}`;
 
       // Recursively translate collaborator
-      const collabTranslator = new LangGraphTranslator(
-        collaborator as unknown as BedrockAgentConfig,
-        this.options,
-        {
-          name: collabName,
-          instruction: collaborator.collaborationInstruction ?? '',
-          relayHistory: collaborator.relayConversationHistory ?? 'DISABLED',
-        }
-      );
+      const collabTranslator = new LangGraphTranslator(collaborator as unknown as BedrockAgentConfig, this.options, {
+        name: collabName,
+        instruction: collaborator.collaborationInstruction ?? '',
+        relayHistory: collaborator.relayConversationHistory ?? 'DISABLED',
+      });
       const collabResult = collabTranslator.translate();
       collaboratorFiles.set(`${fileName}.py`, collabResult.mainPyContent);
       for (const [k, v] of collabResult.collaboratorFiles) {