Using Knowledge Bases

1
from smallestai.atoms.kb import KB
2
from smallestai.atoms import AtomsClient
3
4
# Initialize separate managers
5
kb = KB()
6
client = AtomsClient()
7
8
# Create KB
9
response = kb.create(
10
    name="Product Documentation",
11
    description="Technical specs and troubleshooting guides"
12
)
13
14
kb_id = response["data"]

1
details = kb.get(kb_id)

1
all_kbs = kb.list()
2
3
for item in all_kbs["data"]:
4
    print(f"{item['name']}: {item['_id']}")

1
agent = client.new_agent(
2
    name="Support Agent",
3
    prompt="You are a helpful support agent. Use your knowledge base to answer product questions accurately.",
4
    description="Agent with product KB",
5
    kb_id=kb_id
6
)
7
8
agent_id = agent.data

1
agent_details = client.get_agent_by_id(id=agent_id)
2
3
linked_kb = agent_details["data"]["globalKnowledgeBaseId"]
4
print(f"Agent is linked to KB: {linked_kb}")

1
# First, remove the agent
2
client.delete_agent(id=agent_id)
3
4
# Then delete the KB
5
kb.delete(kb_id)