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