Google ADK

from cintegrity import secure_agent
from google.adk.agents.llm_agent import Agent
from google.adk.runners import Runner
from google.adk.sessions import InMemorySessionService
from google.genai import types

# Wrap your tools
tools, system_prompt = secure_agent.google_adk(
    tools=[read_inbox, send_email]
)

# Create agent
agent = Agent(
    model="gemini-2.0-flash",
    name="secure_assistant",
    instruction=system_prompt,
    tools=tools,
)

# Run
session_service = InMemorySessionService()
session_service.create_session(app_name="demo", user_id="u1", session_id="s1")

runner = Runner(agent=agent, app_name="demo", session_service=session_service)
events = runner.run(
    user_id="u1",
    session_id="s1",
    new_message=types.Content(
        role="user",
        parts=[types.Part(text="Summarize my inbox")]
    )
)

from typing import TypedDict

class SearchDocsArgs(TypedDict):
    query: str

def search_docs(args: SearchDocsArgs) -> dict:
    """Search internal documentation."""
    return {"results": ["Doc 1...", "Doc 2..."]}