Fix problem with deserializer (#80)

* Fix never-ending deserialization agent * Fix never-ending deserialization agent
xebia-functional · May 21, 2023 · a176736 · a176736
1 parent 6405f84
commit a176736
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/DeserializerLLMAgent.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/auto/DeserializerLLMAgent.kt
@@ -143,8 +143,12 @@ suspend fun <A> AIScope.tryDeserialize(
   while (currentAttempts < maxDeserializationAttempts) {
     currentAttempts++
     val result = ensureNotNull(agent().firstOrNull()) { AIError.NoResponse }
-    catch({ json.decodeFromString(serializationConfig.deserializationStrategy, result) }) {
-      e: IllegalArgumentException ->
+    catch({
+      return@tryDeserialize json.decodeFromString(
+        serializationConfig.deserializationStrategy,
+        result
+      )
+    }) { e: IllegalArgumentException ->
       if (currentAttempts == maxDeserializationAttempts)
         raise(AIError.JsonParsing(result, maxDeserializationAttempts, e))
       // else continue with the next attempt