(
prefix: str = "",
suffix: str = "AGENT:",
user_format: str = "USER: {content}\n\n",
agent_format: str = "AGENT: {content}\n\n",
prompt_key: str = "prompt",
)
| 74 | |
| 75 | @staticmethod |
| 76 | def prompt_string( |
| 77 | prefix: str = "", |
| 78 | suffix: str = "AGENT:", |
| 79 | user_format: str = "USER: {content}\n\n", |
| 80 | agent_format: str = "AGENT: {content}\n\n", |
| 81 | prompt_key: str = "prompt", |
| 82 | ): |
| 83 | def prompter(messages: List[Dict[str, str]]): |
| 84 | nonlocal prefix, suffix, user_format, agent_format, prompt_key |
| 85 | prompt = prefix |
| 86 | for item in messages: |
| 87 | if item["role"] == "user": |
| 88 | prompt += user_format.format(content=item["content"]) |
| 89 | else: |
| 90 | prompt += agent_format.format(content=item["content"]) |
| 91 | prompt += suffix |
| 92 | return {prompt_key: prompt} |
| 93 | |
| 94 | return prompter |
| 95 | |
| 96 | |
| 97 | class FastChatAgent(AgentClient): |
nothing calls this directly
no outgoing calls
no test coverage detected