Google Gemini常用代码

Embedding处理使用的代码

from langchain_google_genai import GoogleGenerativeAIEmbeddings

embeddings = GoogleGenerativeAIEmbeddings(model="models/embedding-001")
vector = embeddings.embed_query("hello, world!")
vector[:5]

推理用代码

from langchain_google_genai import GoogleGenerativeAI
llm = GoogleGenerativeAI(model="models/gemini-pro", transport="rest")
print(
    llm.invoke(
        "What are some of the pros and cons of Python as a programming language?"
    )
)