from langchain_core.messages import AIMessage, BaseMessage, HumanMessagefrom langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholderfrom langchain_fireworks import ChatFireworks
prompt = ChatPromptTemplate.from_messages([("system","You are an essay assistant tasked with writing excellent 5-paragraph essays."" Generate the best essay possible for the user's request."" If the user provides critique, respond with a revised version of your previous attempts.",),MessagesPlaceholder(variable_name="messages"),])
llm = ChatFireworks(model="accounts/fireworks/models/mixtral-8x7b-instruct",model_kwargs={"max_tokens": 32768},)
generate = prompt | llmessay = ""request = HumanMessage(content="Write an essay on why the little prince is relevant in modern childhood")
for chunk in generate.stream({"messages": [request]}):print(chunk.content, end="")essay += chunk.content

Title: The Relevance of The Little Prince in Modern Childhood
The Little Prince, a novella by Antoine de Saint-Exupéry, has been a childhood favorite for generations. Despite being published over seven decades ago, its timeless themes continue to resonate with modern children, making it highly relevant in contemporary childhood.
Firstly, the story explores the complex nature of human relationships, which is particularly relevant for modern children growing up in an increasingly connected yet impersonal world. Through the little prince's encounters with various grown-ups on different planets, the book highlights the importance of genuine connections and understanding. In an age where digital communication often replaces face-to-face interaction, this message is more pertinent than ever. The Little Prince encourages children to look beyond superficial relationships and seek deeper connections, fostering empathy and emotional intelligence.
Secondly, the book deals with the concept of responsibility and self-discovery, elements that are integral to a child's growth. The little prince's journey is essentially a quest for self-discovery, leading him to realize his responsibility towards his beloved rose. This narrative encourages modern children to embrace their individuality while understanding the significance of their actions. In a society that often overlooks the emotional well-being of children, The Little Prince offers a refreshing perspective on personal growth and responsibility.
Thirdly, the book addresses the challenging theme of loss and bereavement. The little prince's departure from his asteroid and his subsequent encounters with the fox and the snake are profound reflections on the inevitability of loss and the importance of cherishing relationships. In a time when children are exposed to various forms of loss, from the death of loved ones to environmental degradation, The Little Prince provides a gentle yet powerful way to understand and cope with these experiences.
However, some critics argue that the book's pace and abstract concepts might be challenging for modern children with short attention spans. To address this, a revised version could incorporate more visual elements and interactive activities to engage young readers better. Additionally, supplementary materials explaining the book's themes in simpler terms could be provided for parents and educators to use in discussions with children.
In conclusion, The Little Prince remains relevant in modern childhood due to its exploration of human relationships, self-discovery, and loss. These themes, wrapped in a captivating narrative, offer valuable lessons for modern children. While some adaptations may be necessary to cater to the preferences of today's children, the essence of the story remains a powerful tool for teaching emotional intelligence, personal growth, and resilience.

reflection_prompt = ChatPromptTemplate.from_messages([("system","You are a teacher grading an essay submission. Generate critique and recommendations for the user's submission."" Provide detailed recommendations, including requests for length, depth, style, etc.",),MessagesPlaceholder(variable_name="messages"),])reflect = reflection_prompt | llm

Essay Grade: B+
The essay you submitted provides a clear and well-structured argument about the relevance of The Little Prince in modern childhood. You have demonstrated a strong understanding of the text and its themes, and have effectively applied them to the context of contemporary childhood. However, there are some areas where improvement could be made to enhance the depth, style, and overall flow of your essay.
1. Length: While your essay is well-written and informative, it is relatively brief. Expanding on each point with more detailed analysis and examples would strengthen your argument and demonstrate a more comprehensive understanding of the text. Aim for a minimum of 500 words to allow for a more in-depth exploration of your ideas.
2. Depth: Although you have touched upon the relevance of the novel's themes, further analysis is needed to truly establish its significance in modern childhood. For example, when discussing the complex nature of human relationships, delve into how the digital age affects children's communication skills, and how The Little Prince addresses this issue. Providing concrete examples from the text and connecting them to real-world scenarios will make your argument more compelling.
3. Style: To engage your readers more effectively, consider varying your sentence structure and length. Using a mix of simple, compound, and complex sentences will improve the flow of your essay and make it more engaging to read. Additionally, watch your tense consistency. Ensure that you maintain the same tense throughout your essay to avoid confusion.
4. Recommendations: While your suggestions for adaptation are a good start, they could be expanded upon to provide more comprehensive recommendations. For example, you may want to discuss different methods of incorporating visual elements and interactive activities, such as illustrations, quizzes, or discussion questions. This will demonstrate that you have thoughtfully considered the needs of modern children and have developed strategies to address these challenges.
5. Conclusion: Your conclusion could benefit from a stronger summarization of your key points and an assertive final statement about the relevance of The Little Prince in modern childhood. Tying all your arguments together in a concise and powerful manner will leave a lasting impression on your readers and solidify your position.
Overall, your essay is well-researched and provides a solid foundation for a compelling argument about the relevance of The Little Prince in modern childhood. With some expansion, deeper analysis, and stylistic improvements, your essay can achieve an even higher level of excellence.

for chunk in generate.stream({"messages": [request, AIMessage(content=essay), HumanMessage(content=reflection)]}):print(chunk.content, end="")

from typing import Annotated, List, Sequencefrom langgraph.graph import END, StateGraph, STARTfrom langgraph.graph.message import add_messagesfrom typing_extensions import TypedDict
class State(TypedDict):messages: Annotated[list, add_messages]
async def generation_node(state: Sequence[BaseMessage]):return await generate.ainvoke({"messages": state})
async def reflection_node(messages: Sequence[BaseMessage]) -> List[BaseMessage]:    # Other messages we need to adjustcls_map = {"ai": HumanMessage, "human": AIMessage}    # First message is the original user request. We hold it the same for all nodestranslated = [messages[0]] + [cls_map[msg.type](content=msg.content) for msg in messages[1:]]res = await reflect.ainvoke({"messages": translated})    # We treat the output of this as human feedback for the generatorreturn HumanMessage(content=res.content)
builder = StateGraph(State)builder.add_node("generate", generation_node)builder.add_node("reflect", reflection_node)builder.add_edge(START, "generate")
def should_continue(state: List[BaseMessage]):if len(state) > 6:        # End after 3 iterationsreturn ENDreturn "reflect"
builder.add_conditional_edges("generate", should_continue)builder.add_edge("reflect", "generate")graph = builder.compile()
async for event in graph.astream([HumanMessage(content="Generate an essay on the topicality of The Little Prince and its message in modern life")],):print(event)    print("---")