You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
1.0 KiB
26 lines
1.0 KiB
from _arango import db |
|
from _llm import LLM |
|
from print_color import * |
|
relations = list(db.aql.execute('for doc in relations return doc', count=True)) |
|
|
|
for relation in relations: |
|
interrogation = db.collection('interrogations').get(relation['interrogation']) |
|
if not interrogation: |
|
print_red(relation) |
|
continue |
|
for i in ['to', 'from']: |
|
if 'name' not in interrogation: |
|
interrogation['name'] = interrogation['person'] |
|
db.collection('interrogations').update(interrogation, check_rev=False) |
|
if relation[i] == interrogation['name']: |
|
relation[i] = interrogation['person_id'] |
|
elif relation[i] == interrogation['person_mentioned_as']: |
|
relation[i] = interrogation['person_id'] |
|
for k, v in relation.items(): |
|
print_rainbow(k, v) |
|
print() |
|
for k, v in interrogation.items(): |
|
if k == 'text' or k == 'formatted_text': |
|
continue |
|
print_rainbow(k, v) |
|
db.collection('relations').update(relation, check_rev=False) |