pkg/aflow: make it possible for LLMAgent to return only structured outputs

In some cases there may be not final text reply, only some structured outputs
(e.g. some bool). Don't require final reply, if structured outputs are specified.

1 files changed, 65 insertions, 0 deletions

diff --git a/pkg/aflow/testdata/TestOnlyStructuredOutputs.llm.json b/pkg/aflow/testdata/TestOnlyStructuredOutputs.llm.json
new file mode 100644
index 000000000..c9776f4ae
--- /dev/null
+++ b/pkg/aflow/testdata/TestOnlyStructuredOutputs.llm.json
@@ -0,0 +1,65 @@
+[
+	{
+		"Model": "model",
+		"Config": {
+			"systemInstruction": {
+				"parts": [
+					{
+						"text": "Instructions\n\nUse set-results tool to provide results of the analysis.\nIt must be called exactly once before the final reply.\nIgnore results of this tool.\n"
+					}
+				],
+				"role": "user"
+			},
+			"temperature": 0.3,
+			"tools": [
+				{
+					"functionDeclarations": [
+						{
+							"description": "Use this tool to provide results of the analysis.",
+							"name": "set-results",
+							"parametersJsonSchema": {
+								"additionalProperties": false,
+								"properties": {
+									"Result": {
+										"description": "Result",
+										"type": "integer"
+									}
+								},
+								"required": [
+									"Result"
+								],
+								"type": "object"
+							},
+							"responseJsonSchema": {
+								"additionalProperties": false,
+								"properties": {
+									"Result": {
+										"description": "Result",
+										"type": "integer"
+									}
+								},
+								"required": [
+									"Result"
+								],
+								"type": "object"
+							}
+						}
+					]
+				}
+			],
+			"responseModalities": [
+				"TEXT"
+			]
+		},
+		"Request": [
+			{
+				"parts": [
+					{
+						"text": "Initial Prompt"
+					}
+				],
+				"role": "user"
+			}
+		]
+	}
+]
\ No newline at end of file