Skip to content

ontocast.agent.select_ontology

Ontology selection agent for OntoCast.

This module provides functionality for selecting appropriate ontologies based on the content of text chunks, ensuring that the chosen ontology best matches the domain and requirements of the text.

select_ontology(state, tools)

Select an appropriate ontology for the current chunk.

This function analyzes the current chunk and selects the most appropriate ontology based on its content and requirements.

Parameters:

Name Type Description Default
state AgentState

The current agent state containing the chunk to process.

 required
tools ToolBox

The toolbox instance providing utility functions.

 required

Returns:

Name Type Description
AgentState AgentState

Updated state with selected ontology.
Source code in ontocast/agent/select_ontology.py 
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
def select_ontology(state: AgentState, tools: ToolBox) -> AgentState:
    """Select an appropriate ontology for the current chunk.

    This function analyzes the current chunk and selects the most appropriate
    ontology based on its content and requirements.

    Args:
        state: The current agent state containing the chunk to process.
        tools: The toolbox instance providing utility functions.

    Returns:
        AgentState: Updated state with selected ontology.
    """
    logger.info("Selecting ontology")
    llm_tool = tools.llm
    om_tool: OntologyManager = tools.ontology_manager

    parser = PydanticOutputParser(pydantic_object=OntologySelectorReport)

    if len(om_tool.ontologies) > 0:
        ontologies_desc = "\n\n".join([o.describe() for o in om_tool.ontologies])
        logger.info(f"Retrieved descriptions for {len(om_tool.ontologies)} ontologies")

        if state.current_chunk is None:
            if state.chunks:
                state.current_chunk = state.chunks.pop(0)
            else:
                state.set_failure(
                    FailureStages.NO_CHUNKS_TO_PROCESS, "No chunks to process"
                )

        excerpt = state.current_chunk.text[:1000] + " ..."

        prompt = PromptTemplate(
            template=template_prompt,
            input_variables=["excerpt", "ontologies_desc", "format_instructions"],
        )

        response = llm_tool(
            prompt.format_prompt(
                excerpt=excerpt,
                ontologies_desc=ontologies_desc,
                format_instructions=parser.get_format_instructions(),
            )
        )
        selector = parser.parse(response.content)
        logger.debug(
            f"Parsed selector report - Selected ontology: {selector.short_name}"
        )

        state.current_ontology = om_tool.get_ontology(selector.short_name)
    else:
        state.current_ontology = NULL_ONTOLOGY
    logger.debug(f"Current ontology set to: {state.current_ontology.short_name}")
    return state