在 Python 中调用 API 并实现流式输出通常依赖于服务端支持流式响应，客户端通过逐块接收响应内容实现逐步输出。以下是几种常见方法及实现思路： ### 1. 使用 `requests` 实现基于 HTTP 流式响应的输出 对于支持流式传输的 HTTP API，可以通过设置 `stream=True` 参数来逐步读取响应内容。 ```python import requests response = requests.get('https://api.example.com/stream-endpoint', stream=True) for chunk in response.iter_content(chunk_size=1024): if chunk: print(chunk.decode('utf-8'), end='') ``` 这种方式适用于服务端采用分块传输编码（chunked transfer encoding）的情况。通过 `iter_content()` 或 `iter_lines()` 方法，客户端可以逐步处理接收到的数据[^3]。 --- ### 2. 使用 Google Gemini API 实现流式生成 Google Gemini 提供了 `generate_content_stream()` 方法，可以实现逐块接收生成内容。以下是一个示例： ```python from google import genai client = genai.Client(api_key="GEMINI_API_KEY") response = client.models.generate_content_stream( model="gemini-2.0-flash", contents=["Explain how AI works"] ) for chunk in response: print(chunk.text, end="") ``` 此方法适用于生成式 AI 模型，允许在生成过程中逐步输出结果，实现流式体验[^2]。 --- ### 3. 使用 WebSocket 实现双向流式通信 对于需要双向通信的流式场景（如实时聊天或事件推送），可以使用 WebSocket 协议进行连接。Python 中常用的库包括 `websockets` 和 `socket.io`。 ```python import asyncio import websockets async def listen(): uri = "wss://example.com/websocket" async with websockets.connect(uri) as websocket: async for message in websocket: print(message) asyncio.get_event_loop().run_until_complete(listen()) ``` WebSocket 提供了持久连接，使得服务端可以主动推送数据到客户端，适用于需要实时性的流式输出场景[^1]。 --- ### 4. 自定义流式处理逻辑 在实际应用中，流式输出可能涉及多个处理节点，例如工作流引擎（如 FastGPT、Dify）中的每个节点都需要逐步返回结果。此时可以采用回调函数或事件驱动机制实现动态发送。 ```python def send_stream(callback): for data in data_generator(): callback(data) def handle_data(chunk): print(chunk) send_stream(handle_data) ``` 通过将输出逻辑抽象为回调函数，可以在每一步处理完成后动态发送数据，实现灵活的流式响应机制[^3]。 --- ### 5. 处理流式过程中的后续操作 在流式输出过程中，若希望在 `return` 之后继续执行其他操作，可以采用异步编程或协程方式。Python 中的 `async/await` 结构允许在等待 I/O 操作时执行其他任务。 ```python import asyncio async def stream_output(): for i in range(5): print(f"Chunk {i}") await asyncio.sleep(1) print("Streaming completed") async def main(): task = asyncio.create_task(stream_output()) await asyncio.sleep(0.5) print("Performing other operations...") await task asyncio.run(main()) ``` 上述代码在流式输出的同时，可以执行其他任务，避免阻塞主线程[^3]。 ---