few_shot_message.py

# definition of the few shot messages
messages_few_shots = []
for key, sample in samples.items():
  messages_few_shots.extend([{
    "role": "user",
    "content" : str(sample["input"]),
  },{
    "role": "assistant",
    "content" : str(sample["output"]),
  }
  ])
  
# Set the messages to send to the client
messages = messages_few_shots.copy()
messages.append({
  "role": "user",
  "content" : raw_str_html,
})

# Execute the completion
chat_response = client.chat.complete(
    model = model,
    messages = messages
)