support yaml in assistant instructions (#677)

xebia-functional · Mar 6, 2024 · a7a0b9e · a7a0b9e
1 parent f808e47
commit a7a0b9e
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/assistants/Assistant.kt b/core/src/commonMain/kotlin/com/xebia/functional/xef/llm/assistants/Assistant.kt
@@ -17,6 +17,7 @@ import kotlinx.serialization.json.JsonElement
 import kotlinx.serialization.json.JsonObject
 import kotlinx.serialization.json.JsonPrimitive
 import net.mamoe.yamlkt.Yaml
+import net.mamoe.yamlkt.YamlMap
 import net.mamoe.yamlkt.literalContentOrNull
 import net.mamoe.yamlkt.toYamlElement
 
@@ -110,7 +111,11 @@ class Assistant(
           model = parsed["model"]?.literalContentOrNull ?: error("model is required"),
           name = parsed["name"]?.literalContentOrNull,
           description = parsed["description"]?.literalContentOrNull,
-          instructions = parsed["instructions"]?.literalContentOrNull,
+          instructions =
+            parsed["instructions"]?.literalContentOrNull
+              ?: (parsed["instructions"] as? YamlMap)?.let {
+                Yaml.encodeToString(YamlMap.serializer(), it)
+              },
           tools =
             parsed["tools"]?.let { list ->
               (list as List<*>).map { element ->