| from langchain_community.chat_models import ChatOllama | |
| from langchain.prompts import ChatPromptTemplate | |
| from langchain_core.output_parsers import StrOutputParser | |
| from langchain_core.runnables import RunnablePassthrough | |
| def generate_title(message: str) -> str: | |
| prompt_template = """Generate one concise and descriptive title that is not longer than 3 words. Only generate one that title. Only generate 3 words. No alternatives. No explanation. Generate this title based on the following message: | |
| Message: {message} | |
| Title: """ | |
| prompt = ChatPromptTemplate.from_template(prompt_template) | |
| model = ChatOllama(model="mistral") | |
| title_chain = ( | |
| {"message": RunnablePassthrough()} | |
| | prompt | |
| | model | |
| | StrOutputParser() | |
| ) | |
| result = title_chain.invoke(message) | |
| return result | |