Fix docstring for expected input format

qcri · Jul 30, 2024 · 88e1b9c · 88e1b9c
1 parent 0b2a087
commit 88e1b9c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/llmebench/models/AzureModel.py b/llmebench/models/AzureModel.py
@@ -92,9 +92,10 @@ def prompt(self, processed_input):
 
  Arguments
  ---------
- processed_input : dictionary
- Must be a dictionary with one key "prompt", the value of which
- must be a string.
+ processed_input : list
+ Must be list of dictionaries, where each dictionary has two keys;
+ "role" defines a role in the chat (e.g. "system", "user") and
+ "content" defines the actual message for that turn
 
  Returns
  -------