Harrison/official pre release (#8106)

2025-09-23 19:39:58 +00:00 · 2023-07-21 18:44:32 -07:00
parent 95bcf68802
commit aa0e69bc98
65 changed files with 210 additions and 602 deletions
--- a/libs/experimental/langchain_experimental/cpal/templates/univariate/query.py
+++ b/libs/experimental/langchain_experimental/cpal/templates/univariate/query.py
@@ -0,0 +1,270 @@
+# flake8: noqa E501
+
+
+# fmt: off
+template = (
+    """
+Transform the narrative_input into an SQL expression. If you are
+unsure, then do not guess, instead add a llm_error_msg that explains why you are unsure.
+
+
+{format_instructions}
+
+
+narrative_input: how much money will boris have?
+
+
+# JSON:
+
+    {{
+        "narrative_input": "how much money will boris have?",
+        "llm_error_msg": "",
+        "expression": "SELECT name, value FROM df WHERE name = 'boris'"
+    }}
+
+
+
+narrative_input: How much money does ted have?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "How much money does ted have?",
+        "llm_error_msg": "",
+        "expression": "SELECT name, value FROM df WHERE name = 'ted'"
+    }}
+
+
+
+narrative_input: what is the sum of pet count for all the people?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what is the sum of pet count for all the people?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df"
+    }}
+
+
+
+
+narrative_input: what's the average of the pet counts for all the people?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the average of the pet counts for all the people?",
+        "llm_error_msg": "",
+        "expression": "SELECT AVG(value) FROM df"
+    }}
+
+
+
+
+narrative_input: what's the maximum of the pet counts for all the people?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the maximum of the pet counts for all the people?",
+        "llm_error_msg": "",
+        "expression": "SELECT MAX(value) FROM df"
+    }}
+
+
+
+
+narrative_input: what's the minimum of the pet counts for all the people?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the minimum of the pet counts for all the people?",
+        "llm_error_msg": "",
+        "expression": "SELECT MIN(value) FROM df"
+    }}
+
+
+
+
+narrative_input: what's the number of people with pet counts greater than 10?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the number of people with pet counts greater than 10?",
+        "llm_error_msg": "",
+        "expression": "SELECT COUNT(*) FROM df WHERE value > 10"
+    }}
+
+
+
+
+narrative_input: what's the pet count for boris?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the pet count for boris?",
+        "llm_error_msg": "",
+        "expression": "SELECT name, value FROM df WHERE name = 'boris'"
+    }}
+
+
+
+
+narrative_input: what's the pet count for cindy and marcia?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the pet count for cindy and marcia?",
+        "llm_error_msg": "",
+        "expression": "SELECT name, value FROM df WHERE name IN ('cindy', 'marcia')"
+    }}
+
+
+
+
+narrative_input: what's the total pet count for cindy and marcia?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the total pet count for cindy and marcia?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df WHERE name IN ('cindy', 'marcia')"
+    }}
+
+
+
+
+narrative_input: what's the total pet count for TED?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the total pet count for TED?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df WHERE name = 'TED'"
+    }}
+
+
+
+
+
+narrative_input: what's the total dollar count for TED and cindy?
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the total dollar count for TED and cindy?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df WHERE name IN ('TED', 'cindy')"
+    }}
+
+
+
+
+narrative_input: what's the total pet count for TED and cindy?
+
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the total pet count for TED and cindy?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df WHERE name IN ('TED', 'cindy')"
+    }}
+
+
+
+
+narrative_input: what's the best for TED and cindy?
+
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the best for TED and cindy?",
+        "llm_error_msg": "ambiguous narrative_input, not sure what 'best' means",
+        "expression": ""
+    }}
+
+
+
+
+narrative_input: what's the value?
+
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "what's the value?",
+        "llm_error_msg": "ambiguous narrative_input, not sure what entity is being asked about",
+        "expression": ""
+    }}
+
+
+
+
+
+
+narrative_input: how many total pets do the three have?
+
+
+
+
+
+# JSON:
+
+    {{
+        "narrative_input": "how many total pets do the three have?",
+        "llm_error_msg": "",
+        "expression": "SELECT SUM(value) FROM df"
+    }}
+
+
+
+
+
+
+narrative_input: {narrative_input}
+
+
+
+
+# JSON:
+""".strip()
+    + "\n\n\n"
+)
+# fmt: on