11 releases
0.2.4 | Mar 18, 2024 |
---|---|
0.2.3 | Feb 11, 2024 |
0.1.7 | Jan 18, 2024 |
0.1.4 | Dec 30, 2023 |
#148 in Machine learning
266 downloads per month
46KB
876 lines
Google Generative AI API client (unofficial)
An unofficial rust-based client library to interact with the Google generative AI API.
The goal is to emulate the Google AI Python SDK but in Rust.
The initial focus will be on the newer Gemini models, but the more stable and mature models will hopefully also be supported soon.
⚠️ Warning: initial working out; it will change very frequently. ⚠️
Usage
Start point, gain familiarity with Google's Gemini generative AI.
-
For the public Gemini endpoint, see the Gemini API Overview docs
-
Similarly, for the Vertex AI endpoint, see the Vertex AI Gemini API docs
See examples and follow the in-comment instructions. The code is (hopefully) easy and readable.
Contributing
Yes, please!! Create a fork and branch, make your contribution, and raise a PR.
Please see contributing for the rules; they are standard though.
Work status
I do my best to release working code.
Status today is: "Happy path for both public and Vertex AI endpoints work for Gemini, but some things may be fragile."
Outline tasks
- Create request and response structs
- Create the public API happy path for Gemini
- Create the Vertex AI (private) API happy path for Gemini
- Create basic error handling
- get - see: "https://ai.google.dev/tutorials/rest_quickstart#get_model" and "https://ai.google.dev/tutorials/rest_quickstart#list_models"
- countTokens - see: "https://ai.google.dev/tutorials/rest_quickstart#count_tokens"
- function - see "https://cloud.google.com/vertex-ai/docs/generative-ai/multimodal/function-calling"
- embedContent - see: "https://ai.google.dev/tutorials/rest_quickstart#embedding"
Dependencies
~19–35MB
~672K SLoC