flt/tools
2
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' of github.com:flt6/tools

Browse Source
This commit is contained in:
flt6
2025-05-06 22:39:55 +08:00
parent 71c3e30267 f692f1ed86
commit dff3bcbd7a
1 changed files with 9 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
conversation_trans/main.py
View File

@ -18,23 +18,17 @@ class Msg:
if msg is None: if msg is None:
self.time=0 self.time=0
return return
print(msg) self.id=msg["message_id"]
self.id=msg["id"] self.par=msg["parent_id"]
if msg["metadata"]["is_visually_hidden_from_conversation"]: self.role=msg["role"]
self.par=0 self.thinking_time=sub(r"\\[\(\)]","$",msg["thinking_elapsed_secs"])
self.time=msg["create_time"] self.thinking_content=sub(r"\\[\[\]]","$$",msg["thinking_content"])
return self.content=msg["content"]
self.par=msg["metadata"]["parent_id"] self.time=msg["inserted_at"]
self.role=msg["author"]["role"]
# self.thinking_time=sub(r"\\[\(\)]","$",msg["thinking_elapsed_secs"])
# self.thinking_content=sub(r"\\[\[\]]","$$",msg["thinking_content"])
self.content=msg["content"]["parts"][0]
self.time=msg["create_time"]
if self.par==None:self.par=0 if self.par==None:self.par=0
inp = data["messages"] inp = data["data"]["biz_data"]["chat_messages"]
# inp = data["data"]["biz_data"]["chat_messages"]
msgs:list[Msg]=[Msg(None)] msgs:list[Msg]=[Msg(None)]
child:list[list[Msg]]=[[] for _ in range(len(inp)+1)] child:list[list[Msg]]=[[] for _ in range(len(inp)+1)]
@ -63,8 +57,7 @@ if not dfs(0):
raise RuntimeError("Cannot find latest node in tree.") raise RuntimeError("Cannot find latest node in tree.")
path.pop(0) path.pop(0)
title = "Conversation" title = data['data']['biz_data']['chat_session']['title']
# title = data['data']['biz_data']['chat_session']['title']
out=[f"# {title}"] out=[f"# {title}"]
for i in path: for i in path:
resp=[] resp=[]