We're excited to announce that instructor
now supports structured outputs using tool calling for both the Gemini SDK and the VertexAI SDK.
A special shoutout to Sonal for his contributions to the Gemini Tool Calling support.
Let's walk through a simple example of how to use these new features
To get started, install the latest version of instructor
. Depending on whether you're using Gemini or VertexAI, you should install the following:
This ensures that you have the necessary dependencies to use the Gemini or VertexAI SDKs with instructor.
We recommend using the Gemini SDK over the VertexAI SDK for two main reasons.
- Compared to the VertexAI SDK, the Gemini SDK comes with a free daily quota of 1.5 billion tokens to use for developers.
- The Gemini SDK is significantly easier to setup, all you need is a
GOOGLE_API_KEY
that you can generate in your GCP console. THe VertexAI SDK on the other hand requires a credentials.json file or an OAuth integration to use.
With our provider agnostic API, you can use the same interface to interact with both SDKs, the only thing that changes here is how we initialise the client itself.
Before running the following code, you'll need to make sure that you have your Gemini API Key set in your shell under the alias GOOGLE_API_KEY
.
import instructor
import google.generativeai as genai
from pydantic import BaseModel
class User(BaseModel):
name: str
age: int
client = instructor.from_gemini(
client=genai.GenerativeModel(
model_name="models/gemini-1.5-flash-latest", # (1)!
)
)
resp = client.chat.completions.create(
messages=[
{
"role": "user",
"content": "Extract Jason is 25 years old.",
}
],
response_model=User,
)
print(resp)
#> name='Jason' age=25
- Current Gemini models that support tool calling are
gemini-1.5-flash-latest
and gemini-1.5-pro-latest
.
We can achieve a similar thing with the VertexAI SDK. For this to work, you'll need to authenticate to VertexAI.
There are some instructions here but the easiest way I found was to simply download the GCloud cli and run gcloud auth application-default login
.
import instructor
import vertexai # type: ignore
from vertexai.generative_models import GenerativeModel # type: ignore
from pydantic import BaseModel
vertexai.init()
class User(BaseModel):
name: str
age: int
client = instructor.from_vertexai(
client=GenerativeModel("gemini-1.5-pro-preview-0409"), # (1)!
)
resp = client.chat.completions.create(
messages=[
{
"role": "user",
"content": "Extract Jason is 25 years old.",
}
],
response_model=User,
)
print(resp)
#> name='Jason' age=25
- Current Gemini models that support tool calling are
gemini-1.5-flash-latest
and gemini-1.5-pro-latest
.