From 5a9f7c9000d7a05601234830c9243363deedadc6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Sat, 22 Feb 2025 01:16:25 +0900
Subject: [PATCH] fix: protocol restriction

---
 model/ollama.go | 6 +++---
 model/openai.go | 6 +++---
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/model/ollama.go b/model/ollama.go
index 25fca44..f8b8170 100644
--- a/model/ollama.go
+++ b/model/ollama.go
@@ -61,7 +61,7 @@ func (ollm *OLLaMA) OutputRaw() string {
 	return ollm.Message.Content
 }
 
-func (ollm *OLLaMA) System(prompt string) *OLLaMA {
+func (ollm *OLLaMA) System(prompt string) Protocol {
 	ollm.Messages = make([]Message, 1, 8)
 	ollm.Messages[0] = Message{
 		Role:    "system",
@@ -70,7 +70,7 @@ func (ollm *OLLaMA) System(prompt string) *OLLaMA {
 	return ollm
 }
 
-func (ollm *OLLaMA) User(prompt string) *OLLaMA {
+func (ollm *OLLaMA) User(prompt string) Protocol {
 	ollm.Messages = append(ollm.Messages, Message{
 		Role:    "user",
 		Content: prompt,
@@ -78,7 +78,7 @@ func (ollm *OLLaMA) User(prompt string) *OLLaMA {
 	return ollm
 }
 
-func (ollm *OLLaMA) Assistant(prompt string) *OLLaMA {
+func (ollm *OLLaMA) Assistant(prompt string) Protocol {
 	ollm.Messages = append(ollm.Messages, Message{
 		Role:    "assistant",
 		Content: prompt,
diff --git a/model/openai.go b/model/openai.go
index 33e6dbe..5458f89 100644
--- a/model/openai.go
+++ b/model/openai.go
@@ -76,7 +76,7 @@ func (opai *OpenAI) OutputRaw() string {
 	return opai.Choices[len(opai.Choices)-1].Message.Content
 }
 
-func (opai *OpenAI) System(prompt string) *OpenAI {
+func (opai *OpenAI) System(prompt string) Protocol {
 	opai.Messages = make([]Message, 1, 8)
 	opai.Messages[0] = Message{
 		Role:    "system",
@@ -85,7 +85,7 @@ func (opai *OpenAI) System(prompt string) *OpenAI {
 	return opai
 }
 
-func (opai *OpenAI) User(prompt string) *OpenAI {
+func (opai *OpenAI) User(prompt string) Protocol {
 	opai.Messages = append(opai.Messages, Message{
 		Role:    "user",
 		Content: prompt,
@@ -93,7 +93,7 @@ func (opai *OpenAI) User(prompt string) *OpenAI {
 	return opai
 }
 
-func (opai *OpenAI) Assistant(prompt string) *OpenAI {
+func (opai *OpenAI) Assistant(prompt string) Protocol {
 	opai.Messages = append(opai.Messages, Message{
 		Role:    "assistant",
 		Content: prompt,