update arch-fc parameters

This commit is contained in:
Adil Hafeez 2024-09-27 13:34:10 -07:00
parent 75cf5e5304
commit 8a4e11077c
3 changed files with 7 additions and 6 deletions

View file

@ -4,8 +4,9 @@ FROM Arch-Function-Calling-1.5B-Q4_K_M.gguf
PARAMETER num_predict 1024 PARAMETER num_predict 1024
PARAMETER temperature 0.001 PARAMETER temperature 0.001
PARAMETER top_p 1.0 PARAMETER top_p 1.0
PARAMETER top_k 16000 PARAMETER top_k 50
PARAMETER repeat_penalty 1.0 PARAMETER repeat_penalty 1.0
PARAMETER stop "<|im_start|>"
PARAMETER stop "<|im_end|>" PARAMETER stop "<|im_end|>"
# Set the random number seed to use for generation # Set the random number seed to use for generation
@ -15,7 +16,8 @@ PARAMETER seed 42
TEMPLATE """ TEMPLATE """
{{- if .System }}<|im_start|>system {{- if .System }}<|im_start|>system
{{ .System }}<|im_end|> {{ .System }}<|im_end|>
{{ end }}{{ if .Prompt }}<|im_start|>user {{ end }}{{ if .Prompt }}
<|im_start|>user
{{ .Prompt }}<|im_end|> {{ .Prompt }}<|im_end|>
{{ end }}<|im_start|>assistant {{ end }}<|im_start|>assistant
{{ .Response }}<|im_end|>""" {{ .Response }}<|im_end|>"""

View file

@ -42,9 +42,9 @@ prompt_targets:
description: The policy number for which the insurance claim details are requested. description: The policy number for which the insurance claim details are requested.
type: string type: string
- name: include_expired - name: include_expired
description: Include expired insurance claims in the response. description: whether to include expired insurance claims in the response.
type: string type: bool
default: "false" required: true
endpoint: endpoint:
cluster: api_server cluster: api_server
path: /insurance_claim_details path: /insurance_claim_details

View file

@ -43,7 +43,6 @@ class ArchHandler:
+ ARCH_FUNCTION_CALLING_TOOL_PROMPT.format(tool_text=tool_text) + ARCH_FUNCTION_CALLING_TOOL_PROMPT.format(tool_text=tool_text)
+ "\n\n" + "\n\n"
+ ARCH_FUNCTION_CALLING_FORMAT_PROMPT + ARCH_FUNCTION_CALLING_FORMAT_PROMPT
+ "\n"
) )
return system_prompt return system_prompt