近乎免费的 Gemini Flash,有了结构化输出

科技   2024-08-31 13:03   广东  

本篇内容核心:Google Gemini 1.5 Flash 已支持「结构化输出」,非常经济实惠

官方文档还没出:为了方便大家使用,我写了个 Sample Code,放在了下方


Logan 是原 OpenAI 开发者关系负责人

后去了 Google,也是负责开发者关系



「格式化输出」很重要


上一篇中讲到:结构化输出,是绝大多数 AI 产品和 Agent 的核心,无论是 AI 搜索、有记忆的 bot,还是各类 agent,都是基于结构化输出搭建的。

在上一篇中,你可以看到有关的起源、原理、用例,以及如何用它来搭建一个 AI 项目,这里就不再赘述:

看完这篇,你也能做 AI 搜索:论「结构化输出」


通过结构化输出,可以让 AI 输出一份思维导图或者表格:


而不是成篇的文本:

史蒂夫·乔布斯,1955年2月24日出生,2011年10月5日去世,美国人。他活跃于科技、创新、企业管理和动画领域。乔布斯创立了Apple、NeXT和Pixar公司,推出了Mac、iPod、iPhone等具有划时代意义的产品,重塑了个人电脑、音乐和手机行业,奠定了苹果在全球科技领域的领军地位。作为Pixar的创办人之一,他也在动画领域留下了深远的影响。乔布斯是20世纪末至21世纪初最具影响力的企业家和创新者之一。


Gemini Flash 的结构化输出

Google 家之前也有结构化输出,不过是在他们的贵的模型,Gemini 1.5 Pro 上面


而这次的更新,可以让廉价模型 1.5 Flash 也用上了:

  • 100 万 token 的上下文

  • 每天前 1500 个请求免费

  • 调用价格低至 $0.075 每 100 万 token(长度少于 128k 的上文)

  • 可叠加 GCP 的赞助/优惠

  • 四舍五入不要钱

  • 智谱的 Flash 是完全不要钱

相信做 AI 项目的同学都知道这意味了什么:这便宜大碗,而且 AI 味不重的 Flash,可正儿八经用在决策 workflow 了!

每天 1500 个免费请求

超出部分低至 $0.075 / 1M,外加项目折扣



Sample Code


目前,官方的调用方法还没出(会在本周末更新),但我可以给大家提供一份 sample code


我们把上一篇《看完这篇,你也能做 AI 搜索:论「结构化输出」》中,“将四大名著的信息进行结构化输出”的例子拿来做对比,通过 GPT,代码这么写

from pydantic import BaseModel
class theBook(BaseModel): name: str writer: str
class theFour(BaseModel): books: list[theBook]
completion = client.beta.chat.completions.parse( model="gpt-4o-2024-08-06", messages=[ {"role": "system", "content": "Extract the event information."}, {"role": "user", "content": "告诉我四大名著分别是什么,以及他们的作者是谁"}, ], response_format = theFour,)
response = completion.choices[0].message.parsed

得到的结果是

theFour(books=[theBook(name='《红楼梦》', writer='曹雪芹'), theBook(name='《西游记》', writer='吴承恩'), theBook(name='《三国演义》', writer='罗贯中'), theBook(name='《水浒传》', writer='施耐庵')])


而通过 Flash,代码是类似这样的

"""Install the Google AI Python SDK
$ pip install google-generativeai$ pip install google.ai.generativelanguage"""
import osimport google.generativeai as genaifrom google.ai.generativelanguage_v1beta.types import content
genai.configure(api_key=os.environ["GEMINI_API_KEY"])
# Create the modelgeneration_config = { "temperature": 1, "top_p": 0.95, "top_k": 64, "max_output_tokens": 8192, "response_schema": content.Schema( type = content.Type.OBJECT, enum = "[]", required = "["books"]", properties = { "books": content.Schema( type = content.Type.ARRAY, items = content.Schema( type = content.Type.OBJECT, properties = { "name": content.Schema( type = content.Type.STRING, ), "writer": content.Schema( type = content.Type.STRING, ), }, ), ), }, ), "response_mime_type": "application/json",}
model = genai.GenerativeModel( model_name="gemini-1.5-flash", generation_config=generation_config, # safety_settings = Adjust safety settings # See https://ai.google.dev/gemini-api/docs/safety-settings system_instruction="Extract the event information.",)
chat_session = model.start_chat( history=[ { "role": "user", "parts": [ "告诉我四大名著分别是什么,以及他们的作者是谁", ], }, { "role": "model", "parts": [ "```json\n{\"books\": [{\"name\": \"红楼梦\", \"writer\": \"曹雪芹\"}, {\"name\": \"三国演义\", \"writer\": \"罗贯中\"}, {\"name\": \"水浒传\", \"writer\": \"施耐庵\"}, {\"name\": \"西游记\", \"writer\": \"吴承恩\"}]} \n```", ], }, ])
response = chat_session.send_message("INSERT_INPUT_HERE")
print(response.text)

得到结果:

{"books": [{"name": "红楼梦", "writer": "曹雪芹"}, {"name": "三国演义", "writer": "罗贯中"}, {"name": "水浒传", "writer": "施耐庵"}, {"name": "西游记", "writer": "吴承恩"}]}


结语

以上,就是本期教程,你学废了吗?


以及...

来看看我做的新开源玩具吧,生成的东西,能直接贴入公众号

智谱 GLM-4-Plus 发布,独家附送免费 API,和我整的新活


相关...

在赛博禅心里回复「智谱API」,可获得无限量使用的智谱 API,9月1日前随便用,还剩最后一天

赛博禅心
拜AI古佛,修赛博禅心
 最新文章