# Entity + Relation extraction with structured output

SYSTEM_PROMPT = """
You are a knowledge graph construction agent.
Given a text passage, extract ALL entities and relations.

Return ONLY valid JSON matching this schema:
{
  "entities": [
    { "id": "e1", "text": "Alan Turing", "type": "PERSON",
      "canonical": "Q7251", "confidence": 0.97 }
  ],
  "relations": [
    { "subject": "e1", "predicate": "works_at",
      "object": "e2", "confidence": 0.91 }
  ]
}
"""

async def extract_graph_from_chunk(chunk: str) -> dict:
    response = await llm.complete(
        system=SYSTEM_PROMPT,
        user=f"Extract entities and relations:\n\n{chunk}",
        temperature=0.0,    # deterministic
        response_format="json_object"
    )
    data = json.loads(response.text)
    return resolve_and_upsert(data)   # dedup + write to graph