egutierrez
63a9cb5273
Datascience: aggregate_by_group, deduplicate_entities/relations, detect_drift, diff_entities/relations, extract_entities/relations_llm, hotness_score, melt, merge_graphs, pivot, build_entity/relation_schema_prompt. Finance: avellaneda_stoikov_quotes, generate_gbm_prices, generate_taker_order, hawkes_intensity + módulo finance.py. Cybersecurity: envelope_encrypt/decrypt + módulo cybersecurity.py. Pipelines: extraction_pipeline, monte_carlo_market, run_market_sim. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
23 lines
808 B
Python
23 lines
808 B
Python
"""Genera la seccion del system prompt con los tipos de relacion permitidos."""
|
|
|
|
|
|
def build_relation_schema_prompt(relation_types: list[str]) -> str:
|
|
"""Genera texto legible para el LLM describiendo los tipos de relacion permitidos.
|
|
|
|
Formatea la lista de tipos de relacion en una seccion del system prompt que
|
|
indica al LLM que relaciones puede extraer entre entidades.
|
|
|
|
Args:
|
|
relation_types: Lista de strings con los tipos de relacion permitidos.
|
|
Ejemplo: ["funds", "employs", "communicates_with"]
|
|
|
|
Returns:
|
|
String formateado con la seccion del prompt. Retorna string vacio si
|
|
la lista esta vacia.
|
|
"""
|
|
if not relation_types:
|
|
return ""
|
|
|
|
joined = ", ".join(relation_types)
|
|
return f"Allowed relation types:\n{joined}"
|