三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

用 Python 接生图接口:从同步到异步并发的完整演进

用 Python 接生图接口:从同步到异步并发的完整演进

Python 怎么调 AI 生图 API?三十行跑通 nano banana pro(含重试与异步并发)

网上 Python 调图像生成接口的例子大多绑死某一家 SDK,换个服务就得重写。其实完全不用 SDK——该服务兼容 OpenAI 的chat/completions协议,requests一个库就够了。

这篇给一份能直接抄进项目的实现:同步版三十行跑通,再补上重试和asyncio并发。

一、接口信息

Base URLhttps://API_HOST
接口POST /v1/chat/completions
鉴权Authorization: Bearer {你的API密钥}
数据格式application/json,UTF-8

二、最小实现

importosimportrequests ENDPOINT="https://API_HOST/v1/chat/completions"defgenerate(prompt:str,model:str="nano-banana-pro",timeout:int=120)->dict:"""调 AI 绘画API 出图。生图慢,timeout 一定要给足。"""resp=requests.post(ENDPOINT,headers={"Authorization":f"Bearer{os.environ['TTQ_API_KEY']}","Content-Type":"application/json",},json={"model":model,"messages":[{"role":"user","content":prompt}]},timeout=timeout,)resp.raise_for_status()returnresp.json()if__name__=="__main__":data=generate("白底电商主图,无线蓝牙耳机,居中构图,柔和顶光")print(data)

两个新手最容易踩的坑:密钥别写死在代码里(走环境变量,别提交进仓库);timeout别用默认值——requests默认是不超时的,网络一抖请求就永久挂住,跑批时整个进程卡死。

三、加重试

生图 API偶发 5xx 和网络抖动是常态,没有重试的批量脚本基本跑不完:

importtimefromrequests.exceptionsimportRequestExceptiondefgenerate_with_retry(prompt:str,model:str="nano-banana-pro",attempts:int=3)->dict:last=Noneforiinrange(attempts):try:returngenerate(prompt,model)exceptRequestExceptionase:last=e# 指数退避:2s、4s、8s。别用固定间隔,服务端抖动时会同时重试放大压力time.sleep(2**(i+1))raiselast

注意只对网络异常和 5xx重试。4xx 是请求本身有问题(密钥错、参数错、积分不足),重试多少次都一样,白白浪费时间。

四、异步并发

批量出图用asyncio+httpx比多线程干净得多,用信号量把并发压在配额之内:

importasyncioimporthttpxasyncdefbatch_generate(prompts:list[str],model:str="nano-banana2",workers:int=16):sem=asyncio.Semaphore(workers)# 同时在飞的请求数headers={"Authorization":f"Bearer{os.environ['TTQ_API_KEY']}"}asyncwithhttpx.AsyncClient(timeout=120)asclient:asyncdefone(idx:int,prompt:str):asyncwithsem:try:r=awaitclient.post(ENDPOINT,headers=headers,json={"model":model,"messages":[{"role":"user","content":prompt}],})r.raise_for_status()returnidx,r.json()exceptException:returnidx,None# 失败留空,跑完统一挑出来重试results=awaitasyncio.gather(*(one(i,p)fori,pinenumerate(prompts)))out=[None]*len(prompts)foridx,datainresults:out[idx]=datareturnout

结果按下标写回是关键——跑完一眼就能定位哪几条失败,不用整批重来。workers按自己配额调,16 起步比较稳。

五、模型怎么选

model字段直接换字符串,函数签名都不用动:

模型代码说明消耗
nano-banana2Flash 级,速度快,适合跑量8 积分/次
nano-banana-proPro 级,支持 1K/2K/4K10 积分/次
gpt-image-2细节质感好,适合终稿按次计费
Veo 3.1 视频生成/Sora 2生视频120 / 130 积分/次

实践上的分工:批量草稿用nano-banana2终稿切nano-banana-progpt-image-2再出一版对比,挑最好的用。

六、小结

Python 接生图接口的要点就四条:密钥走环境变量、timeout给足、只对 5xx 重试、并发用信号量压住。

该服务的 OpenAI 兼容协议让 Python 侧完全不需要 SDK,requestshttpx直接调。模型全(nano banana pro、gpt-image-2、Veo 3.1、Sora 2 一个密钥全接)、价格便宜(积分制明码标价)、并发高出图快稳定


接口服务:甜甜圈API(dashengfenshen.cn)

← 返回列表