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
860 B
26 lines
860 B
from _arango import arango |
|
from _llm import LLM |
|
db = arango.db |
|
|
|
db.create_collection("rumors_relations", edge=True) |
|
|
|
q = 'for doc in rumors return doc' |
|
rumors = list(db.aql.execute(q)) |
|
persons = list(db.collection("persons").all()) |
|
persons_string = "\n".join([i["name"] for i in persons]) |
|
llm = LLM(chat=False) |
|
|
|
for rumor in rumors: |
|
|
|
heard_from = rumor["heard_from"] |
|
|
|
prompt = f'Nedan är en lista med personer:\n\n{persons_string}\n\nVem av dessa tror du att {heard_from} är?\n\nHär är mer kontext som det hjälper:\n\n{rumor['sexual_summary']} \n\nSvara med namn och hur personen nämns i texten på formen "namn;hur personen nämns\n".' |
|
response = llm.generate(prompt) |
|
|
|
db.collection("rumors_relations").insert( |
|
{ |
|
"_from": person["_id"], |
|
"_to": rumor["_id"], |
|
"context": "rumor", |
|
} |
|
) |