Prompt engineering to encourage LLM to always label its x and y axes in plots... throwback to grading lab reports in grad school...

Author: kaleko

lib/idp_common_pkg/idp_common/agents/analytics/agent.py

@@ -83,6 +83,7 @@ def create_analytics_agent(
     - Only write python code to generate plots or tables. Do not use python for any other purpose.
     - The python code should read the query results from "query_results.csv" file provided, for example with a line like `df = pd.read_csv("query_results.csv")`
     - Make sure the python code will output json representing either "table" or "plotData" responseType as described in the above description of the result format.
+    - Any time you generate a plot, make sure to label the x and y axes clearly.
     - Use built in python libraries, optionally with pandas or matplotlib.
     - Always use the execute_python tool to execute your python code, and be sure to include the reset_state=True flag each time you call this tool.
     

lib/idp_common_pkg/idp_common/agents/analytics/config.py

@@ -321,7 +321,17 @@ def load_result_format_description() -> str:
   "options": {
     "scales": {
       "y": {
-        "beginAtZero": true
+        "beginAtZero": true,
+        "title": {
+          "display": true,
+          "text": "Number of Documents"
+        }
+      },
+      "x": {
+        "title": {
+          "display": true,
+          "text": "Document Types"
+        }
       }
     },
     "responsive": true,