mirror of
https://github.com/googleapis/genai-toolbox.git
synced 2026-01-09 07:28:05 -05:00
1e67810740
## Description ## PR Checklist > Thank you for opening a Pull Request! Before submitting your PR, there are a > few things you can do to make sure it goes smoothly: - [x] Make sure you reviewed [CONTRIBUTING.md](https://github.com/googleapis/genai-toolbox/blob/main/CONTRIBUTING.md) - [ ] Make sure to open an issue as a [bug/issue](https://github.com/googleapis/genai-toolbox/issues/new/choose) before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea - [x] Ensure the tests and linter pass - [ ] Code coverage does not decrease (if any source code was changed) - [x] Appropriate docs were updated (if necessary) - [ ] Make sure to add `!` if this involve a breaking change 🛠️ Fixes #<issue_number_goes_here>
56 lines
2.1 KiB
JavaScript
56 lines
2.1 KiB
JavaScript
import { InMemoryRunner, LlmAgent, LogLevel } from '@google/adk';
|
|
import { ToolboxClient } from '@toolbox-sdk/adk';
|
|
|
|
const prompt = `
|
|
You're a helpful hotel assistant. You handle hotel searching, booking, and
|
|
cancellations. When the user searches for a hotel, mention its name, id,
|
|
location and price tier. Always mention hotel ids while performing any
|
|
searches. This is very important for any operations. For any bookings or
|
|
cancellations, please provide the appropriate confirmation. Be sure to
|
|
update checkin or checkout dates if mentioned by the user.
|
|
Don't ask for confirmations from the user.
|
|
`;
|
|
|
|
const queries = [
|
|
"Find hotels with Basel in its name.",
|
|
"Can you book the Hilton Basel for me?",
|
|
"Oh wait, this is too expensive. Please cancel it and book the Hyatt Regency instead.",
|
|
"My check in dates would be from April 10, 2024 to April 19, 2024.",
|
|
];
|
|
|
|
process.env.GOOGLE_GENAI_API_KEY = process.env.GOOGLE_API_KEY || 'your-api-key'; // Replace it with your API key
|
|
|
|
export async function main() {
|
|
const userId = 'test_user';
|
|
const client = new ToolboxClient('http://127.0.0.1:5000');
|
|
const tools = await client.loadToolset("my-toolset");
|
|
|
|
const rootAgent = new LlmAgent({
|
|
name: 'hotel_agent',
|
|
model: 'gemini-2.5-flash',
|
|
description: 'Agent for hotel bookings and administration.',
|
|
instruction: prompt,
|
|
tools: tools,
|
|
});
|
|
|
|
const appName = rootAgent.name;
|
|
const runner = new InMemoryRunner({ agent: rootAgent, appName, logLevel: LogLevel.ERROR, });
|
|
const session = await runner.sessionService.createSession({ appName, userId });
|
|
|
|
for (const query of queries) {
|
|
await runPrompt(runner, userId, session.id, query);
|
|
}
|
|
}
|
|
|
|
async function runPrompt(runner, userId, sessionId, prompt) {
|
|
const content = { role: 'user', parts: [{ text: prompt }] };
|
|
const stream = runner.runAsync({ userId, sessionId, newMessage: content });
|
|
const responses = await Array.fromAsync(stream);
|
|
const accumulatedResponse = responses
|
|
.flatMap((e) => e.content?.parts?.map((p) => p.text) ?? [])
|
|
.join('');
|
|
|
|
console.log(`\nMODEL RESPONSE: ${accumulatedResponse}\n`);
|
|
}
|
|
|
|
main(); |