chat_response.py

@action(reads=["prompt", "chat_history", "mode"], writes=["response"])
def chat_response(
        state: State,
        prepend_prompt: str,
        model: str = "gpt-4o",
) -> Tuple[dict, State]:
    chat_history = state["chat_history"].copy()
    chat_history_to_send = chat_history[:-1]
    prompt = f"{prepend_prompt}: {chat_history[-1]['content']}"
    response = _query_openai(
        prompt=prompt,
        chat_history=chat_history_to_send,
        model=model
    )
    return state.update(response={"content": response, "type": "text", "role": "assistant"})