用Python调用google的Gemini API
以下是Gemini自己给出的具体步骤和代码,我直接参考一次测试通过,大家可以参考。
Goodmao:Python call gemini API
Gemini:
Absolutely, calling the Gemini API with Python involves a few steps:
Prerequisites:
-
API Key: You'll need an API key to access Gemini. You can create one easily in Google AI Studio: [invalid URL removed].
-
Python library: Install the
google-generativeai
library using pip:Bash
pip install -U -q google-generativeai
请谨慎使用代码。
Calling the API:
Here's a basic structure for calling the Gemini API for text generation:
Python
import googleapiclient.discovery as discovery
# Replace with your API key obtained from AI Studio
API_KEY = "YOUR_API_KEY"
# Define the project ID where your Gemini model resides
PROJECT_ID = "YOUR_PROJECT_ID"
# Discovery URL for the Gemini API
service = discovery.build(
"generativeai", "v1beta1", developerKey=API_KEY
)
# Text prompt for the model
prompt = "Write a poem about nature."
# Define the model you want to use (replace with the actual model name)
model = "your_model_name"
# Send the request with prompt and model details
body = {"prompt": prompt, "model": model}
response = service.projects().locations().models().generateText(
parent=f"projects/{PROJECT_ID}/locations/global/models/{model}", body=body
)
# Access the generated text from the response
generated_text = response["text"]
print(generated_text)
请谨慎使用代码。
Additional functionalities:
- List Models: Use the
service.projects().locations().models().list()
method to explore available models. - Chat Conversation: Initiate a chat session with
chat = model.start_chat()
and interact with the model turn-by-turn. - Function Calling: Enable function calling during chat by setting
enable_automatic_function_calling=True
instart_chat()
.
Resources:
- Get started with Gemini API (Python): https://ai.google.dev/gemini-api/docs/get-started/python
- Gemini API: Function calling with Python: https://ai.google.dev/tutorials/function_calling_python_quickstart
= = =