升级新库

bbit_ai/app/llm/titleChain.py

@@ -1,10 +1,10 @@
+from langchain_core.prompts import PromptTemplate
 
-from langchain.prompts import PromptTemplate
 from config.llm import llm
 
 titlePrompt = PromptTemplate(
     input_variables=["userStr"],
-    template = """
+    template="""
         请将用户的这句话总结成一个简短、精准的对话标题，要求：
         1. 不超过10个字（可根据需要调整长度）。
         2. 直接概括本次对话的核心内容。
@@ -12,9 +12,10 @@ titlePrompt = PromptTemplate(
         4. 保持自然、易懂、专业或有趣（可根据场景调整风格）。
         5. 不能出现标点符号。
         用户原话："{userStr}"
-    """
+    """,
 )
 titleChain = titlePrompt | llm
 
+
 def get_title(userInput: str):
-    return titleChain.invoke({"userStr": userInput}).content
+    return titleChain.invoke({"userStr": userInput}).content