主干Ai实验室后端项目

bbit_ai/app/llm/titleChain.py

@@ -0,0 +1,19 @@
+
+from langchain.prompts import PromptTemplate
+from config.llm import llm
+
+titlePrompt = PromptTemplate(
+    input_variables=["userStr"],
+    template = """
+        请将用户的这句话总结成一个简短、精准的对话标题，要求：
+        1. 不超过10个字（可根据需要调整长度）。
+        2. 直接概括本次对话的核心内容。
+        3. 避免使用笼统或无意义的词语，如“讨论”、“聊天”等。
+        4. 保持自然、易懂、专业或有趣（可根据场景调整风格）。
+        用户原话："{userStr}"
+    """
+)
+titleChain = titlePrompt | llm
+
+def get_title(userInput: str):
+    return titleChain.invoke({"userStr": userInput}).content