Google VertexAI API 接入

import vertexai

import os

#此步非常重要，否则无法访问，去GCP创建服务账号密钥。

os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = "服务账号json格式key"

from vertexai.generative_models import GenerativeModel, Part
# TODO (developer): Update and un-comment below lines
# project_id = "PROJECT_ID"
vertexai.init(project="GCP上创建的项目ID", location="us-central1")
model = GenerativeModel(model_name="gemini-1.5-pro-preview-0409")
prompt = """
请提取文档的目录结构。
"""
pdf_file_uri = 'gs://jacky_bucket/H3.pdf'#"gs://cloud-samples-data/generative-ai/pdf/2403.05530.pdf"
pdf_file = Part.from_uri(pdf_file_uri, mime_type="application/pdf")
contents = [pdf_file, prompt]
response = model.generate_content(contents)
print(response.text)

返回结果：

```json
[{"level": 1, "text": "核心观点"}, {"level": 1, "text": "一、 福昕软件：PDF 办公软件领军企业"}, {"level": 2, "text": "1.1 发展历程：公司注册时间短，聚焦软件服务"}, {"level": 2, "text": "1.2 公司业务分类：聚焦 PDF 版式软件与服务"}, {"level": 3, "text": "a. PDF 编辑与处理业务"}, {"level": 3, "text": "b. PDF 工具及转换引擎"}, {"level": 3, "text": "c. 软件开发平台与服务"}, {"level": 3, "text": "d. 企业文档协作解决方案"}, {"level": 3, "text": "e. PDF 行业应用解决方案"}, {"level": 1, "text": "二、 所处行业：行业龙头稳居，公司持续发力"}, {"level": 2, "text": "2.1 行业发展：文档电子化办公已成必然，PDF 获得编辑与协作等新功能"}, {"level": 2, "text": "2.2 受益公司分析：公司产品矩阵成熟，用户粘性高，具有较强规模效应"}, {"level": 3, "text": "2.2.1 据守 PDF 软件市场，拓展海外业务是公司发展重点"}, {"level": 3, "text": "2.2.2 福昕 PDF Editor 收获忠实用户好评"}, {"level": 1, "text": "三、“订阅+渠道”双轮驱动，稳健经营"}, {"level": 2, "text": "3.1 订阅模式：公司积极进行产品及渠道改革，公司业绩展现发展韧性"}, {"level": 3, "text": "3.1.1 订阅模式下，公司自身经营展现韧性"}, {"level": 3, "text": "3.1.2 公司经营优势明显，财务状况良好"}, {"level": 2, "text": "3.2 渠道策略：公司不断拓展线下销售渠道，加速拓展全球市场"}, {"level": 3, "text": "3.2.1 线上线下多渠道并举，拓展全球市场"}, {"level": 3, "text": "3.2.2 从价格优势到服务优势，最终形成品牌优势"}, {"level": 1, "text": "四、海外内需：海外行业稳健延展，信创行业市场空间广阔"}, {"level": 2, "text": "4.1 海外市场：全球 PDF 软件市场稳步发展，公司积极拓展海外业务"}, {"level": 3, "text": "4.1.1 海外建筑行业解决方案"}, {"level": 3, "text": "4.1.2 海外律师网络审查解决方案"}, {"level": 3, "text": "4.1.3 电子签名解决方案"}, {"level": 2, "text": "4.2 行业信创：积极拓展行业应用"}, {"level": 1, "text": "五、 AIGC 赋能佐证行情，产学研助力稳步发展"}, {"level": 2, "text": "5.1 Foxit PDF Editor  AI 功能上线，佐证 ChatGPT 热潮"}, {"level": 3, "text": "5.1.1 文档摘要：对输入文本进行简单而准确的概括"}, {"level": 3, "text": "5.1.2 文档问答：根据上下文回答用户问题"}, {"level": 3, "text": "5.1.3 文档改写：改写文档，并保持文本基本意思"}, {"level": 3, "text": "5.1.4 文档翻译：支持多种语言的文档翻译"}, {"level": 3, "text": "5.1.5 文档分析：利用专业领域模型分析文档"}, {"level": 3, "text": "5.1.6 内容安全检测：用户可快速找到文档中的敏感词组，同时，AI 助手的参与也减少了用户的负担"}, {"level": 2, "text": "5.2  嵌入 ChatGPT，Foxit PDF Editor Cloud 功能实用性显著提升"}, {"level": 1, "text": "六、 盈利预测、估值与投资评级"}, {"level": 2, "text": "6.1 盈利预测"}, {"level": 2, "text": "6.2 估值与投资评级"}, {"level": 1, "text": "风险因素"}, {"level": 1, "text": "附录"}, {"level": 1, "text": "分析团队简介"}, {"level": 1, "text": "分析师声明"}, {"level": 1, "text": "投资评级的说明"}, {"level": 1, "text": "风险提示"}]
```