From 276fe0f9fbf92a3b39d385a4ab27591ed562b63d Mon Sep 17 00:00:00 2001
From: Damien Ostler <Damien113@users.noreply.huggingface.co>
Date: Fri, 7 Feb 2025 07:56:20 +0000
Subject: [PATCH] Update Modelfile

---
 Modelfile | 27 +++++++++++----------------
 1 file changed, 11 insertions(+), 16 deletions(-)

diff --git a/Modelfile b/Modelfile
index d047018..0f49387 100644
--- a/Modelfile
+++ b/Modelfile
@@ -1,10 +1,5 @@
-
 FROM /content/Damien113/data_identification_manager_test/unsloth.F16.gguf
 TEMPLATE """
-
-You are an employee of Data443 Risk Mitigation, the creater of Data Identification Manager. Your job is to support that product and talk about it and nothing else. If the input you recieve is not about that, respond with a message saying you are not suppose to talk about that when you are working.
-
-
 {{ if .Messages }}
 {{- if or .System .Tools }}<|start_header_id|>system<|end_header_id|>
 {{- if .System }}
@@ -13,24 +8,24 @@ You are an employee of Data443 Risk Mitigation, the creater of Data Identificati
 {{- end }}
 {{- if .Tools }}
 
-You are a helpful assistant with tool calling capabilities. When you receive a tool call response, use the output to format an answer to the original use question.
+You are a helpful assistant with tool-calling capabilities. When you receive a tool call response, use the output to format an answer to the original user question.
 {{- end }}
 {{- end }}<|eot_id|>
 {{- range $i, $_ := .Messages }}
 {{- $last := eq (len (slice $.Messages $i)) 1 }}
 {{- if eq .Role "user" }}<|start_header_id|>user<|end_header_id|>
-{{- if and $.Tools $last }}
-
-Given the following functions, please respond with a JSON for a function call with its proper arguments that best answers the given prompt.
-
-Respond in the format {"name": function name, "parameters": dictionary of argument name and its value}. Do not use variables.
-
-
-{{ $.Tools }}
-{{- end }}
 
 {{ .Content }}<|eot_id|>{{ if $last }}<|start_header_id|>assistant<|end_header_id|>
 
+{{- if (or (contains .Content "Data Identification Manager") 
+            (contains .Content "DIM") 
+            (contains .Content "document tagging") 
+            (contains .Content "metadata search")) }}
+
+{{ else }}
+I'm only able to discuss topics related to Data Identification Manager. Let me know if you have any questions about it.
+{{ end }}
+
 {{ end }}
 {{- else if eq .Role "assistant" }}<|start_header_id|>assistant<|end_header_id|>
 {{- if .ToolCalls }}
@@ -60,4 +55,4 @@ PARAMETER stop "<|end_header_id|>"
 PARAMETER stop "<|eot_id|>"
 PARAMETER stop "<|eom_id|>"
 PARAMETER temperature 1.5
-PARAMETER min_p 0.1
\ No newline at end of file
+PARAMETER min_p 0.1