fix: align extension schemas in TypeScript with Python schemas (#1375)

Author: PetrBulanek

apps/beeai-sdk-ts/src/client/a2a/extensions/services/embedding.ts

@@ -10,8 +10,8 @@ import type { A2AServiceExtension } from '../types';
 const URI = 'https://a2a-extensions.beeai.dev/services/embedding/v1';
 
 const embeddingDemandSchema = z.object({
-  description: z.string().nullable(),
-  suggested: z.array(z.string()).nullable(),
+  description: z.string().nullish(),
+  suggested: z.array(z.string()).nullish(),
 });
 
 const embeddingDemandsSchema = z.object({
@@ -23,7 +23,7 @@ const embeddingFulfillmentsSchema = z.object({
   embedding_fulfillments: z.record(
     z.string(),
     z.object({
-      identifier: z.string().optional(),
+      identifier: z.string().nullish(),
       api_base: z.string(),
       api_key: z.string(),
       api_model: z.string(),

apps/beeai-sdk-ts/src/client/a2a/extensions/services/llm.ts

@@ -10,8 +10,8 @@ import type { A2AServiceExtension } from '../types';
 const URI = 'https://a2a-extensions.beeai.dev/services/llm/v1';
 
 const llmDemandSchema = z.object({
-  description: z.string().nullable(),
-  suggested: z.array(z.string()).nullable(),
+  description: z.string().nullish(),
+  suggested: z.array(z.string()).nullish(),
 });
 
 const llmDemandsSchema = z.object({
@@ -23,7 +23,7 @@ const llmFulfillmentSchema = z.object({
   llm_fulfillments: z.record(
     z.string(),
     z.object({
-      identifier: z.string().optional(),
+      identifier: z.string().nullish(),
       api_base: z.string(),
       api_key: z.string(),
       api_model: z.string(),

apps/beeai-sdk-ts/src/client/a2a/extensions/services/mcp.ts

@@ -13,9 +13,9 @@ const mcpTransportTypesEnum = z.enum(['streamable_http', 'stdio']);
 type MCPTransportType = z.infer<typeof mcpTransportTypesEnum>;
 
 const mcpDemandSchema = z.object({
-  description: z.string().nullable(),
-  suggested: z.array(z.string()).nullable(),
-  allowed_transports: z.array(mcpTransportTypesEnum).nullable(),
+  description: z.string().nullish(),
+  suggested: z.array(z.string()).nullish(),
+  allowed_transports: z.array(mcpTransportTypesEnum).nullish(),
 });
 
 const mcpDemandsSchema = z.object({

apps/beeai-sdk-ts/src/client/a2a/extensions/services/secrets.ts

@@ -11,7 +11,7 @@ const URI = 'https://a2a-extensions.beeai.dev/auth/secrets/v1';
 
 const secretDemandSchema = z.object({
   name: z.string(),
-  description: z.string().nullable(),
+  description: z.string().nullish(),
 });
 export type SecretDemand = z.infer<typeof secretDemandSchema>;
 

apps/beeai-sdk-ts/src/client/a2a/extensions/ui/agent-detail.ts

@@ -12,31 +12,29 @@ const URI = 'https://a2a-extensions.beeai.dev/ui/agent-detail/v1';
 
 const contributorSchema = z.object({
   name: z.string(),
-  email: z.string().nullable(),
-  url: z.string().nullable(),
+  email: z.string().nullish(),
+  url: z.string().nullish(),
 });
 
 const toolSchema = z.object({
   name: z.string(),
   description: z.string(),
 });
 
-const schema = z
-  .object({
-    interaction_mode: z.union([interactionModeSchema, z.string()]).nullable(),
-    user_greeting: z.string().nullable(),
-    input_placeholder: z.string().nullable(),
-    tools: z.array(toolSchema).nullable(),
-    framework: z.string().nullable(),
-    license: z.string().nullable(),
-    programming_language: z.string().nullable(),
-    homepage_url: z.string().nullable(),
-    source_code_url: z.string().nullable(),
-    container_image_url: z.string().nullable(),
-    author: contributorSchema.nullable(),
-    contributors: z.array(contributorSchema).nullable(),
-  })
-  .partial();
+const schema = z.object({
+  interaction_mode: z.union([interactionModeSchema, z.string()]).nullish(),
+  user_greeting: z.string().nullish(),
+  input_placeholder: z.string().nullish(),
+  tools: z.array(toolSchema).nullish(),
+  framework: z.string().nullish(),
+  license: z.string().nullish(),
+  programming_language: z.string().nullish(),
+  homepage_url: z.string().nullish(),
+  source_code_url: z.string().nullish(),
+  container_image_url: z.string().nullish(),
+  author: contributorSchema.nullish(),
+  contributors: z.array(contributorSchema).nullish(),
+});
 
 export type AgentDetailTool = z.infer<typeof toolSchema>;
 export type AgentDetailContributor = z.infer<typeof contributorSchema>;

apps/beeai-sdk-ts/src/client/a2a/extensions/ui/citation.ts

@@ -9,15 +9,13 @@ import type { A2AUiExtension } from '../types';
 
 const URI = 'https://a2a-extensions.beeai.dev/ui/citation/v1';
 
-const citationSchema = z
-  .object({
-    url: z.string(),
-    start_index: z.number(),
-    end_index: z.number(),
-    title: z.string(),
-    description: z.string(),
-  })
-  .partial();
+const citationSchema = z.object({
+  url: z.string().nullish(),
+  start_index: z.number().nullish(),
+  end_index: z.number().nullish(),
+  title: z.string().nullish(),
+  description: z.string().nullish(),
+});
 
 const schema = z.object({
   citations: z.array(citationSchema),

apps/beeai-sdk-ts/src/client/a2a/extensions/ui/form.ts

@@ -50,8 +50,8 @@ const fileFieldValue = z.object({
     .array(
       z.object({
         uri: z.string(),
-        name: z.string().optional(),
-        mime_type: z.string().optional(),
+        name: z.string().nullish(),
+        mime_type: z.string().nullish(),
       }),
     )
     .nullish(),

apps/beeai-sdk-ts/src/client/a2a/extensions/ui/trajectory.ts

@@ -9,12 +9,10 @@ import type { A2AUiExtension } from '../types';
 
 const URI = 'https://a2a-extensions.beeai.dev/ui/trajectory/v1';
 
-const schema = z
-  .object({
-    title: z.string(),
-    content: z.string(),
-  })
-  .partial();
+const schema = z.object({
+  title: z.string().nullish(),
+  content: z.string().nullish(),
+});
 
 export type TrajectoryMetadata = z.infer<typeof schema>;
 

apps/beeai-ui/src/api/a2a/utils.ts

@@ -146,8 +146,8 @@ export function createTrajectoryPart(metadata: TrajectoryMetadata): UITrajectory
   const trajectoryPart: UITrajectoryPart = {
     kind: UIMessagePartKind.Trajectory,
     id: uuid(),
-    title,
-    content: truncate(content, { length: MAX_CONTENT_CHARS_LENGTH }),
+    title: title ?? undefined,
+    content: truncate(content ?? undefined, { length: MAX_CONTENT_CHARS_LENGTH }),
   };
 
   return trajectoryPart;