dependencies {
    implementation(libs.koog.firebase)
}

dependencies {
    implementation("dev.ynagai.koog:koog-firebase:0.8.0")
}

import ai.koog.agents.core.agent.AIAgent
import dev.ynagai.koog.firebase.FirebaseModels
import dev.ynagai.koog.firebase.simpleFirebaseExecutor

val agent = AIAgent(
    promptExecutor = simpleFirebaseExecutor(),
    systemPrompt = "You are a helpful assistant.",
    llmModel = FirebaseModels.Gemini3_7Flash
)

val result = agent.run("Hello!")

import dev.ynagai.firebase.Firebase
import dev.ynagai.firebase.ai.GenerativeBackend
import dev.ynagai.koog.firebase.simpleFirebaseExecutor

// Use Vertex AI backend.
// Gemini 3.x models are only available in the "global" location; vertexAI()
// defaults to "us-central1", which will 404 for those models.
val executor = simpleFirebaseExecutor(
    app = Firebase.app,
    backend = GenerativeBackend.vertexAI("global")
)

import ai.koog.prompt.dsl.prompt
import dev.ynagai.firebase.ai.Tool
import dev.ynagai.koog.firebase.FirebaseLLMParams
import dev.ynagai.koog.firebase.FirebaseMetadataKeys
import kotlinx.serialization.json.jsonArray
import kotlinx.serialization.json.jsonObject
import kotlinx.serialization.json.jsonPrimitive

val prompt = prompt("search", params = FirebaseLLMParams(
    builtInTools = listOf(Tool.googleSearch(), Tool.urlContext()),
)) {
    user("What changed in the latest Kotlin release? See https://kotlinlang.org/docs/whatsnew.html")
}

val response = executor.execute(prompt, FirebaseModels.Gemini3_7Flash).first()

// Grounding sources and Search Suggestions are exposed in the response metadata.
val grounding = response.metaInfo.metadata?.get(FirebaseMetadataKeys.GROUNDING_METADATA)?.jsonObject
val sources = grounding?.get("groundingChunks")?.jsonArray
    ?.mapNotNull { it.jsonObject["web"]?.jsonObject?.get("uri")?.jsonPrimitive?.content }
val searchSuggestionsHtml = grounding?.get("searchEntryPoint")?.jsonObject
    ?.get("renderedContent")?.jsonPrimitive?.content

import dev.ynagai.koog.firebase.tools.GoogleSearchTool
import dev.ynagai.koog.firebase.tools.UrlContextTool
import kotlinx.serialization.json.jsonObject
import kotlinx.serialization.json.jsonPrimitive

val toolRegistry = ToolRegistry {
    tool(GoogleSearchTool(executor, FirebaseModels.Gemini3_7Flash, onGroundingMetadata = { grounding ->
        // Google's terms require showing the Search Suggestions when Search grounding is used.
        // The callback runs on the tool's coroutine, not the main thread.
        val html = grounding["searchEntryPoint"]?.jsonObject?.get("renderedContent")?.jsonPrimitive?.content
        html?.let { showSearchSuggestions(it) }
    }))
    tool(UrlContextTool(executor, FirebaseModels.Gemini3_7Flash))
    tool(MyOwnTool())
}