fumiama/deepinfra
1
0
Fork 0
mirror of https://github.com/fumiama/deepinfra.git synced 2026-06-30 01:30:29 +08:00
Code Activity

fix: protocol restriction

Browse Source
This commit is contained in:
源文雨
2025-02-22 01:16:25 +09:00
parent ed1290950d
commit 5a9f7c9000
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
model/ollama.go
View File

@@ -61,7 +61,7 @@ func (ollm *OLLaMA) OutputRaw() string {
return ollm.Message.Content 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 = make([]Message, 1, 8)
ollm.Messages[0] = Message{ ollm.Messages[0] = Message{
Role: "system", Role: "system",
@@ -70,7 +70,7 @@ func (ollm *OLLaMA) System(prompt string) *OLLaMA {
return ollm return ollm
} }
func (ollm *OLLaMA) User(prompt string) *OLLaMA { func (ollm *OLLaMA) User(prompt string) Protocol {
ollm.Messages = append(ollm.Messages, Message{ ollm.Messages = append(ollm.Messages, Message{
Role: "user", Role: "user",
Content: prompt, Content: prompt,
@@ -78,7 +78,7 @@ func (ollm *OLLaMA) User(prompt string) *OLLaMA {
return ollm return ollm
} }
func (ollm *OLLaMA) Assistant(prompt string) *OLLaMA { func (ollm *OLLaMA) Assistant(prompt string) Protocol {
ollm.Messages = append(ollm.Messages, Message{ ollm.Messages = append(ollm.Messages, Message{
Role: "assistant", Role: "assistant",
Content: prompt, Content: prompt,

6
model/openai.go
View File

@@ -76,7 +76,7 @@ func (opai *OpenAI) OutputRaw() string {
return opai.Choices[len(opai.Choices)-1].Message.Content 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 = make([]Message, 1, 8)
opai.Messages[0] = Message{ opai.Messages[0] = Message{
Role: "system", Role: "system",
@@ -85,7 +85,7 @@ func (opai *OpenAI) System(prompt string) *OpenAI {
return opai return opai
} }
func (opai *OpenAI) User(prompt string) *OpenAI { func (opai *OpenAI) User(prompt string) Protocol {
opai.Messages = append(opai.Messages, Message{ opai.Messages = append(opai.Messages, Message{
Role: "user", Role: "user",
Content: prompt, Content: prompt,
@@ -93,7 +93,7 @@ func (opai *OpenAI) User(prompt string) *OpenAI {
return opai return opai
} }
func (opai *OpenAI) Assistant(prompt string) *OpenAI { func (opai *OpenAI) Assistant(prompt string) Protocol {
opai.Messages = append(opai.Messages, Message{ opai.Messages = append(opai.Messages, Message{
Role: "assistant", Role: "assistant",
Content: prompt, Content: prompt,