Skip to content

Commit 72d74e5

Browse files
author
Amaad Martin
committed
Audit fixes: export symbols and clean up imports
1 parent 408e495 commit 72d74e5

3 files changed

Lines changed: 10 additions & 11 deletions

File tree

core/src/common.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,10 @@ export type {
209209
ToolInputParameters,
210210
ToolOptions,
211211
} from './tools/function_tool.js';
212+
export {
213+
GoogleSearchAgentTool,
214+
createGoogleSearchAgent,
215+
} from './tools/google_search_agent_tool.js';
212216
export {GOOGLE_SEARCH, GoogleSearchTool} from './tools/google_search_tool.js';
213217
export {
214218
LOAD_ARTIFACTS,

core/src/tools/google_search_agent_tool.ts

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ export function createGoogleSearchAgent(model: string | BaseLlm): LlmAgent {
3535
* A tool that wraps a sub-agent that only uses google_search tool.
3636
*
3737
* This is a workaround to support using google_search tool with other tools.
38-
* @experimental
3938
*/
4039
@experimental
4140
export class GoogleSearchAgentTool extends AgentTool {

core/test/tools/google_search_agent_tool_test.ts

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -5,24 +5,20 @@
55
*/
66

77
import {
8+
BaseLlm,
89
BaseTool,
910
Context,
1011
createEventActions,
12+
createGoogleSearchAgent,
13+
GOOGLE_SEARCH,
14+
GoogleSearchAgentTool,
15+
GoogleSearchTool,
1116
LlmAgent,
17+
LlmRequest,
1218
Runner,
1319
} from '@google/adk';
1420
import {Tool} from '@google/genai';
1521
import {describe, expect, it, vi} from 'vitest';
16-
import {BaseLlm} from '../../src/models/base_llm.js';
17-
import {LlmRequest} from '../../src/models/llm_request.js';
18-
import {
19-
createGoogleSearchAgent,
20-
GoogleSearchAgentTool,
21-
} from '../../src/tools/google_search_agent_tool.js';
22-
import {
23-
GOOGLE_SEARCH,
24-
GoogleSearchTool,
25-
} from '../../src/tools/google_search_tool.js';
2622

2723
vi.mock('../../src/runner/runner.js', async (importOriginal) => {
2824
const actual =

0 commit comments

Comments
 (0)