Merge pull request #5196 from ollama/mxyng/messages-2

include modelfile messages
This commit is contained in:
Michael Yang
2024-07-31 10:18:17 -07:00
committed by GitHub
4 changed files with 36 additions and 40 deletions

View File

@@ -70,7 +70,7 @@ type Model struct {
License []string
Digest string
Options map[string]interface{}
Messages []Message
Messages []api.Message
Template *template.Template
}
@@ -191,11 +191,6 @@ func (m *Model) String() string {
return modelfile.String()
}
type Message struct {
Role string `json:"role"`
Content string `json:"content"`
}
type ConfigV2 struct {
ModelFormat string `json:"model_format"`
ModelFamily string `json:"model_family"`