How to Create Custom Agent Tool Types

Note

The functionality described on this page is only available to customers who have licensed Veeva AI.

To create a custom agent tool type with Vault Java SDK:

1. Create a Tool Type Runtime Handler by implementing AIToolTypeHandler:

   Copy to clipboard

   @ExecuteAs(ExecuteAsUser.REQUEST_OWNER) @AiContextTypeRuntimeHandlerInfo(scope = AiScope.DOCUMENT) public class MyContextTypeRuntimeHandler implements AiContextTypeRuntimeHandler{ @Override public AiContextTypeResolveResponse onResolve(AiContextTypeResolveContext context){ AiDocumentScopeSource documentScopeSource = context.getScopeSource(AiScopeType.DOCUMENT); DocumentVersion documentVersion = documentScopeSource.getDocumentVersion(); String documentName = documentVersion.getValue("name__v", ValueType.STRING); AiContextTypeOutputItem contextOutput = context.newOutputItemBuilder() .withText("The current document name is [" + documentName + "]") .build(); return context.newSuccessResponseBuilder() .appendOutputItem(contextOutput) .build(); } }

2. Create an Aitooltyperuntimecode record referencing the AIToolTypeHandler, and create Aitooltyperuntimecode MDL records:

   Copy to clipboard

   RECREATE Aitooltype my_custom_tool_type__c ( label('My Tool Type'), admin_configurable(true), description ('Sample tool to create a Tool Invocations record'), runtime_code('Aitooltyperuntimecode.com.veeva.vault.custom.context.MyToolTypeRuntimeHandler') );