⚠️ google-generative-ai has become deprecated. The same goes for this library. Please use FunctionCalling-FirebaseVertexAI
[Deprecated] Google AI Swift SDK for the Gemini API
With Gemini 2.0, we took the chance to create a unified SDK for mobile developers who want to use Google's GenAI models (Gemini, Veo, Imagen, etc). As part of that process, we took all of the feedback from this SDK and what developers like about other SDKs in the ecosystem to direclty work with the Firebase SDK. We don't plan to add anything to this SDK or making any further changes. We know how disruptive an SDK change can be and don't take this change lightly, but our goal is to create an extremely simple and clear path for developers to build with our models so it felt necessary to make this change.
This library simplifies the integration of the FunctionCalling macro into GoogleGenerativeAI. By using this library, you can directly generate Tool
objects from Swift native functions, which can then be specified as FunctionCalling when invoking VertexAI.
import FunctionCalling
import FunctionCalling_GoogleGenerativeAI
import GoogleGenerativeAI
// MARK: Declare the container and functions for the tools to be called from FunctionCalling.
@FunctionCalling(service: .llamaOrGemini)
struct MyFunctionTools {
@CallableFunction
/// Get the current stock price for a given ticker symbol
///
/// - Parameter: The stock ticker symbol, e.g. GOOG for Google Inc.
func getStockPrice(ticker: String) async throws -> String {
// code to return stock price of passed ticker
}
}
// MARK: You can directly pass the tools generated from objects to the model in GoogleGenerativeAI.
let model = GenerativeModel(
modelName: "gemini-1.5-flash",
tools: MyFunctionTools().googleGenerativeAITools
)
let package = Package(
name: "MyPackage",
products: [...],
targets: [
.target(
"YouAppModule",
dependencies: [
.product(name: "FunctionCalling-GoogleGenerativeAI", package: "FunctionCalling-GoogleGenerativeAI")
]
)
],
dependencies: [
.package(url: "https://github.com/FunctionCalling/FunctionCalling-GoogleGenerativeAI", from: "0.1.0")
]
)
Pull requests are welcome. For major changes, please open an issue first to discuss what you would like to change.Please make sure to update tests as appropriate.
The MIT License