Skip to content
Last updated

Creating a Knowledge Base

A Knowledge Base is a tool that gives the reference data to Agent, which is critical capability for agent to generate more accurate and desired output. AI Agent Foundry offers 2 types of Knowledge Base, one is Database and the other is Text. Follow the instruction below to create a new Knowledge Base.

  1. Open TD Console.
  2. Navigate to AI Agent Foundry.
  3. Select Knowledge Base and then select Create knowledge base.

Create a Database Knowledge Base

Enter Knowledge Base Details​

  1. Enter a Knowledge Base name specifying how it is used
  2. Select Database type, then select a database.

Database knowledge base configuration form

Enter Table Information

  1. Click Add table button to define the table information

  2. Complete the form.

FieldDescription
NameEnter a name for your knowledge base.
TypeSelect Database.
TD Database NameEnter the Treasure Data database where queries run.
The database must exist in Plazma DB.
TableDefine the table that the knowledge base supports.
TD QueryProvide the Trino SQL that defines the table using the selected database.
Exclude Columns: list every column you want to keep.
Exclude Records: filter records with a WHERE clause.
NameEnter the table name to reference.
+ Add tableSelect to add more tables to your knowledge base.
  1. Select Save.

After you select Save , the data loading process starts. Before using the knowledge base, wait until it has a status of Success.

Creating Text Knowledge Base

Enter Knowledge Base Details​

  1. Enter a Knowledge Base name specifying how it is used
  2. Select Text type, then complete the form
    Text knowledge base form
FieldDescription
NameEnter a name for your knowledge base.
TypeSelect Text.
Text-inputEnter up to 18,000 characters of reference text.

Creating Parent Segment Knowledge Base

Parent Segment Knowledge Base is automatically created when the "LLM enabled" switch is enabled on the Parent Segment Config. Currently, there is no way to create on the Custom LLM project.

Next Step

Check your Knowledge Base setup through query simulations.