Function get_few_shot_messages

byod/api/app/utils.py:31–45

(mode: str = "text_to_sql")

def get_few_shot_messages(mode: str = "text_to_sql") -> List[Dict]:
    global IN_CONTEXT_EXAMPLES_DICT
    
    examples = IN_CONTEXT_EXAMPLES_DICT.get(mode, [])
    messages = []
    for example in examples:
        messages.append({
            "role": "user",
            "content": example["user"],
        })
        messages.append({
            "role": "assistant",
            "content": example["assistant"],
        })
    return messages


def get_assistant_message(