diff --git a/model/api.go b/model/api.go index 44834a2..8d02d07 100644 --- a/model/api.go +++ b/model/api.go @@ -22,7 +22,7 @@ type Requester interface { } type Cloner interface { - Clone() Cloner + Clone() Protocol } type MessageBuilder[T any] interface { diff --git a/model/genai.go b/model/genai.go index 4984fa3..9c51c19 100644 --- a/model/genai.go +++ b/model/genai.go @@ -189,7 +189,7 @@ func (opai *GenAI) Assistant(prompt ...Content) Protocol { return opai } -func (opai *GenAI) Clone() Cloner { +func (opai *GenAI) Clone() Protocol { x := new(GenAI) *x = *opai x.SystemInstruction = nil diff --git a/model/ollama.go b/model/ollama.go index 0a4ba25..f28c07d 100644 --- a/model/ollama.go +++ b/model/ollama.go @@ -102,7 +102,7 @@ func (ollm *OLLaMA) Assistant(prompt ...Content) Protocol { return ollm } -func (ollm *OLLaMA) Clone() Cloner { +func (ollm *OLLaMA) Clone() Protocol { x := new(OLLaMA) *x = *ollm x.Messages = nil diff --git a/model/openai.go b/model/openai.go index d044902..4298f38 100644 --- a/model/openai.go +++ b/model/openai.go @@ -156,7 +156,7 @@ func (opai *OpenAI) Assistant(prompt ...Content) Protocol { return opai.normal("assistant", prompt...) } -func (opai *OpenAI) Clone() Cloner { +func (opai *OpenAI) Clone() Protocol { x := new(OpenAI) *x = *opai x.Choices = nil