Skip to content
This repository was archived by the owner on Jan 10, 2025. It is now read-only.

Commit 276c84c

Browse files
authored
Update docs for create_task (#32)
1 parent 3a5ece7 commit 276c84c

File tree

1 file changed

+19
-4
lines changed

1 file changed

+19
-4
lines changed

docs/python-sdk.md

Lines changed: 19 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -362,13 +362,28 @@ refuel_client.create_task(
362362
dataset='<DATASET NAME>',
363363
input_columns=['col 1', 'col 2' ...],
364364
context = '...',
365-
fields = [{'name': '...', 'guidelines': ...}]
365+
fields = [{'name': '...', 'guidelines': ...}],
366+
model = '...'
366367
)
367368
```
368369

369-
- task_type is one of: `classification`, `multilabel_classification` or `attribute_extraction`
370-
- input_columns is the subset of columns from the dataset that will be used as input for LLM
371-
- fields is a list of dictionaries. Each dictionary contains a fixed set of keys: name (name of the LLM label field as it will be appear in the exported dataset), guidelines (labeling guidelines for the LLM) and labels (list of valid labels, this field is only required for classification type tasks)
370+
- `task_type` is one of: `classification`, `multilabel_classification` or `attribute_extraction`
371+
- `input_columns` is the subset of columns from the dataset that will be used as input for LLM
372+
- `fields` is a list of dictionaries. Each dictionary contains a fixed set of keys:
373+
- `name` (name of the LLM label field as it will be appear in the exported dataset)
374+
- `guidelines` (labeling guidelines for the LLM)
375+
- `labels` (list of valid labels, this field is only required for classification type tasks)
376+
- `model` is an optional parameter to select the LLM that will be used for this task. If not specified, we will use the default LLM set for your team. Here is the list of LLMs currently supported:
377+
- GPT-4 Turbo
378+
- GPT-4
379+
- GPT-3.5 Turbo
380+
- GPT-3.5 Turbo (16K)
381+
- Claude 3 (Opus)
382+
- Claude 3 (Sonnet)
383+
- Claude 3 (Haiku)
384+
- Gemini (Pro)
385+
- Mistral Small
386+
- Mistral Large
372387

373388
### Get Tasks
374389

0 commit comments

Comments
 (0)