summaryrefslogtreecommitdiffstats
path: root/config.example.yaml
diff options
context:
space:
mode:
authorsigoden <sigoden@gmail.com>2024-01-13 19:52:07 +0800
committerGitHub <noreply@github.com>2024-01-13 19:52:07 +0800
commitfe35cfd9419302f01baf9672493c0b0a4b41d889 (patch)
tree94e763745fb7989c97af39cc1dfb44250440a5eb /config.example.yaml
parent4e99df4c1bd4028a77251bdb00ff23c664372b5f (diff)
downloadaichat-fe35cfd9419302f01baf9672493c0b0a4b41d889.tar.gz
feat: supports model capabilities (#297)
1. automatically switch to the model that has the necessary capabilities. 2. throw an error if the client does not have a model with the necessary capabilities
Diffstat (limited to 'config.example.yaml')
-rw-r--r--config.example.yaml7
1 files changed, 5 insertions, 2 deletions
diff --git a/config.example.yaml b/config.example.yaml
index d360a7b..ce12cce 100644
--- a/config.example.yaml
+++ b/config.example.yaml
@@ -36,8 +36,11 @@ clients:
api_key: xxx
chat_endpoint: /chat/completions # Optional field
models:
- - name: gpt4all-j
+ - name: mistral
max_tokens: 8192
+ - name: llava
+ max_tokens: 8192
+ capabilities: text,vision # Optional field, possible values: text, vision
# See https://github.com/jmorganca/ollama
- type: ollama
@@ -45,7 +48,7 @@ clients:
api_key: Basic xxx # Set authorization header
chat_endpoint: /chat # Optional field
models:
- - name: gpt4all-j
+ - name: mistral
max_tokens: 8192
# See https://learn.microsoft.com/en-us/azure/ai-services/openai/chatgpt-quickstart