Skip to content

ontocast.agent.select_ontology_catalog

LLM pick of a single catalog ontology for a text excerpt (full-TTL per-unit path).

select_catalog_ontology_for_excerpt(ontology_manager, llm_tool, excerpt, ontology_selection_user_instruction='') async

Use the LLM to select one catalog ontology, or :data:NULL_ONTOLOGY if none fit.

The excerpt is usually a content unit's text. Empty excerpt or an empty catalog yields NULL_ONTOLOGY without calling the model.

Source code in ontocast/agent/select_ontology_catalog.py 
19
20
21
22
23
24
25
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
async def select_catalog_ontology_for_excerpt(
    ontology_manager: OntologyManager,
    llm_tool: LLMTool,
    excerpt: str,
    ontology_selection_user_instruction: str = "",
) -> Ontology:
    """Use the LLM to select one catalog ontology, or :data:`NULL_ONTOLOGY` if none fit.

    The excerpt is usually a content unit's text. Empty excerpt or an empty
    catalog yields ``NULL_ONTOLOGY`` without calling the model.
    """
    text = excerpt.strip()
    if not text or not ontology_manager.has_ontologies:
        return NULL_ONTOLOGY

    ontologies = ontology_manager.ontologies
    num_ontologies = len(ontologies)
    if num_ontologies == 0:
        return NULL_ONTOLOGY

    lines: list[str] = []
    for i, o in enumerate(ontologies, start=1):
        lines.append(f"{i}. {o.describe()}")
    ontologies_list = "\n\n".join(lines)
    none_index = num_ontologies + 1

    model_cls = create_ontology_selector_report_model(num_ontologies)
    parser = PydanticOutputParser(pydantic_object=model_cls)
    prompt = PromptTemplate(
        template=template_prompt,
        input_variables=[
            "excerpt",
            "ontologies_list",
            "num_ontologies",
            "none_index",
            "ontology_selection_user_instruction",
            "format_instructions",
        ],
    )

    selector = await call_llm_with_retry(
        llm_tool=llm_tool,
        prompt=prompt,
        parser=parser,
        prompt_kwargs={
            "excerpt": text,
            "ontologies_list": ontologies_list,
            "num_ontologies": num_ontologies,
            "none_index": none_index,
            "ontology_selection_user_instruction": ontology_selection_user_instruction.strip(),
            "format_instructions": parser.get_format_instructions(),
        },
    )

    idx = selector.answer_index
    if idx == none_index:
        logger.debug("LLM selected: no suitable catalog ontology (none index)")
        return NULL_ONTOLOGY
    if 1 <= idx <= num_ontologies:
        return ontologies[idx - 1]
    logger.warning("Invalid answer_index %s from selector; using NULL_ONTOLOGY", idx)
    return NULL_ONTOLOGY