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