Update config.json

config.json

@@ -1,29 +1,17 @@
 {
- "model_name": "slim-sa-ner-tool",
- "model_ft_base": "slim-sa-ner",
+ "model_name": "slim-extract-tool",
+ "model_ft_base": "slim-extract",
  "quantization": "4Q_K_M GGUF",
  "model_base": "stabilityai/stablelm-3b-4e1t",
  "model_type": "stablelm",
- "tokenizer": "llmware/slim-sa-ner",
+ "tokenizer": "llmware/slim-extract",
  "parameters": "2.7 billion",
- "description": "slim-sa-ner is a function-calling model, fine-tuned to output structured dictionaries corresponding to sentiment analysis classification and major named entity categories",
+ "description": "slim-extract is a function-calling model, fine-tuned to output structured dictionaries corresponding to a custom extraction key",
  "prompt_wrapper": "human_bot",
- "prompt_format": "<human> {context_passage} <classify> sentiment, people, place, organization, misc </classify>\n<bot>:",
- "output_format": "{'sentiment': [], 'people': [], 'place': [], 'organization': [], 'misc':[]}",
- "primary_keys": [
-  "sentiment",
-  "people",
-  "place",
-  "organization",
-  "misc"
- ],
- "output_values": [
-  "sentiment",
-  "people",
-  "place",
-  "organization",
-  "misc"
- ],
+ "prompt_format": "<human> {context_passage} <classify> {custom extraction key} </classify>\n<bot>:",
+ "output_format": "{custom_extraction_key: ['list of items found in the text corresponding to key']}",
+ "primary_keys": ["any custom key provided"],
+ "output_values": ["dictionary with custom_key and value consisting of list of extracted values"],
  "publisher": "llmware",
  "release_date": "march 2024",
  "test_set": [