Extension Points

from cogos import UniversalSchemaDomain

medical = UniversalSchemaDomain("medical_record")
medical.create_field("patient.name", "", "Patient full name")
medical.create_field("diagnosis.current", "", "Current diagnosis")
medical.create_field("medication.current", "", "Current medications")
medical.create_field("medication.allergies", "", "Drug allergies")

cogos = CogOS(config).register(medical)

def my_cm_prompt(state, registry):
    schema_overview = registry.inspect_all(max_depth=3)
    return f"""You are a medical records agent.

Available schemas:
{schema_overview}

Extract: diagnoses, medications, allergies, vitals.
When done: {{"continue": false}}"""

cogos = CogOS(config).cm_prompt(my_cm_prompt).chatbot_prompt(my_template)

my_template = """You are a helpful medical assistant.

Patient records:
{recalled_data}

{rag_section}

Respond based on the patient's records above."""

from cogos.converters import InputConverter

class MyConverter(InputConverter):
    def can_handle(self, data) -> bool:
        return isinstance(data, MyCustomType)

    def convert(self, data) -> str:
        return str(data)