找回密码
 立即注册
首页 业界区 业界 AI开发-python-langchain框架(3-13-串行流程 ) ...

AI开发-python-langchain框架(3-13-串行流程 )

胥望雅 7 小时前
现在有这样一个需求,要求使用AI智能化完成这个任务。

  • 1、先处理大纲:构建大纲提示词 → 获得大纲内容
  • 2、再处理注意事项:构建注意事项提示词 → 获得注意事项内容
  • 3、最后整合生成文章:整合所有信息 → 构建最终提示词 → 获得完整文章
流程图如下:
1.png

代码实现:
  1. from langchain_openai import ChatOpenAI
  2. from langchain_core.prompts import ChatPromptTemplate
  3. from langchain_core.output_parsers import StrOutputParser
  4. #目标写一个文章 先让大模型生成大纲和需要注意事项 然后仍给大模型生成文章
  5. #模型接入
  6. DEEPSEEK_API_KEY = "123"  # 替换为实际的 API Key
  7. llm = ChatOpenAI(
  8.     api_key=DEEPSEEK_API_KEY,
  9.     base_url="http://qwen3-32b-awq.model.xjipc.com/openai/v1",
  10.     model="qwen3-32b-awq",
  11.     temperature=0.3,
  12.     max_tokens=10240,
  13. )
  14. #生成大纲提示词
  15. outlinePromptTemplate = '''主题:{theme}
  16. 如果要根据主题写一篇文章,请列出文章的大纲。'''
  17. outlinePrompt = ChatPromptTemplate.from_template(outlinePromptTemplate)
  18. #生成注意事项提示词
  19. tipsPromptTemplate = '''主题:{theme}
  20. 如果要根据主题写一篇文章,应该需要注意哪些方面,才能把这篇文章写好。
  21. '''
  22. tipsPrompt = ChatPromptTemplate.from_template(tipsPromptTemplate)
  23. tipsPrompt
  24. #主题
  25. query = "2025年中国经济走向与运行趋势"
  26. strParser = StrOutputParser()
  27. #生成大纲
  28. outlineChain = outlinePrompt | llm | strParser
  29. outline = outlineChain.invoke({"theme":query})
  30. #生成注意事项
  31. tipsChain = tipsPrompt | llm | strParser
  32. tips = tipsChain.invoke({"theme":query})
  33. print('------------大纲--------------')
  34. print(outline)
  35. print('------------注意事项--------------')
  36. print(tips)
  37. #最终生成文章提示词
  38. articlePromptTemplate = '''主题:{theme}
  39. 大纲:
  40. {outline}
  41. 注意事项:
  42. {tips}
  43. 请根据上面的主题、大纲和注意事项写出丰富的完整文章内容。
  44. '''
  45. articlePrompt = ChatPromptTemplate.from_template(articlePromptTemplate)
  46. #生成最终文章
  47. articleChain = articlePrompt | llm | strParser
  48. resutl = articleChain.invoke({
  49.     "theme":query,
  50.     "outline":outline,
  51.     "tips":tips
  52. })
  53. print('------------生成最终文章--------------')
  54. print(resutl)
复制代码
 
缺点时间效率低


  • 总耗时累加:总执行时间 = 生成大纲时间 + 生成注意事项时间 + 生成文章时间
  • 如果每个LLM调用需要3秒,总耗时至少9秒
如何优化改成并行,下一篇对该流程进行优化改进。
 
 
更多学习资料尽在 老虎网盘资源
来源:程序园用户自行投稿发布,如果侵权,请联系站长删除
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!

相关推荐

您需要登录后才可以回帖 登录 | 立即注册