From 2731096264253993cd9cf661f5d08fbd0b63a1cf Mon Sep 17 00:00:00 2001
From: Aldrin <imagesai32@gmail.com>
Date: Wed, 12 Mar 2025 21:56:09 +0530
Subject: [PATCH] fix: react agent default prompt

Closes https://github.com/meta-llama/llama-stack-client-python/issues/194
---
 src/llama_stack_client/lib/agents/react/prompts.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/llama_stack_client/lib/agents/react/prompts.py b/src/llama_stack_client/lib/agents/react/prompts.py
index 4ce228f2..8c864afc 100644
--- a/src/llama_stack_client/lib/agents/react/prompts.py
+++ b/src/llama_stack_client/lib/agents/react/prompts.py
@@ -37,7 +37,7 @@
     "thought": "I need to transform the image that I received in the previous observation to make it green.",
     "action": {
         "tool_name": "image_transformer",
-        "tool_params": {"image": "image_1.jpg"}
+        "tool_params": [{"image": "image_1.jpg"}]
     },
     "answer": null
 }
@@ -61,7 +61,7 @@
     "thought": "I will proceed step by step and use the following tools: `document_qa` to find the oldest person in the document, then `image_generator` to generate an image according to the answer.",
     "action": {
         "tool_name": "document_qa",
-        "tool_params": {"document": "document.pdf", "question": "Who is the oldest person mentioned?"}
+        "tool_params": [{"document": "document.pdf", "question": "Who is the oldest person mentioned?"}]
     },
     "answer": null
 }
@@ -73,7 +73,7 @@
     "thought": "I will now generate an image showcasing the oldest person.",
     "action": {
         "tool_name": "image_generator",
-        "tool_params": {"prompt": "A portrait of John Doe, a 55-year-old man living in Canada."}
+        "tool_params": [{"prompt": "A portrait of John Doe, a 55-year-old man living in Canada."}]
     },
     "answer": null
 }
@@ -93,7 +93,7 @@
     "thought": "I will use python code evaluator to compute the result of the operation and then return the final answer using the `final_answer` tool",
     "action": {
         "tool_name": "python_interpreter",
-        "tool_params": {"code": "5 + 3 + 1294.678"}
+        "tool_params": [{"code": "5 + 3 + 1294.678"}]
     },
     "answer": null
 }
@@ -113,7 +113,7 @@
     "thought": "I need to get the populations for both cities and compare them: I will use the tool `search` to get the population of both cities.",
     "action": {
         "tool_name": "search",
-        "tool_params": {"query": "Population Guangzhou"}
+        "tool_params": [{"query": "Population Guangzhou"}]
     },
     "answer": null
 }
@@ -124,7 +124,7 @@
     "thought": "Now let's get the population of Shanghai using the tool 'search'.",
     "action": {
         "tool_name": "search",
-        "tool_params": {"query": "Population Shanghai"}
+        "tool_params": [{"query": "Population Shanghai"}]
     },
     "answer": null
 }