checkpoint(Saved system prompt presets)

src/models.py

@@ -315,3 +315,33 @@ class FileItem:
    auto_aggregate=data.get("auto_aggregate", True),
    force_full=data.get("force_full", False),
   )
+
+@dataclass
+class Preset:
+ name: str
+ system_prompt: str
+ temperature: Optional[float] = None
+ top_p: Optional[float] = None
+ max_output_tokens: Optional[int] = None
+
+ def to_dict(self) -> Dict[str, Any]:
+  res = {
+   "system_prompt": self.system_prompt,
+  }
+  if self.temperature is not None:
+   res["temperature"] = self.temperature
+  if self.top_p is not None:
+   res["top_p"] = self.top_p
+  if self.max_output_tokens is not None:
+   res["max_output_tokens"] = self.max_output_tokens
+  return res
+
+ @classmethod
+ def from_dict(cls, name: str, data: Dict[str, Any]) -> "Preset":
+  return cls(
+   name=name,
+   system_prompt=data.get("system_prompt", ""),
+   temperature=data.get("temperature"),
+   top_p=data.get("top_p"),
+   max_output_tokens=data.get("max_output_tokens"),
+  )