材料驱动图谱构建机制

# 构建用户消息
user_message = self._build_user_message(
    document_texts,
    simulation_requirement,
    additional_context
)

messages = [
    {"role": "system", "content": system_prompt},
    {"role": "user", "content": user_message}
]

result = self.llm_client.chat_json(
    messages=messages,
    temperature=0.3,
    max_tokens=None,
    max_attempts=2,
)

result = self._validate_and_process(result)
return result

entity_names = {e["name"] for e in result["entity_types"]}
has_person = "Person" in entity_names
has_organization = "Organization" in entity_names

fallbacks_to_add = []
if not has_person:
    fallbacks_to_add.append(person_fallback)
if not has_organization:
    fallbacks_to_add.append(organization_fallback)

if fallbacks_to_add:
    current_count = len(result["entity_types"])
    needed_slots = len(fallbacks_to_add)
    if current_count + needed_slots > MAX_ONTOLOGY_TYPES:
        to_remove = current_count + needed_slots - MAX_ONTOLOGY_TYPES
        result["entity_types"] = result["entity_types"][:-to_remove]
    result["entity_types"].extend(fallbacks_to_add)

result["entity_types"] = result["entity_types"][:MAX_ONTOLOGY_TYPES]

if not graph_id:
    raise ValueError("graph_id is required")
self.validate_batch_chunks(chunks, batch_size=batch_size)

total_chunks = len(chunks)
operation_id = self.build_operation_id(graph_id, chunks)
if batch_created_callback:
    batch_created_callback(None, operation_id)

try:
    batch = self.client.batch.create(
        metadata={
            "mirofish_operation_id": operation_id,
            "graph_id": graph_id,
            "chunk_count": total_chunks,
        }
    )
except Exception as error:
    if not is_retryable_zep_error(error):
        raise
    batch = self._find_batch_by_operation_id(graph_id, operation_id)
    if batch is None:
        raise RuntimeError(
            "Zep batch creation is unconfirmed and no matching operation was found"
        ) from error