Spaces:
Sleeping
Sleeping
File size: 601 Bytes
b60402f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
"""
Chat workflows.
"""
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from app.api.scraper.services.prompts import chat_prompt
from app.core.config import settings
async def generate_chat_name(query: str) -> str:
"""
Generate a name for a chat.
"""
prompt = ChatPromptTemplate.from_messages(
[("system", chat_prompt.generate_chat_name), ("human", query)]
)
chain = prompt | settings.get_llm(model="gpt-4.1-nano", temperature=0.3) | StrOutputParser()
return await chain.ainvoke({"query": query})
|