feat: ✨ add prediction support and enhance processing

microsoft · Dec 27, 2024 · b5d6267 · b5d6267
1 parent fdae125
commit b5d6267
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/packages/core/src/llms.json b/packages/core/src/llms.json
@@ -18,13 +18,15 @@
             "id": "azure",
             "detail": "Azure OpenAI deployment",
             "listModels": false,
-            "bearerToken": false
+            "bearerToken": false,
+            "prediction": false
         },
         {
             "id": "azure_serverless",
             "detail": "Azure AI OpenAI (serverless deployments)",
             "listModels": false,
             "bearerToken": false,
+            "prediction": false,
             "aliases": {
                 "large": "gpt-4o",
                 "small": "gpt-4o-mini",

diff --git a/packages/sample/genaisrc/dup.genai.mjs b/packages/sample/genaisrc/dup.genai.mjs
@@ -7,4 +7,9 @@ script({
 def("FILE", "hello world")
 def("FILE", "hello world")
 
-$`If <FILE> is defined twice in the prompt, respond with FAIL; otherwise respond SUCCESS`
+defOutputProcessor((output) => {
+    const msg = output.messages.find(m => m.role === "user")
+    console.log({ msg })
+    const fileCount = (msg.content.match(/<FILE>/g) || []).length;
+    if (fileCount > 2) throw new Error("Too many files")
+})
diff --git a/packages/sample/genaisrc/tools.genai.mjs b/packages/sample/genaisrc/tools.genai.mjs
@@ -17,7 +17,7 @@ defTool(
         required: ["location"],
     },
     ({ location }) => {
-        return location === "Paris" ? "sunny" : "unknown"
+        return /paris/i.test(location) ? "sunny" : "unknown"
     }
 )