(messages: List[Dict[str, str]])
| 81 | agent_role: str = "agent", |
| 82 | ): |
| 83 | def prompter(messages: List[Dict[str, str]]): |
| 84 | nonlocal message_key, role_key, content_key, user_role, agent_role |
| 85 | role_dict = { |
| 86 | "user": user_role, |
| 87 | "agent": agent_role, |
| 88 | } |
| 89 | prompt = [] |
| 90 | for item in messages: |
| 91 | prompt.append( |
| 92 | {role_key: role_dict[item["role"]], content_key: item["content"]} |
| 93 | ) |
| 94 | return {message_key: prompt} |
| 95 | |
| 96 | return prompter |
| 97 |
no test coverage detected