From f0311cd8b97e75d0dd557c02d4c8fe17fdd58ce2 Mon Sep 17 00:00:00 2001 From: Hao Wu Date: Fri, 26 Apr 2024 21:50:09 +0800 Subject: [PATCH] remove redundent ollama files --- api/ollama.go | 69 --------------------------------------------------- 1 file changed, 69 deletions(-) delete mode 100644 api/ollama.go diff --git a/api/ollama.go b/api/ollama.go deleted file mode 100644 index 1dbf067a..00000000 --- a/api/ollama.go +++ /dev/null @@ -1,69 +0,0 @@ -package main - -import ( - "fmt" - "strings" -) - -func formatMinstralPrompt(chat_compeletion_messages []Message) string { - var sb strings.Builder - - for _, message := range chat_compeletion_messages { - - if message.Role != "assistant" { - sb.WriteString(fmt.Sprintf("[INST] %s[/INST]\n", message.Content)) - } else { - sb.WriteString(fmt.Sprintf("%s \n", message.Content)) - } - } - prompt := sb.String() - print(prompt) - return prompt -} - -// ### Prompt Format -// ### System: -// {system} -// ### User: -// {usr} -// ### Assistant: -func formatNeuralChatPrompt(chat_compeletion_messages []Message) string { - var sb strings.Builder - - for _, message := range chat_compeletion_messages { - if message.Role == "system" { - sb.WriteString(fmt.Sprintf("### System:\n%s\n", message.Content)) - } else if message.Role == "user" { - sb.WriteString(fmt.Sprintf("### User:\n%s\n", message.Content)) - } else { - sb.WriteString(fmt.Sprintf("### Assistant:\n%s\n", message.Content)) - } - } - sb.WriteString("### Assitant:\n") - prompt := sb.String() - print(prompt) - return prompt -} - -// <|im_start|>system -// {system}<|im_end|> -// <|im_start|>user -// {user}<|im_end|> -// <|im_start|>assistant -// {asistant}<|im_end|> -func formatOpenHermesNeuralChatPrompt(chat_compeletion_messages []Message) string { - var sb strings.Builder - - for _, message := range chat_compeletion_messages { - if message.Role == "system" { - sb.WriteString(fmt.Sprintf("<|im_start|>system\n%s<|im_end|>\n", message.Content)) - } else if message.Role == "user" { - sb.WriteString(fmt.Sprintf("<|im_start|>user\n%s<|im_end|>\n", message.Content)) - } else { - sb.WriteString(fmt.Sprintf("<|im_start|>assistant\n%s<|im_end|>\n", message.Content)) - } - } - prompt := sb.String() - print(prompt) - return prompt -}