Merge pull request #72 from aftersnow/add-more-cap

feat: add more fields for capabilities

Author: chlins

docs/config.md

@@ -167,6 +167,14 @@ The following terms are used in this section:
 
     Whether the model can use external tools or APIs to perform tasks.
 
+  - **embedding** _boolean_, OPTIONAL
+
+    Whether the model can perform embedding tasks.
+
+  - **reward** _boolean_, OPTIONAL
+
+    Whether the model is a reward model.
+
 ## Example
 
 Here is an example model artifact configuration JSON document:
@@ -208,6 +216,8 @@ Here is an example model artifact configuration JSON document:
       "knowledgeCutoff": "2024-05-21T00:00:00Z",
       "reasoning": true,
       "toolUsage": false
+      "embedding": false,
+      "reward": false
     }
   },
   "modelfs": {

specs-go/v1/config.go

@@ -123,6 +123,12 @@ type ModelCapabilities struct {
 	// ToolUsage indicates whether the model can use external tools
 	// such as a calculator, a search engine, etc.
 	ToolUsage *bool `json:"toolUsage,omitempty"`
+
+	// Embedding indicates whether the model can perform embedding tasks
+	Embedding *bool `json:"embedding,omitempty"`
+
+	// Reward indicates whether the model is a reward model
+	Reward *bool `json:"reward,omitempty"`
 }
 
 // Model defines the basic information of a model.