惯性聚合 高效追踪和阅读你感兴趣的博客、新闻、科技资讯
阅读原文 在惯性聚合中打开

推荐订阅源

宝玉的分享
宝玉的分享
OSCHINA 社区最新新闻
OSCHINA 社区最新新闻
云风的 BLOG
云风的 BLOG
V
V2EX
The GitHub Blog
The GitHub Blog
T
Threatpost
Microsoft Azure Blog
Microsoft Azure Blog
Security Latest
Security Latest
A
Arctic Wolf
Google DeepMind News
Google DeepMind News
D
Darknet – Hacking Tools, Hacker News & Cyber Security
V
Vulnerabilities – Threatpost
P
Proofpoint News Feed
W
WeLiveSecurity
Cloudbric
Cloudbric
C
Check Point Blog
Google Online Security Blog
Google Online Security Blog
MyScale Blog
MyScale Blog
T
The Exploit Database - CXSecurity.com
U
Unit 42
cs.CV updates on arXiv.org
cs.CV updates on arXiv.org
PCI Perspectives
PCI Perspectives
cs.CL updates on arXiv.org
cs.CL updates on arXiv.org
Spread Privacy
Spread Privacy
Hugging Face - Blog
Hugging Face - Blog
O
OpenAI News
N
News and Events Feed by Topic
The Last Watchdog
The Last Watchdog
P
Proofpoint News Feed
Last Week in AI
Last Week in AI
MongoDB | Blog
MongoDB | Blog
让小产品的独立变现更简单 - ezindie.com
让小产品的独立变现更简单 - ezindie.com
P
Privacy & Cybersecurity Law Blog
T
Threat Research - Cisco Blogs
博客园 - 聂微东
V
Visual Studio Blog
美团技术团队
Security Archives - TechRepublic
Security Archives - TechRepublic
G
GRAHAM CLULEY
Recent Commits to openclaw:main
Recent Commits to openclaw:main
C
CERT Recently Published Vulnerability Notes
Know Your Adversary
Know Your Adversary
小众软件
小众软件
Engineering at Meta
Engineering at Meta
NISL@THU
NISL@THU
K
KPMG report finds enterprise disconnect between AI and its ROI | CIO
人人都是产品经理
人人都是产品经理
Cisco Talos Blog
Cisco Talos Blog
Martin Fowler
Martin Fowler
T
Troy Hunt's Blog

博客园_首页

Plist 二进制格式 Milvus 和 PGVector,哪个更好? OpenClaw 已过时?在 VS Code 中运行 Hermes Agent! 第30篇文章:一个大三计科生的自白 Manim如何在数学公式中完美显示中文? Docker 部署 RocketMQ 5 并发编程核心概念辨析 C#事务处理最佳实践:别再让“主表存了、明细丢了”的破事发生 CLI 是什么?为什么大厂突然集体卷命令行? 【从0到1构建一个ClaudeAgent】协作-自主Agent UIImageView 设置图片不生效的原因排查 最小二乘问题详解20:无先验约束下的增量式SFM自由网平差 痞子衡嵌入式:大话双核i.MXRT1180之XIP应用里借助MU实现可靠Flash IAP的方法 AI Chat 封装, SemanticKerne.AiProvider.Unified 已发布 Windows下右键编辑js文件无法打开记事本——在注册表中使用环境变量 在后台服务中使用 Scoped 服务,为什么总是报错? H200 安装驱动并使用sglang启动模型 wireshark 抓包Trap上报告警内容 我用 AI 辅助开发了一系列小工具(2):图片压缩工具 [A Primer On MC and CC] 2.1 Memory Consistency 1 - 指令重排序和 SC 模型 Oracle数据库SCN推进技术详解与实践指南 玩转控件:封装个带图片的Label控件 Claude Code 4.7 真正该升级的不是模型,而是你的工作流 前端小白一句话,AI 帮我做了个颜值拉满的桌面媒体播放器。当代码不再是门槛,一句话编程就是现实。 5. WorkBuddy: 小龙虾的灵魂三件套,让你的小龙虾不只是工具 SQLite 分片方案实战:三种分片策略的深度对比 告别简陋 UI!一款基于 Fluent Design 和基于 WinUI 的开源免费、现代化的 Avalonia UI 控件库 关于二进制排列组合枚举的总结 AI开发-python-LangGraph框架(3-27-LangGraph从零实现大模型智能决策工作流) ElasticSearch主分片和副本分片概念详解 【002】HTTPS 粗解:证书、TLS 握手与对后端配置的影响 Hermes Agent 一周暴涨五万 Star,但我劝你别急着追 明明连接的是Redis的DB0,为什么能查到DB3的数据? 【从0到1构建一个ClaudeAgent】协作-Agent团队 熟悉电子元器件之后,电子小白下一步该怎么走? MAF快速入门(23)通过C#类定义Skills .NET 高级开发 | 手写一个对象映射框架 FastAPI数据库ORM怎么选?我肝了三个Demo后,终于不再纠结了 mysqldump 参数拾遗:在遗忘与铭记之间 C# .NET 周刊|2026年3月5期 Claude code入门 - 陈彦斌 一文学习入门 ThingsBoard 开源物联网平台 GitHub 热门项目 | 2026年04月16日 如何为GIT设置全局勾子,为每次提交追加信息 Number.isFinite和isFinite与isNaN()和Number.isNaN的区别 PortSwigger SQL注入LAB2 推荐一个测试人必备的Skills,从功能到性能全搞定(附详细实操和安装下载方式) 筑基期:掌握Odoo基础核心知识点02(Odoo XML 开发方式详解) GLM模型这么火,咱们用vllm也咧一个呗! 深入理解 AbortController:从底层原理到跨语言设计哲学 字符串学习笔记 多租户系统框架的基础模块设计和分析设计 Apache SeaTunnel Zeta 为什么能做到“又快又稳”? AI开发-python-LangGraph框架(3-26-LangGraph基本概念及第一个简单样例) Vue 3 组件通信,别只会用 Props 和 Emits 了,这几个狠活儿你得看看 ElasticSearch7.X版本配置密码 用Manim实现动态交点计算--从一个动点问题说起 团结引擎+Addressable+Instant Game打包抖音小游戏 function call 实战:让 LLM 自动判断 pod 异常、调用日志工具并完成故障分析 bubseek —— 让 Agent 的足迹,变成团队的洞察 通过 C# 读取并导出 PDF 书签 如何用 GitHub Actions 实现 Steam 自动化发布 【从0到1构建一个ClaudeAgent】并发-后台任务 .NET 高级开发 | 定制 ASP.NET Core 框架 电子小白:什么是运算放大器(运放) zero2Agent:面向大厂面试的 Agent 工程教程,从概念到生产的完整学习路线 堆上的ORW HC32F460 USB CDC通信异常:非对齐访问异常排查 20260413-Hyperbridge 攻击事件:发生在默克尔山上的验证绕过 那些喊着AI 要淘汰你的人,正在靠你的焦虑赚大钱! 深度学习进阶(八)Swin Transformer 最小二乘问题详解19:带先验约束的增量式SFM优化与实现 SnapTranslate 3.0 正式发布:全局划词翻译 + 完整英语学习闭环,一站式搞定查词、记词、复习 工作的意义、工作的困难认知再思考 .NET + AI 进阶实战:基于类的技能开发 - 打造可治理的 Agent 能力模块 【从0到1构建一个ClaudeAgent】规划与协调-技能 上周热点回顾(4.6-4.12) 电子小白的工具三件套:面包板、杜邦线、万能板 单表五亿数据的查询优化 | Mysql、StarRocks 2. WorkBuddy:从“我是谁”到“帮我干活” C# 如何减少代码运行时间:7 个实战技巧 基于HelixToolkit.SharpDX 渲染3D模型 - 笺上知微 从零开始的双臂具身VLA起源及现阶段发展综述 - SkyXZ 记对 xonsh shell 的使用, 脚本编写, 迁移及调优 - pluvium27 受够了Vibe Coding的失控?换个起点,让AI事半功倍 从开始配置漏洞环境到漏洞复现流程 - 難しい 关于10年工作经验的程序员对OpenClaw的实战经验分享以及看法 - 虚无境 Any metadata 的内存布局 C# .NET 周刊|2026年3月2期 - InCerry 我帮你测过了,测试圈排名第二的 Skill 依然很牛逼 Skill Discovery | 无监督技能发现的经典工作总结 - MoonOut 上下文工程是什么?过时了么?一文讲明白! - 一枫说码 开了 TUN 模式还是直连?90% 的人都踩过这个坑 AScript扩展多种脚本语言 - rockey627 AI 学习笔记:Agent 的记忆机制 你能被装进一个文件里吗?——7 万人把同事"蒸馏"成了 AI - 我没有三颗心脏 Claude Code 通关手册(七):给 AI 装上技能包——Skills 完全指南 - 暮色之狐 在浏览器中快速编辑代码:VSCode Web 集成实践 - Newbe36524 蒸馏自己 skill?基于 Deepseek 的蒸馏器,丐版蒸馏方式,简单便捷 - To_Carpe_Diem Spring AI Aliababa和AgentScope,哪个更好? - 苏三说技术
12. LangChain 6大核心调用方法:invoke/stream/batch同步异步全解析,新手也能轻松学会
老陈说编程 · 2026-05-07 · via 博客园_首页

做 AI 应用开发的朋友都有体会,不同业务场景对大模型的调用方式,要求天差地别。为了适配各类开发需求,帮大家省去重复造轮子的麻烦,LangChain 针对大模型对话交互,封装了 6 种核心调用模式,几乎覆盖了绝大多数开发场景。

选同步还是异步,最直白的判断标准,就是怕不怕「卡住」。

如果不怕卡,或者说卡了完全不影响 —— 比如本地跑单条脚本、离线处理小批量数据,就安安稳稳等模型返回结果再往下走,那直接选同步方法(invoke/stream/batch)就行,逻辑简单写着省事,不用进行额外处理。

但凡你怕卡住,卡了会出问题 —— 比如做 Web 后端接口、聊天机器人前端交互,不能因为等模型返回,就把整个服务、整个页面卡死,没法接新请求、响应用户中断操作,那就选异步方法(ainvoke/astream/abatch)。异步的核心就是不阻塞主线程,等模型结果的空档,照样能处理其他任务,完全不会出现 “点一下整个界面都动不了” 的情况。

选同步还是异步,就看有没有卡住。

2

 按如下流程图走一遍流程,便可直接选对方法。

首先看有多少个任务。如果只有1个任务,接着看是否需要逐块显示;如果是,就在异步环境或UI界面里选astream,否则选 stream。如果不需要逐块显示,就在异步环境或UI界面里选ainvoke,否则选 invoke。如果是多个任务,就在异步环境或UI界面里选abatch,否则选batch。

image

 以下为.env 环境配置文件的内容,记得必将其中的 API_KEY 占位值,替换为您自行在对应平台申请的有效API密钥。

QWEN_API_KEY="你的QWEN API KEY"
QWEN_BASE_URL="https://dashscope.aliyuncs.com/compatible-mode/v1"

前面我们已经实操掌握了 invoke 单次调用和 stream 流式输出,接下来学习 batch 批量调用。我们先定义多条任务请求,虽说传入了多个任务,但批量调用的语法和 invoke 完全一致。批量任务执行完毕后,通过循环遍历,依次输出每一个任务的返回结果。

import os
from dotenv import load_dotenv
from langchain.chat_models import init_chat_model
 
load_dotenv()
prefix = "QWEN"
llm = init_chat_model(
    model_provider="openai",
    configurable_fields=["model", "api_key", "base_url"],
    config_prefix=prefix,
    temperature=0.5,
    max_tokens=200
)
config = {
    "configurable": {
        f"{prefix}_model": os.getenv(f"{prefix}_MODEL"),
        f"{prefix}_api_key": os.getenv(f"{prefix}_API_KEY"),
        f"{prefix}_base_url": os.getenv(f"{prefix}_BASE_URL")
    }
}
prompts = [
    "生成Java打印Hello World的代码,只提供一种最标准的写法",
    "生成C#打印Hello World的代码,只提供一种最标准的写法",
    "生成Python打印Hello World的代码,只提供一种最标准的写法"
]
responses = llm.batch(prompts, config=config)
for res in responses:
    print(res.content)

熟悉同步 batch 用法后,我们再进阶切换到异步abatch。

首先定义异步函数 async_batch,在函数内部调用大模型的 abatch 方法。

最后用 asyncio.run 启动并运行这个异步函数。

import os
from dotenv import load_dotenv
from langchain.chat_models import init_chat_model
import asyncio
 
load_dotenv()
prefix = "QWEN"
llm = init_chat_model(
    model_provider="openai",
    configurable_fields=["model", "api_key", "base_url"],
    config_prefix=prefix,
    temperature=0.5,
    max_tokens=200
)
config = {
    "configurable": {
        f"{prefix}_model": os.getenv(f"{prefix}_MODEL"),
        f"{prefix}_api_key": os.getenv(f"{prefix}_API_KEY"),
        f"{prefix}_base_url": os.getenv(f"{prefix}_BASE_URL")
    }
}
prompts = [
    "生成Java打印Hello World的代码,只提供一种最标准的写法",
    "生成C#打印Hello World的代码,只提供一种最标准的写法",
    "生成Python打印Hello World的代码,只提供一种最标准的写法"
]
async def async_batch():
    responses = await llm.abatch(prompts, config=config)
    for resp in responses:
        print(resp.content)
 
asyncio.run(async_batch())

接下来,我们来学习异步单次调用方法 ainvoke。

我们把函数名修改一下,内部通过ainvoke,传入单个提示词和配置参数,等待异步请求完成后,直接打印返回内容。

同样使用 asyncio.run 运行函数,就完成了大模型的异步单次调用。

import os
from dotenv import load_dotenv
from langchain.chat_models import init_chat_model
import asyncio
 
load_dotenv()
prefix = "QWEN"
llm = init_chat_model(
    model_provider="openai",
    configurable_fields=["model", "api_key", "base_url"],
    config_prefix=prefix,
    temperature=0.5,
    max_tokens=200
)
config = {
    "configurable": {
        f"{prefix}_model": os.getenv(f"{prefix}_MODEL"),
        f"{prefix}_api_key": os.getenv(f"{prefix}_API_KEY"),
        f"{prefix}_base_url": os.getenv(f"{prefix}_BASE_URL")
    }
}
 
prompt ="生成Python打印Hello World的代码,只提供一种最标准的写法"
 
async def async_invoke():
    response = await llm.ainvoke(prompt, config=config)
    print(response.content)
 
asyncio.run(async_invoke())

继续进阶,学习异步流式输出 astream 方法。

提示词保持不变,重新定义异步函数 async_stream。它不会等待完整结果返回,而是通过 async for循环,监听llm.astream的实时数据流。

大模型每生成一段内容片段,就立即打印输出,设置end不换行、flush强制刷新输出,实现逐字打字机效果。

最后依旧用 asyncio.run 运行函数,就能看到实时流式输出的效果。

import os
from dotenv import load_dotenv
from langchain.chat_models import init_chat_model
import asyncio
 
load_dotenv()
prefix = "QWEN"
llm = init_chat_model(
    model_provider="openai",
    configurable_fields=["model", "api_key", "base_url"],
    config_prefix=prefix,
    temperature=0.5,
    max_tokens=200
)
config = {
    "configurable": {
        f"{prefix}_model": os.getenv(f"{prefix}_MODEL"),
        f"{prefix}_api_key": os.getenv(f"{prefix}_API_KEY"),
        f"{prefix}_base_url": os.getenv(f"{prefix}_BASE_URL")
    }
}
 
prompt = "生成Python打印Hello World的代码,只提供一种最标准的写法"
async def async_stream():
    async for chunk in llm.astream(prompt, config=config):
        print(chunk.content, end="", flush=True)
asyncio.run(async_stream())

接下来,我们将通过一个完整的实战项目,带你将astream流式响应功能集成到 PySide6 搭建的UI界面中。

本项目结构清晰,只有3 个文件,分别为环境变量配置文件 .env、环境变量读写工具模块 env_util.py,以及 PySide6 界面交互与 astream 流式功能集成的核心业务文件 chat_page.py。

在正式编写代码、运行项目前,你需要先通过 pip 命令安装项目所需的全部依赖开发包,具体安装指令如下:

pip install pyside6 dotenv langchain langchain-openai

完成依赖安装后,我们先从项目的基础环境配置入手,.env 环境变量配置文件的完整内容如下:

# ========== 阿里巴巴-通义千问 ==========
QWEN_API_KEY="你的QWEN API KEY"
QWEN_BASE_URL='https://dashscope.aliyuncs.com/compatible-mode/v1'
QWEN_CONSOLE_URL='https://dashscope.console.aliyun.com/'
QWEN_MODELS='qwen3.6-plus,qwen-3.5-plus,qwen-3.5-72b-instruct'
 
# ========== 月之暗面 - Kimi ==========
KIMI_API_KEY=''
KIMI_BASE_URL='https://api.moonshot.cn/v1'
KIMI_CONSOLE_URL='https://platform.moonshot.cn/'
KIMI_MODELS='moonshot-v1-8k,moonshot-v1-128k,moonshot-v1-256k'
 
# ========== MiniMax ==========
MINIMAX_API_KEY=''
MINIMAX_BASE_URL='https://api.minimaxi.com/v1'
MINIMAX_CONSOLE_URL='https://platform.minimax.chat/'
MINIMAX_MODELS='minimax-m2.7,abab-6.5-pro,minimax-m2-her'
 
# ========== 智谱AI-智谱清言 ==========
ZHIPU_API_KEY=''
ZHIPU_BASE_URL='https://open.bigmodel.cn/api/paas/v4'
ZHIPU_CONSOLE_URL='https://open.bigmodel.cn/'
ZHIPU_MODELS='glm-4.5-flash,glm-4.5-pro,glm-4-air'
 
# ========== 字节跳动-豆包 ==========
DOUBAO_API_KEY=''
DOUBAO_BASE_URL='https://ark.cn-beijing.volces.com/api/v3'
DOUBAO_CONSOLE_URL='https://console.volcengine.com/ark/'
# 重要:豆包不直接使用模型名称,必须替换为火山方舟创建的推理接入点ID
DOUBAO_MODELS='doubao-pro-32k,doubao-lite-32k,doubao-pro-128k'
 
# ========== 百度-文心一言 ==========
ERNIE_API_KEY=''
ERNIE_BASE_URL='https://qianfan.baidubce.com/v2'
ERNIE_CONSOLE_URL='https://console.bce.baidu.com/qianfan/'
ERNIE_MODELS='ernie-4.0-turbo-8k,ernie-3.5-turbo-128k,ernie-speed-128k'
 
# ========== 腾讯-混元 ==========
HUNYUAN_API_KEY=''
HUNYUAN_BASE_URL='https://api.hunyuan.cloud.tencent.com/v1'
HUNYUAN_CONSOLE_URL='https://console.cloud.tencent.com/hunyuan/'
HUNYUAN_MODELS='hunyuan-turbo,hunyuan-pro,hunyuan-lite'
 
# ========== DeepSeek ==========
DEEPSEEK_API_KEY=''
DEEPSEEK_BASE_URL='https://api.deepseek.com/v1'
DEEPSEEK_CONSOLE_URL='https://platform.deepseek.com/'
DEEPSEEK_MODELS='deepseek-chat,deepseek-reasoner'

为了便捷、安全地管理.env文件中的配置项,我们专门封装了环境变量处理工具模块,env_util.py的完整源代码如下:

import os
from dotenv import load_dotenv, set_key
 
 
# @老陈说编程 哔哩哔哩、今日头条
class EnvUtil:
    def __init__(self, env_path=None):
        if env_path is None:
            self.env_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), ".env")
        else:
            self.env_path = env_path
        load_dotenv(self.env_path)
 
    def load_config(self, providers):
        config = {}
        for provider in providers:
            api_key = os.getenv(f"{provider}_API_KEY", "")
            base_url = os.getenv(f"{provider}_BASE_URL", "")
            console_url = os.getenv(f"{provider}_CONSOLE_URL", "")
            models_str = os.getenv(f"{provider}_MODELS", "")
            models = [m.strip() for m in models_str.split(",")] if models_str else []
 
            config[provider] = {
                "api_key": api_key,
                "base_url": base_url,
                "console_url": console_url,
                "models": models
            }
        return config
 
    def save_config(self, config):
        if not os.path.exists(self.env_path):
            return
 
        for provider, config_item in config.items():
            set_key(self.env_path, f"{provider}_API_KEY", config_item["api_key"])
            set_key(self.env_path, f"{provider}_BASE_URL", config_item["base_url"])
            set_key(self.env_path, f"{provider}_MODELS", ",".join(config_item["models"]))
 
        load_dotenv(self.env_path, override=True)

完成环境配置与工具模块的封装后,就进入了项目的核心实现环节。以下是基于 PySide6 实现 UI 界面渲染、并完整集成 astream 流式响应能力的chat_page.py 完整源代码:

import sys
import os
from PySide6.QtCore import Qt, QSize, Signal, QThread, QTimer
from PySide6.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QTextEdit, QTextBrowser, QFrame, QSizePolicy, QComboBox, QApplication, QPushButton
from PySide6.QtGui import QIcon
import asyncio
from env_util import EnvUtil
 
# @老陈说编程 哔哩哔哩、今日头条
class IconButton(QPushButton):
    def __init__(self, icon_name, parent=None):
        super().__init__(parent)
        self.icon_name = icon_name
        self.setFixedSize(40, 40)
        self.setStyleSheet("""
            QPushButton {
                border: none;
                background-color: transparent;
            }
            QPushButton:hover {
                background-color: rgba(0, 0, 0, 0.1);
                border-radius: 8px;
            }
        """)
        self.set_icon(icon_name)
    
    def set_icon(self, icon_name):
        self.icon_name = icon_name
        icon_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "Icons", icon_name)
        if os.path.exists(icon_path):
            self.setIcon(QIcon(icon_path))
            self.setIconSize(QSize(24, 24))
 
 
class ChatPage(QWidget):
 
    def __init__(self, parent=None):
        super().__init__(parent)
        self.current_output = ""
        self.worker = None
        self.thinking_timer = QTimer()
        self.thinking_timer.timeout.connect(self.update_thinking)
        self.dot_count = 0
        self.thinking_prefix = ""
        self.is_thinking = False
        
        self.env_util = EnvUtil()
        
        self.model_configs = {}
        self.all_models = []
        self.load_models_from_env()
 
        layout = QVBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(0)
 
        self.model_combo = QComboBox()
        self.model_combo.addItems(self.all_models)
        self.model_combo.setFixedWidth(180)
        self.apply_combo_style(self.model_combo)
        
        self.model_combo.currentTextChanged.connect(self.on_model_changed)
        self.current_model = self.model_combo.currentText()
 
        chat_area = QWidget()
        chat_layout = QVBoxLayout(chat_area)
        chat_layout.setContentsMargins(40, 20, 40, 0)
        chat_layout.setSpacing(0)
 
        output_wrapper = QWidget()
        output_wrapper_layout = QHBoxLayout(output_wrapper)
        output_wrapper_layout.setContentsMargins(0, 0, 0, 0)
        output_wrapper_layout.addStretch()
 
        self.output_text = QTextBrowser()
        self.output_text.setMinimumWidth(820)
        self.output_text.setMaximumWidth(1040)
        self.output_text.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.output_text.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.output_text.setStyleSheet("""
            QTextBrowser {
                background-color: #ffffff;
                border: none;
                padding: 12px;
                font-size: 14px;
                color: #333333;
            }
        """)
 
        output_wrapper_layout.addWidget(self.output_text)
        output_wrapper_layout.addStretch()
 
        chat_layout.addWidget(output_wrapper, 1)
        layout.addWidget(chat_area, 1)
 
        bottom_area = QWidget()
        bottom_area.setStyleSheet("background-color: #ffffff;")
        bottom_layout = QVBoxLayout(bottom_area)
        bottom_layout.setContentsMargins(40, 0, 40, 20)
        bottom_layout.setSpacing(0)
 
        input_frame = QFrame()
        input_frame.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Fixed)
        input_frame.setMinimumWidth(800)
        input_frame.setMaximumWidth(1000)
        self.input_frame = input_frame
        self.apply_input_frame_style()
        frame_layout = QVBoxLayout(input_frame)
        frame_layout.setContentsMargins(16, 12, 16, 12)
        frame_layout.setSpacing(4)
 
        self.input_text = QTextEdit()
        self.input_text.setPlaceholderText("请输入消息...")
        self.input_text.setMaximumHeight(60)
        self.input_text.setMinimumHeight(20)
        self.input_text.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.input_text.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.apply_input_text_style()
 
        bottom_bar = QWidget()
        bottom_bar_layout = QHBoxLayout(bottom_bar)
        bottom_bar_layout.setContentsMargins(0, 0, 0, 0)
 
        bottom_bar_layout.addWidget(self.model_combo)
        bottom_bar_layout.addSpacing(8)
        bottom_bar_layout.addStretch()
 
        self.clear_btn = IconButton("clear.png")
        self.clear_btn.setFixedSize(40, 40)
        self.clear_btn.hide()
        self.clear_btn.clicked.connect(self.clear_input)
 
        self.send_btn = IconButton("send_default.png")
        self.send_btn.setFixedSize(40, 40)
        self.send_btn.clicked.connect(self.on_send_clicked)
 
        bottom_bar_layout.addWidget(self.clear_btn)
        bottom_bar_layout.addWidget(self.send_btn)
 
        frame_layout.addWidget(self.input_text)
        frame_layout.addWidget(bottom_bar)
 
        center_layout = QHBoxLayout()
        center_layout.addStretch()
        center_layout.addWidget(input_frame)
        center_layout.addStretch()
 
        bottom_layout.addLayout(center_layout)
        layout.addWidget(bottom_area)
 
        self.setStyleSheet("background-color: #ffffff;")
        self.input_text.installEventFilter(self)
        self.input_text.textChanged.connect(self.on_text_changed)
 
        self.init_chat_model()
    
    def closeEvent(self, event):
        # 窗口关闭时清理线程
        if self.worker and self.worker.isRunning():
            self.worker.stop()
            self.worker.wait()
        self.thinking_timer.stop()
        event.accept()
 
    def apply_input_frame_style(self):
        self.input_frame.setStyleSheet("""
            QFrame {
                border: 1px solid #cccccc;
                border-radius: 16px;
            }
        """)
 
    def apply_input_text_style(self):
        self.input_text.setStyleSheet("""
            QTextEdit {
                background-color: transparent;
                border: none;
                font-size: 14px;
                color: #333333;
                padding: 0;
            }
            QTextEdit:focus {
                outline: none;
            }
        """)
 
    def apply_combo_style(self, combo):
        combo.setStyleSheet("""
            QComboBox {
                background-color: #f5f5f5;
                color: #333333;
                border: 1px solid #cccccc;
                border-radius: 4px;
                padding: 4px 8px;
            }
            QComboBox::drop-down {
                border: none;
                width: 20px;
            }
            QComboBox::down-arrow {
                width: 12px;
                height: 12px;
            }
            QComboBox QAbstractItemView {
                background-color: #ffffff;
                color: #333333;
                selection-background-color: #0078d4;
                selection-color: white;
            }
        """)
 
    def clear_input(self):
        self.input_text.clear()
 
    def on_text_changed(self):
        has_text = len(self.input_text.toPlainText().strip()) > 0
        if has_text:
            self.clear_btn.show()
            self.send_btn.set_icon("send.png")
        else:
            self.clear_btn.hide()
            self.send_btn.set_icon("send_default.png")
 
    def eventFilter(self, obj, event):
        if obj == self.input_text and event.type() == event.Type.KeyPress:
            if event.key() == Qt.Key.Key_Return or event.key() == Qt.Key.Key_Enter:
                if not event.modifiers() & Qt.KeyboardModifier.ShiftModifier:
                    self.on_send_clicked()
                    return True
        return super().eventFilter(obj, event)
 
    def scroll_to_bottom(self):
        cursor = self.output_text.textCursor()
        cursor.movePosition(cursor.MoveOperation.End)
        self.output_text.setTextCursor(cursor)
        self.output_text.ensureCursorVisible()
 
    def clear_chat(self):
        self.current_output = ""
        self.is_thinking = False
        self.dot_count = 0
        self.thinking_prefix = ""
        self.output_text.clear()
        self.input_text.clear()
        self.send_btn.setEnabled(True)
        self.input_text.setEnabled(True)
        self.worker = None
        self.thinking_timer.stop()
 
    def load_models_from_env(self):
        providers = ["QWEN", "KIMI", "MINIMAX", "ZHIPU", "DOUBAO", "ERNIE", "HUNYUAN", "DEEPSEEK"]
        config = self.env_util.load_config(providers)
        
        for provider, provider_config in config.items():
            for model in provider_config["models"]:
                self.model_configs[model] = {
                    "api_key": provider_config["api_key"],
                    "base_url": provider_config["base_url"],
                    "provider": provider.lower()
                }
                self.all_models.append(model)
 
    def on_model_changed(self, model_name):
        if model_name in self.model_configs:
            self.init_chat_model(model_name)
 
    def init_chat_model(self, model_name=None):
        from langchain.chat_models import init_chat_model
        
        if model_name is None:
            model_name = self.model_combo.currentText()
        
        if model_name in self.model_configs:
            config = self.model_configs[model_name]
            self.model = init_chat_model(
                model=model_name,
                model_provider="openai",
                api_key=config["api_key"],
                base_url=config["base_url"],
                temperature=0.5
            )
 
    def update_thinking(self):
        self.dot_count = (self.dot_count + 1) % 4
        dots = "." * self.dot_count
        self.thinking_prefix = "思考中" + dots
        if "\n\n" in self.current_output:
            temp_output = self.current_output.rsplit("\n\n", 1)[0]
        else:
            temp_output = self.current_output
        self.output_text.setPlainText(temp_output + "\n\n" + self.thinking_prefix)
        self.scroll_to_bottom()
 
    def on_send_clicked(self):
        user_input = self.input_text.toPlainText().strip()
        if not user_input or self.worker:
            return
        
        self.current_model = self.model_combo.currentText()
        
        if self.current_output:
            self.current_output += "\n\n"
        self.current_output += user_input
        self.output_text.setPlainText(self.current_output)
        self.scroll_to_bottom()
        
        self.current_output += "\n\n思考中"
        self.thinking_prefix = "思考中"
        self.output_text.setPlainText(self.current_output)
        self.scroll_to_bottom()
        
        self.input_text.clear()
        self.send_btn.setEnabled(False)
        self.input_text.setEnabled(False)
        
        self.dot_count = 0
        self.is_thinking = True
        self.thinking_timer.start(500)
        
        self.worker = AsyncWorker(self.model, user_input)
        self.worker.text_received.connect(self.on_text_received)
        self.worker.finished.connect(self.on_finished)
        self.worker.start()
 
    def on_text_received(self, ai_text):
        if self.thinking_timer.isActive():
            self.thinking_timer.stop()
        if self.is_thinking:
            self.is_thinking = False
            self.current_output = self.current_output.rsplit("\n\n", 1)[0]
            self.current_output += "\n\n"
        new_output = self.current_output + ai_text
        current_text = self.output_text.toPlainText()
        if current_text != new_output:
            self.output_text.setPlainText(new_output)
            self.scroll_to_bottom()
 
    def on_finished(self):
        self.thinking_timer.stop()
        self.is_thinking = False
        self.send_btn.setEnabled(True)
        self.input_text.setEnabled(True)
        self.worker = None
 
 
class AsyncWorker(QThread):
    text_received = Signal(str)
    finished = Signal()
    
    def __init__(self, model, user_input):
        super().__init__()
        self.model = model
        self.user_input = user_input
        self.full_text = ""
        self.last_emit_time = 0
        self._is_running = True
    
    def stop(self):
        self._is_running = False
    
    def run(self):
        import time
        import asyncio
        
        try:
            asyncio.run(self.async_run())
        except Exception as e:
            print(f"AsyncWorker error: {e}")
        finally:
            self.finished.emit()
    
    async def async_run(self):
        import time
        try:
            async for chunk in self.model.astream(self.user_input):
                if not self._is_running:
                    break
                if chunk.content:
                    self.full_text += chunk.content
                    current_time = time.time()
                    if current_time - self.last_emit_time > 0.033:
                        self.text_received.emit(self.full_text)
                        self.last_emit_time = current_time
            if self._is_running:
                self.text_received.emit(self.full_text)
        except Exception as e:
            print(f"async_run error: {e}")
            if self._is_running:
                self.text_received.emit(self.full_text)
 
 
if __name__ == "__main__":
    app = QApplication(sys.argv)
    chat_page = ChatPage()
    chat_page.setWindowTitle("AI 聊天")
    chat_page.resize(1200, 800)
    chat_page.show()
    sys.exit(app.exec())

代码中用到的图标,放在Icons目录下,下载地址:图标下载