fixed error in enum description (#733)

Co-authored-by: José Carlos Montañez <jc@MacBook-Pro.local>
xebia-functional · May 6, 2024 · 1aef223 · 1aef223
1 parent 5fb134f
commit 1aef223
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/functions/JsonSchema.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/models/functions/JsonSchema.kt
@@ -327,15 +327,21 @@ private fun JsonObjectBuilder.applyJsonSchemaDefaults(
     if (descriptor.kind == SerialKind.ENUM) {
       this["enum"] = descriptor.elementNames
       descriptor.elementNames
-        .mapIndexed { index, name ->
-          "$name (${descriptor.getElementAnnotations(index).lastOfInstance<Description>()?.value})"
+        .mapIndexedNotNull { index, name ->
+          val enumDescription =
+            descriptor.getElementAnnotations(index).lastOfInstance<Description>()?.value
+          if (enumDescription != null) {
+            "$name ($enumDescription)"
+          } else {
+            null
+          }
         }
         .joinToString("\n - ")
     } else null
 
   if (annotations.isNotEmpty()) {
     val description = annotations.filterIsInstance<Description>().firstOrNull()?.value
-    if (additionalEnumDescription != null) {
+    if (!additionalEnumDescription.isNullOrEmpty()) {
       this["description"] = "$description\n - $additionalEnumDescription"
     } else {
       this["description"] = description