template extra args

2023-09-03 18:29:54 -04:00
parent adaa13088b
commit cdc9aa14ed
2 changed files with 8 additions and 5 deletions
--- a/api/types.go
+++ b/api/types.go
@@ -32,11 +32,12 @@ func (e StatusError) Error() string {
 }

 type GenerateRequest struct {
-	Model    string `json:"model"`
-	Prompt   string `json:"prompt"`
-	System   string `json:"system"`
-	Template string `json:"template"`
-	Context  []int  `json:"context,omitempty"`
+	Model    string         `json:"model"`
+	Prompt   string         `json:"prompt"`
+	System   string         `json:"system"`
+	Template string         `json:"template"`
+	Context  []int          `json:"context,omitempty"`
+	Args     map[string]any `json:"args,omitempty"`

 	Options map[string]interface{} `json:"options"`
 }