|
|
|
|
|
|
|
|
if conv["reference"]: |
|
|
if conv["reference"]: |
|
|
messages = conv["messages"] |
|
|
messages = conv["messages"] |
|
|
message_num = 0 |
|
|
message_num = 0 |
|
|
chunk_num = 0 |
|
|
|
|
|
while message_num < len(messages): |
|
|
while message_num < len(messages): |
|
|
if message_num != 0 and messages[message_num]["role"] != "user": |
|
|
if message_num != 0 and messages[message_num]["role"] != "user": |
|
|
chunk_list = [] |
|
|
chunk_list = [] |
|
|
if "chunks" in conv["reference"][chunk_num]: |
|
|
|
|
|
chunks = conv["reference"][chunk_num]["chunks"] |
|
|
|
|
|
|
|
|
if "chunks" in conv["reference"][message_num]: |
|
|
|
|
|
chunks = conv["reference"][message_num]["chunks"] |
|
|
for chunk in chunks: |
|
|
for chunk in chunks: |
|
|
new_chunk = { |
|
|
new_chunk = { |
|
|
"id": chunk.get("chunk_id", chunk.get("id")), |
|
|
"id": chunk.get("chunk_id", chunk.get("id")), |
|
|
|
|
|
|
|
|
} |
|
|
} |
|
|
|
|
|
|
|
|
chunk_list.append(new_chunk) |
|
|
chunk_list.append(new_chunk) |
|
|
chunk_num += 1 |
|
|
|
|
|
messages[message_num]["reference"] = chunk_list |
|
|
messages[message_num]["reference"] = chunk_list |
|
|
message_num += 1 |
|
|
message_num += 1 |
|
|
del conv["reference"] |
|
|
del conv["reference"] |