with schemas this time

Author: mchenco

bin/fetch-ai-models.js

@@ -3,7 +3,7 @@ import fs from "fs";
 fetch("https://ai.cloudflare.com/api/models")
 	.then((res) => res.json())
 	.then((data) => {
-		data.models.forEach((model) => {
+		data.models.map((model) => {
 			const fileName = model.name.split("/")[2];
 			fs.writeFileSync(
 				`./src/content/workers-ai-models/${fileName}.json`,

src/content/workers-ai-models/bge-m3.json

@@ -10,5 +10,74 @@
     },
     "tags": [],
     "properties": [],
-    "schema": {}
+    "schema": {
+        "input": {
+            "type": "object",
+            "properties": {
+                "query": {
+                    "type": "string",
+                    "minLength": 1,
+                    "description": "A query you wish to perform against the provided contexts. If no query is provided the model with respond with embeddings for contexts"
+                },
+                "contexts": {
+                    "type": "array",
+                    "items": {
+                        "type": "object",
+                        "properties": {
+                            "text": {
+                                "type": "string",
+                                "minLength": 1,
+                                "description": "One of the provided context content"
+                            }
+                        }
+                    },
+                    "description": "List of provided contexts. Note that the index in this array is important, as the response will refer to it."
+                }
+            },
+            "required": [
+                "contexts"
+            ]
+        },
+        "output": {
+            "type": "object",
+            "contentType": "application/json",
+            "oneOf": [
+                {
+                    "title": "Query",
+                    "properties": {
+                        "response": {
+                            "type": "array",
+                            "items": {
+                                "type": "object",
+                                "properties": {
+                                    "id": {
+                                        "type": "integer",
+                                        "description": "Index of the context in the request"
+                                    },
+                                    "score": {
+                                        "type": "number",
+                                        "description": "Score of the context under the index."
+                                    }
+                                }
+                            }
+                        }
+                    }
+                },
+                {
+                    "title": "Embedding",
+                    "properties": {
+                        "response": {
+                            "type": "array",
+                            "items": {
+                                "type": "array",
+                                "items": {
+                                    "type": "number"
+                                }
+                            }
+                        }
+                    }
+                }
+            ]
+        }
+    }
 }

src/content/workers-ai-models/bge-reranker-base.json

@@ -10,5 +10,62 @@
     },
     "tags": [],
     "properties": [],
-    "schema": {}
+    "schema": {
+      "input": {
+        "type": "object",
+        "properties": {
+          "query": {
+            "type": "string",
+            "minLength": 1,
+            "description": "A query you wish to perform against the provided contexts."
+          },
+          "top_k": {
+            "type": "integer",
+            "default": null,
+            "minimum": 1,
+            "description": "Number of returned results starting with the best score."
+          },
+          "contexts": {
+            "type": "array",
+            "items": {
+              "type": "object",
+              "properties": {
+                "text": {
+                  "type": "string",
+                  "minLength": 1,
+                  "description": "One of the provided context content"
+                }
+              }
+            },
+            "description": "List of provided contexts. Note that the index in this array is important, as the response will refer to it."
+          }
+        },
+        "required": [
+          "query",
+          "contexts"
+        ]
+      },
+      "output": {
+        "type": "object",
+        "contentType": "application/json",
+        "properties": {
+          "response": {
+            "type": "array",
+            "items": {
+              "type": "object",
+              "properties": {
+                "id": {
+                  "type": "integer",
+                  "description": "Index of the context in the request"
+                },
+                "score": {
+                  "type": "number",
+                  "description": "Score of the context under the index."
+                }
+              }
+            }
+          }
+        }
+      }
+    }
 }