@brandon_9527/tcode 1.0.9 → 1.0.11

package/dist/python-src/src/middlewares/summary.py

@@ -27,7 +27,7 @@ ContextFraction=tuple[Literal[_D],float]
 ContextTokens=tuple[Literal[_E],int]
 ContextMessages=tuple[Literal[_B],int]
 ContextSize=ContextFraction|ContextTokens|ContextMessages
-def dk(model):
+def de(model):
 	if model._llm_type=='anthropic-chat':return partial(count_tokens_approximately,chars_per_token=3.3)
 	return count_tokens_approximately
 class SummaryState(AgentState):compact:NotRequired[bool]=_F
@@ -44,34 +44,34 @@ class SummarizationMiddleware(AgentMiddleware):
 		if isinstance(C,str):C=init_chat_model(C)
 		A.model=C
 		if B is _A:A.trigger=_A;G=[]
-		elif isinstance(B,list):I=[A.df(B,N)for B in B];A.trigger=I;G=I
-		else:J=A.df(B,N);A.trigger=J;G=[J]
-		A._trigger_conditions=G;A.keep=A.df(F,'keep')
-		if H is count_tokens_approximately:A.token_counter=dk(A.model)
+		elif isinstance(B,list):I=[A.da(B,N)for B in B];A.trigger=I;G=I
+		else:J=A.da(B,N);A.trigger=J;G=[J]
+		A._trigger_conditions=G;A.keep=A.da(F,'keep')
+		if H is count_tokens_approximately:A.token_counter=de(A.model)
 		else:A.token_counter=H
 		A.summary_prompt=summary_prompt;A.trim_tokens_to_summarize=trim_tokens_to_summarize;K=any(A[0]==_D for A in A._trigger_conditions)
 		if A.keep[0]==_D:K=_C
-		if K and A.dn()is _A:O='Model profile information is required to use fractional token limits, and is unavailable for the specified model. Please use absolute token counts instead, or pass `\n\nChatModel(..., profile={"max_input_tokens": ...})`.\n\nwith a desired integer value of the model\'s maximum input tokens.';raise ValueError(O)
-	def do(B,state):A=state.get(_G,_F);return bool(A)
+		if K and A.dk()is _A:O='Model profile information is required to use fractional token limits, and is unavailable for the specified model. Please use absolute token counts instead, or pass `\n\nChatModel(..., profile={"max_input_tokens": ...})`.\n\nwith a desired integer value of the model\'s maximum input tokens.';raise ValueError(O)
+	def dh(B,state):A=state.get(_G,_F);return bool(A)
 	@override
 	def before_model(self,state,runtime):
-		C=state;A=self;B=C[_B];A.da(B);E=A.token_counter(B);F=A.do(C);G=A.db(B,E)
+		C=state;A=self;B=C[_B];A.dc(B);E=A.token_counter(B);F=A.dh(C);G=A.do(B,E)
 		if not G and not F:return
-		D=A.dh(B)
+		D=A.dg(B)
 		if D<=0:return
-		H,I=A.di(B,D);J=A.de(H);K=A.dg(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
+		H,I=A.df(B,D);J=A.dm(H);K=A.dn(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
 	@override
 	async def abefore_model(self,state,runtime):
-		C=state;A=self;B=C[_B];A.da(B);E=A.token_counter(B);F=A.do(C);G=A.db(B,E)
+		C=state;A=self;B=C[_B];A.dc(B);E=A.token_counter(B);F=A.dh(C);G=A.do(B,E)
 		if not G and not F:return
-		D=A.dh(B)
+		D=A.dg(B)
 		if D<=0:return
-		H,I=A.di(B,D);J=await A._acreate_summary(H);K=A.dg(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
+		H,I=A.df(B,D);J=await A._acreate_summary(H);K=A.dn(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
 	def dj(B,messages,threshold):
 		A=next((A for A in reversed(messages)if isinstance(A,AIMessage)),_A)
 		if isinstance(A,AIMessage)and A.usage_metadata is not _A and(C:=A.usage_metadata.get('total_tokens',-1))and C>=threshold and(D:=A.response_metadata.get('model_provider'))and D==B.model._get_ls_params().get('ls_provider'):return _C
 		return _F
-	def db(A,messages,total_tokens):
+	def do(A,messages,total_tokens):
 		F=total_tokens;E=messages
 		if not A._trigger_conditions:return _F
 		for(B,C)in A._trigger_conditions:
@@ -79,26 +79,26 @@ class SummarizationMiddleware(AgentMiddleware):
 			if B==_E and F>=C:return _C
 			if B==_E and A.dj(E,C):return _C
 			if B==_D:
-				G=A.dn()
+				G=A.dk()
 				if G is _A:continue
 				D=int(G*C)
 				if D<=0:D=1
 				if F>=D:return _C
 				if A.dj(E,D):return _C
 		return _F
-	def dh(A,messages):
+	def dg(A,messages):
 		B=messages;D,E=A.keep
 		if D in{_E,_D}:
-			C=A.dd(B)
+			C=A.db(B)
 			if C is not _A:return C
 			return A.dl(B,_DEFAULT_MESSAGES_TO_KEEP)
 		return A.dl(B,cast('int',E))
-	def dd(C,messages):
+	def db(C,messages):
 		A=messages
 		if not A:return 0
 		H,I=C.keep
 		if H==_D:
-			J=C.dn()
+			J=C.dk()
 			if J is _A:return
 			F=int(J*I)
 		elif H==_E:F=int(I)
@@ -114,8 +114,8 @@ class SummarizationMiddleware(AgentMiddleware):
 		if B>=len(A):
 			if len(A)==1:return 0
 			B=len(A)-1
-		return C.dc(A,B)
-	def dn(C):
+		return C.dd(A,B)
+	def dk(C):
 		try:A=C.model.profile
 		except AttributeError:return
 		if not isinstance(A,Mapping):return
@@ -123,7 +123,7 @@ class SummarizationMiddleware(AgentMiddleware):
 		if not isinstance(B,int):return
 		return B
 	@staticmethod
-	def df(context,parameter_name):
+	def da(context,parameter_name):
 		E=context;C=parameter_name;D,B=E
 		if D==_D:
 			if not 0<B<=1:A=f"Fractional {C} values must be between 0 and 1, got {B}.";raise ValueError(A)
@@ -132,19 +132,19 @@ class SummarizationMiddleware(AgentMiddleware):
 		else:A=f"Unsupported context size type {D} for {C}.";raise ValueError(A)
 		return E
 	@staticmethod
-	def dg(summary):return[HumanMessage(content=f"Here is a summary of the conversation to date:\n\n{summary}",additional_kwargs={'lc_source':'summarization'})]
+	def dn(summary):return[HumanMessage(content=f"Here is a summary of the conversation to date:\n\n{summary}",additional_kwargs={'lc_source':'summarization'})]
 	@staticmethod
-	def da(messages):
+	def dc(messages):
 		for A in messages:
 			if A.id is _A:A.id=str(uuid.uuid4())
 	@staticmethod
-	def di(conversation_messages,cutoff_index):B=cutoff_index;A=conversation_messages;C=A[:B];D=A[B:];return C,D
+	def df(conversation_messages,cutoff_index):B=cutoff_index;A=conversation_messages;C=A[:B];D=A[B:];return C,D
 	def dl(C,messages,messages_to_keep):
 		B=messages_to_keep;A=messages
 		if len(A)<=B:return 0
-		D=len(A)-B;return C.dc(A,D)
+		D=len(A)-B;return C.dd(A,D)
 	@staticmethod
-	def dc(messages,cutoff_index):
+	def dd(messages,cutoff_index):
 		B=cutoff_index;A=messages
 		if B>=len(A)or not isinstance(A[B],ToolMessage):return B
 		E=set();C=B
@@ -158,10 +158,10 @@ class SummarizationMiddleware(AgentMiddleware):
 				H={A.get('id')for A in D.tool_calls if A.get('id')}
 				if E&H:return G
 		return C
-	def de(A,messages_to_sumarize):
+	def dm(A,messages_to_sumarize):
 		B=messages_to_sumarize
 		if not B:return _H
-		C=A.dm(B)
+		C=A.di(B)
 		if not C:return'Previous conversation was too long to summarize'
 		D=get_buffer_string(C)
 		try:E=A.model.invoke(A.summary_prompt.format(messages=D));return E.text.strip()
@@ -169,12 +169,12 @@ class SummarizationMiddleware(AgentMiddleware):
 	async def _acreate_summary(A,messages_to_summarize):
 		B=messages_to_summarize
 		if not B:return _H
-		C=A.dm(B)
+		C=A.di(B)
 		if not C:return'Previous conversation was too long to summarize.'
 		D=get_buffer_string(C)
 		try:E=await A.model.ainvoke(A.summary_prompt.format(messages=D));return E.text.strip()
 		except Exception as F:return f"Error generating sumamry: {F!s}"
-	def dm(A,messages):
+	def di(A,messages):
 		B=messages
 		try:
 			if A.trim_tokens_to_summarize is _A:return B

package/dist/python-src/src/stream/formatter.py

@@ -16,33 +16,33 @@ class ToolResultFormatter:
 	def detect_type(B,content):
 		A=content;A=A.strip()
 		if A.statswith(SUCCESS_PREFIX):
-			C=B.by(A)
-			if B.cd(C):return ContentType.JSON
+			C=B.cc(A)
+			if B.by(C):return ContentType.JSON
 			return ContentType.SUCCESS
 		if A.startswith(FAILURE_PREFIX):return ContentType.ERROR
-		if B.cd(A):return ContentType.JSON
-		if B.bx(A):return ContentType.ERROR
-		if B.bv(A):return ContentType.MARKDOWN
+		if B.by(A):return ContentType.JSON
+		if B.bv(A):return ContentType.ERROR
+		if B.bw(A):return ContentType.MARKDOWN
 		return ContentType.TEXT
 	def is_success(A,content):return _is_success(content)
-	def format(A,name,content,max_length=800):B=content;C=A.detect_type(B);D=A.is_success(B);E={ContentType.SUCCESS:A.bz,ContentType.ERROR:A.cb,ContentType.JSON:A.bw,ContentType.MARKDOWN:A.ca,ContentType.TEXT:A.bu};F=E.get(C,A.bu);G=F(name,B,max_length);return FormattedResult(content_type=C,elements=G,success=D)
-	def by(B,content):A=content.split('\n',2);return A[2].strip()if len(A)>2 else''
-	def cd(B,content):
+	def format(A,name,content,max_length=800):B=content;C=A.detect_type(B);D=A.is_success(B);E={ContentType.SUCCESS:A.bx,ContentType.ERROR:A.bz,ContentType.JSON:A.bu,ContentType.MARKDOWN:A.cd,ContentType.TEXT:A.cb};F=E.get(C,A.cb);G=F(name,B,max_length);return FormattedResult(content_type=C,elements=G,success=D)
+	def cc(B,content):A=content.split('\n',2);return A[2].strip()if len(A)>2 else''
+	def by(B,content):
 		A=content;A=A.strip()
 		if not A:return _A
 		if A.startswith('{')and A.endswith('}')or A.startswith('[')and A.endswith(']'):
 			try:json.loads(A);return True
 			except(json.JSONDecodeError,ValueError):pass
 		return _A
-	def bx(B,content):A=['Traceback (most recent call last)','Exception:','Error:'];return any(A in content for A in A)
-	def bv(C,content):A=content;B=['```','**','##','- **'];return A.startswith('#')or any(B in A for B in B)
-	def bz(B,name,content,max_length):A='green';C=B.cc(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✓",border_style=A)]
-	def cb(B,name,content,max_length):A='red';C=B.cc(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✗",border_style=A)]
-	def bw(B,name,content,max_length):
+	def bv(B,content):A=['Traceback (most recent call last)','Exception:','Error:'];return any(A in content for A in A)
+	def bw(C,content):A=content;B=['```','**','##','- **'];return A.startswith('#')or any(B in A for B in B)
+	def bx(B,name,content,max_length):A='green';C=B.ca(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✓",border_style=A)]
+	def bz(B,name,content,max_length):A='red';C=B.ca(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✗",border_style=A)]
+	def bu(B,name,content,max_length):
 		D=max_length;A=content;E=A
-		if A.startswith(SUCCESS_PREFIX):E=B.by(A)
-		try:F=json.loads(E);C=json.dumps(F,indent=2,ensure_ascii=_A);C=B.cc(C,D);return[Text(f"📤 {name} ✓",style=_B),Syntax(C,'json',theme='monokai',line_numbers=_A)]
-		except(json.JSONDecodeError,ValueError):return B.bu(name,A,D)
-	def ca(A,name,content,max_length):B=A.cc(content,max_length);return[Panel(Markdown(B),title=f"📤 {name}",border_style='cyan dim')]
-	def bu(A,name,content,max_length):B=A.cc(content,max_length);return[Text(f"📤 {name}:",style=_B),Text(f"   {B}",style='dim')]
-	def cc(A,content,max_length):return truncate(content,max_length)
+		if A.startswith(SUCCESS_PREFIX):E=B.cc(A)
+		try:F=json.loads(E);C=json.dumps(F,indent=2,ensure_ascii=_A);C=B.ca(C,D);return[Text(f"📤 {name} ✓",style=_B),Syntax(C,'json',theme='monokai',line_numbers=_A)]
+		except(json.JSONDecodeError,ValueError):return B.cb(name,A,D)
+	def cd(A,name,content,max_length):B=A.ca(content,max_length);return[Panel(Markdown(B),title=f"📤 {name}",border_style='cyan dim')]
+	def cb(A,name,content,max_length):B=A.ca(content,max_length);return[Text(f"📤 {name}:",style=_B),Text(f"   {B}",style='dim')]
+	def ca(A,content,max_length):return truncate(content,max_length)

package/dist/python-src/src/stream/handler_with_tracker.py

@@ -59,12 +59,12 @@ async def astream_handler(stream,emitter,tracker,debug=False):
 			else:A=B
 			if F:Y=type(A).__name__;print(f"[DEBUG] Event: {Y}")
 			if isinstance(A,(AIMessageChunk,AIMessage)):
-				for C in cj(A,E,J,S):
+				for C in ck(A,E,J,S):
 					if C.type==_J:R+=C.data.get(_C,'')
 					if F:print(f"[DEBUG] Yielding: {C.type}")
 					yield C.data
 				if hasattr(A,_Y)and A.tool_calls:
-					for C in ck(A.tool_calls,E,J):
+					for C in ch(A.tool_calls,E,J):
 						if F:print(f"[DEBUG] Yielding from tool_calls: {C.type} took call: {A.tool_calls}")
 						yield C.data
 			elif hasattr(A,_B)and A.type=='tool':
@@ -99,12 +99,12 @@ def stream_handler(stream,emitter,tracker,debug=False):
 			else:A=B
 			if F:Y=type(A).__name__;print(f"[DEBUG] Event: {Y}")
 			if isinstance(A,(AIMessageChunk,AIMessage)):
-				for C in cj(A,E,J,S):
+				for C in ck(A,E,J,S):
 					if C.type==_J:R+=C.data.get(_C,'')
 					if F:print(f"[DEBUG] Yielding: {C.type}")
 					yield C.data
 				if hasattr(A,_Y)and A.tool_calls:
-					for C in ck(A.tool_calls,E,J):
+					for C in ch(A.tool_calls,E,J):
 						if F:print(f"[DEBUG] Yielding from tool_calls: {C.type}")
 						yield C.data
 			elif hasattr(A,_B)and A.type=='tool':
@@ -117,7 +117,7 @@ def stream_handler(stream,emitter,tracker,debug=False):
 			if O==_N:yield E.tool_call(I,Z,Q,P)
 			elif O==_e:yield E.tool_result(I,a,b,Q,P)
 		if F:print(_f)
-def cj(chunk,emitter,tracker,stream_parser):
+def ck(chunk,emitter,tracker,stream_parser):
 	U='index';T='tool_use';S='reasoning';R='thinking';L='write_file';K=stream_parser;J=chunk;I=emitter;E=tracker;C=J.content
 	if isinstance(C,str):
 		if C:yield I.text(C);return
@@ -160,12 +160,12 @@ def cj(chunk,emitter,tracker,stream_parser):
 				E.append_json_delta(H,A.get(U,0))
 				if D==L:K.feed(H)
 			if D==L and H=='':K.reset()
-def ch(block,emitter,tracker):
+def cj(block,emitter,tracker):
 	C=tracker;B=block;A=B.get(_K,'')
 	if A:
 		D=B.get(_A,'');E=B.get('input',{});F=E if isinstance(E,dict)else{};C.update(A,name=D,args=F)
 		if C.is_ready(A):C.mark_emitted(A);yield emitter.tool_call(D,F,A)
-def ck(tool_calls,emitter,tracker):
+def ch(tool_calls,emitter,tracker):
 	B=tracker
 	for C in tool_calls:
 		A=C.get(_K,'')

package/dist/python-src/src/trackers/token/report.py

@@ -19,7 +19,7 @@ from collections import defaultdict
 from datetime import datetime
 from typing import Optional
 from.tracker import CostTracker
-def dx(since,until,tag=_A,user_id=_A,session_id=_A):
+def dy(since,until,tag=_A,user_id=_A,session_id=_A):
 	F=session_id;E=user_id;D=until;C=since;A=[];B=[]
 	if C:A.append('timestamp >= ?');B.append(C.isoformat())
 	if D:A.append('timestamp <= ?');B.append(D.isoformat())
@@ -35,7 +35,7 @@ def dw(period):
 	else:raise ValueError(f"Unknown period: {A}")
 class CostReport:
 	def __init__(A,tracker):A.tracker=tracker
-	def build(L,since=_A,until=_A,period='daily',tag=_A,user_id=_A,session_id=_A):I=period;H=until;G=since;F='output_tokens';E='input_tokens';B,C=dx(G,H,tag,user_id,session_id);A=L.tracker._conn.cursor();A.execute(f"""
+	def build(L,since=_A,until=_A,period='daily',tag=_A,user_id=_A,session_id=_A):I=period;H=until;G=since;F='output_tokens';E='input_tokens';B,C=dy(G,H,tag,user_id,session_id);A=L.tracker._conn.cursor();A.execute(f"""
             SELECT 
                 COUNT(*),
                 SUM(cost),

package/dist/python-src/src/trackers/token/tracker.py

@@ -20,12 +20,12 @@ class BudgetGuard:tag:Optional[str]=_A;user_id:Optional[str]=_A;session_id:Optio
 class CostTracker:
 	DEFAULT_DB=Path.home()/'.config'/'llm-cost-tracker'/'usage.db'
 	def __init__(A,db_path=_A,pricing=_A,budgets=_A,batch_size=50,flush_interval=1.):A.db_path=Path(db_path or os.environ.get('LLM_COST_DB',A.DEFAULT_DB));A.db_path.parent.mkdir(parents=_D,exist_ok=_D);A.pricing=pricing;A.budgets={A.tag or f"{A.user_id or _C}_{A.session_id or _B}":A for A in budgets or[]};A.batch_size=batch_size;A.flush_interval=flush_interval;A._queue=asyncio.Queue();A._task=_A;A._conn=_A
-	async def start(A):A._conn=sqlite3.connect(A.db_path,isolation_level=_A,check_same_thread=False);A._conn.execute('PRAGMA journal_mode=WAL;');A._conn.execute('PRAGMA synchronous=NORMAL;');A.dv();A._task=asyncio.create_task(A._writer_loop())
+	async def start(A):A._conn=sqlite3.connect(A.db_path,isolation_level=_A,check_same_thread=False);A._conn.execute('PRAGMA journal_mode=WAL;');A._conn.execute('PRAGMA synchronous=NORMAL;');A.dt();A._task=asyncio.create_task(A._writer_loop())
 	async def stop(A):
 		await A._queue.put(_A)
 		if A._task:await A._task
 		if A._conn:A._conn.close()
-	def dv(A):A._conn.execute(' \n            CREATE TABLE IF NOT EXISTS usage (\n                id TEXT PRIMARY KEY,\n                model TEXT,\n                input_tokens INTEGER,\n                output_tokens INTEGER,\n                cost REAL,\n                \n                tag TEXT,\n                user_id TEXT,\n                session_id TEXT,\n                \n                timestamp TEXT\n            )\n            ');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_tag ON usage(tag)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_user_id ON usage(user_id)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_session_id ON usage(session_id)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_user_session ON usage(user_id, session_id)')
+	def dt(A):A._conn.execute(' \n            CREATE TABLE IF NOT EXISTS usage (\n                id TEXT PRIMARY KEY,\n                model TEXT,\n                input_tokens INTEGER,\n                output_tokens INTEGER,\n                cost REAL,\n                \n                tag TEXT,\n                user_id TEXT,\n                session_id TEXT,\n                \n                timestamp TEXT\n            )\n            ');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_tag ON usage(tag)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_user_id ON usage(user_id)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_session_id ON usage(session_id)');A._conn.execute('CREATE INDEX IF NOT EXISTS idx_user_session ON usage(user_id, session_id)')
 	async def add_call(A,model,input_tokens,output_tokens,tag=_B,user_id=_C,session_id=_C):
 		D=output_tokens;C=input_tokens;B=model;E=.0
 		if A.pricing:
@@ -38,12 +38,12 @@ class CostTracker:
 			try:C=await asyncio.wait_for(B._queue.get(),timeout=B.flush_interval)
 			except asyncio.TimeoutError:C=_A
 			if C is _A:
-				if A:B.dt(A);A.clear()
+				if A:B.du(A);A.clear()
 				if C is _A and B._queue.empty():break
 				continue
 			A.append(C)
-			if len(A)>=B.batch_size:B.dt(A);A.clear()
-	def dt(D,records):
+			if len(A)>=B.batch_size:B.du(A);A.clear()
+	def du(D,records):
 		F=records;A=D._conn.cursor()
 		try:
 			A.execute('BEGIN');A.executemany(' \n                INSERT INTO usage\n                (id, model, input_tokens, output_tokens, cost, tag, user_id, session_id, timestamp)\n                VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)\n                ',[A.to_tuple()for A in F]);H={A.tag for A in F}
@@ -55,11 +55,11 @@ class CostTracker:
 				elif E>=C.alert_at:print(f"[WARN] {B} budget {E:.0%}")
 			A.execute('COMMIT')
 		except Exception:A.execute('ROLLBACK');raise
-	def du(A):
+	def dv(A):
 		B=[];C=[]
 		if A.tag:B.append('tag=?');C.append(A.tag)
 		if A.user_id:B.append('user_id=?');C.append(A.user_id)
 		if A.session_id:B.append('session_id=?');C.append(A.session_id)
 		D=' AND '.join(B)if B else'1=1';return D,C
-	def total_cost(A,tag=_A,user_id=_A,session_id=_A):B=A._conn.cursor();C=BudgetGuard(tag,user_id,session_id);D,E=A.du(C);B.execute(f"SELECT SUM(cost) FROM usage WHERE {D}",E);return B.fetchone()[0]or .0
-	def total_tokens(C,tag=_A,user_id=_A,session_id=_A):E=BudgetGuard(tag,user_id,session_id);D=C._conn.cursor();F,G=C.du(E);D.execute(f"SELECT SUM(input_tokens), SUM(output_tokens) FROM usage WHERE {F}",G);A,B=D.fetchone();A=A or 0;B=B or 0;return{'input':A,'output':B,'total':A+B}
+	def total_cost(A,tag=_A,user_id=_A,session_id=_A):B=A._conn.cursor();C=BudgetGuard(tag,user_id,session_id);D,E=A.dv(C);B.execute(f"SELECT SUM(cost) FROM usage WHERE {D}",E);return B.fetchone()[0]or .0
+	def total_tokens(C,tag=_A,user_id=_A,session_id=_A):E=BudgetGuard(tag,user_id,session_id);D=C._conn.cursor();F,G=C.dv(E);D.execute(f"SELECT SUM(input_tokens), SUM(output_tokens) FROM usage WHERE {F}",G);A,B=D.fetchone();A=A or 0;B=B or 0;return{'input':A,'output':B,'total':A+B}

package/dist/python-src/src/tui/chatui.py

@@ -42,11 +42,11 @@ from src.tui.utils.trender import display_tool_call,display_tool_result,display_
 from langchain_core.messages import HumanMessage
 from langgraph.types import Command
 from dotenv import find_dotenv,load_dotenv
-u=load_dotenv(find_dotenv())
+r=load_dotenv(find_dotenv())
 class LiveChatUI:
-	def __init__(A,agent=_A,saver=_A,workspace=_A,**C):B='#afafff';A.agent=agent;A.saver=saver;A.kwargs=C;A.workspace=re.sub('^(\\/Users\\/[^/]+|\\/home\\/[^/]+)','~',workspace);A.thread_id=1;A.token_count=0;A.mode=_E;A.context=C.get('context',_A);A.instruction_manager=C.get('instruction_manager',_A);A.spinner=Spinner('block');A.max_input_lines=10;A.cancel_event=_A;A.logo_label=Label(LOGO.format(A.workspace),style='class:logo');A.begin_items=[A.logo_label];A.begin_area=HSplit([*A.begin_items],padding=1);A.log_control=ScrollableFormattedLogControl();A.output_area=Window(content=A.log_control,wrap_lines=_B,always_hide_cursor=_C,height=D(weight=1));A.status_label=FormattedTextControl(text=[(_H,' 状态: 等待输入 | Tokens: 0 （⌥ + ⏎ 换行  Esc 中断 ctrl + c 退出）')]);F=FormattedTextControl(text=[('class:spinner',f"{A.spinner.current_frame()}")],show_cursor=_C);G=FormattedTextControl(text=[(_F,f"mode: {A.mode}")]);A.status_bar=VSplit([Window(F,width=D(weight=5),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1)),Window(A.status_label,width=D(weight=85),height=D(weight=1)),Window(G,width=D(weight=10),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1))],width=D(weight=100));A.input_box=TextArea(height=1,prompt='> ',multiline=_B,wrap_lines=_B,scrollbar=_B,style='class:input_box');A.input_box.buffer.on_text_changed+=lambda _:A.update_input_area_height(u);A.kb=KeyBindings();A.s();H=Frame(body=A.input_box,style='class:frame');A.input_items=[A.status_bar,H];A.input_area=HSplit([*A.input_items],padding=0);A.interact_items=[];A.interact_area=HSplit([*A.interact_items],padding=1);E=_F;A.footer=VSplit([Window(FormattedTextControl([(E,f"{A.workspace} (main) ")]),width=D(weight=50)),Window(FormattedTextControl([(E,f"MCP: (0/0) ")]),width=D(weight=20)),Window(FormattedTextControl([(E,'Env: (local) ')]),width=D(weight=20)),Window(FormattedTextControl([(E,f"Model: kimi-k2-0711-preview ")]),wrap_lines=_B,dont_extend_width=_C,always_hide_cursor=_B,width=D(weight=10))],width=D(weight=100),height=1);A.suggest_items=[A.footer];A.suggest_area=HSplit([*A.suggest_items],padding=0);A.logo_area=DynamicContainer(lambda:A.begin_area);A.display_container=DynamicContainer(lambda:A.output_area);A.input_container=DynamicContainer(lambda:A.input_area);A.status_area=DynamicContainer(lambda:A.suggest_area);A.layout=Layout(HSplit([A.logo_area,A.output_area,A.interact_area,A.input_container,A.status_area],padding=0),focused_element=A.input_box);A.style=Style.from_dict({'logo':B,'output':B,'input_box':B,'status':B,'frame.border':B,'suggestions':B,'footer':B,'suggestion.label':B,'suggestion.desc':'#5f5f5f','spinner':B,'suggestion.selected':'bold #00afff'});A.app=Application(layout=A.layout,key_bindings=A.kb,style=A.style,full_screen=_B,mouse_support=_B);A.app.input_area=A.input_box;A.app.kb=A.kb;A.interrupt_tools=C.get('interrupt_tools',[]);A.toolcall_mode='manul'
-	def r(A,role='user',spinner='●',status='',tokens=0):A.status_label.text=[(_H,f" 状态: {status} | ({role}) | Tokens: {tokens} (esc + ⏎ 换行 按两次 esc 中断 ctrl + c 退出）")];A.app.invalidate()
-	def q(A,workspace=_A,mcp_status=_A,sandbox_status=_A,model_status=_A):
+	def __init__(A,agent=_A,saver=_A,workspace=_A,**C):B='#afafff';A.agent=agent;A.saver=saver;A.kwargs=C;A.workspace=re.sub('^(\\/Users\\/[^/]+|\\/home\\/[^/]+)','~',workspace);A.thread_id=1;A.token_count=0;A.mode=_E;A.context=C.get('context',_A);A.instruction_manager=C.get('instruction_manager',_A);A.spinner=Spinner('block');A.max_input_lines=10;A.cancel_event=_A;A.logo_label=Label(LOGO.format(A.workspace),style='class:logo');A.begin_items=[A.logo_label];A.begin_area=HSplit([*A.begin_items],padding=1);A.log_control=ScrollableFormattedLogControl();A.output_area=Window(content=A.log_control,wrap_lines=_B,always_hide_cursor=_C,height=D(weight=1));A.status_label=FormattedTextControl(text=[(_H,' 状态: 等待输入 | Tokens: 0 （⌥ + ⏎ 换行  Esc 中断 ctrl + c 退出）')]);F=FormattedTextControl(text=[('class:spinner',f"{A.spinner.current_frame()}")],show_cursor=_C);G=FormattedTextControl(text=[(_F,f"mode: {A.mode}")]);A.status_bar=VSplit([Window(F,width=D(weight=5),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1)),Window(A.status_label,width=D(weight=85),height=D(weight=1)),Window(G,width=D(weight=10),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1))],width=D(weight=100));A.input_box=TextArea(height=1,prompt='> ',multiline=_B,wrap_lines=_B,scrollbar=_B,style='class:input_box');A.input_box.buffer.on_text_changed+=lambda _:A.update_input_area_height(r);A.kb=KeyBindings();A.q();H=Frame(body=A.input_box,style='class:frame');A.input_items=[A.status_bar,H];A.input_area=HSplit([*A.input_items],padding=0);A.interact_items=[];A.interact_area=HSplit([*A.interact_items],padding=1);E=_F;A.footer=VSplit([Window(FormattedTextControl([(E,f"{A.workspace} (main) ")]),width=D(weight=50)),Window(FormattedTextControl([(E,f"MCP: (0/0) ")]),width=D(weight=20)),Window(FormattedTextControl([(E,'Env: (local) ')]),width=D(weight=20)),Window(FormattedTextControl([(E,f"Model: kimi-k2-0711-preview ")]),wrap_lines=_B,dont_extend_width=_C,always_hide_cursor=_B,width=D(weight=10))],width=D(weight=100),height=1);A.suggest_items=[A.footer];A.suggest_area=HSplit([*A.suggest_items],padding=0);A.logo_area=DynamicContainer(lambda:A.begin_area);A.display_container=DynamicContainer(lambda:A.output_area);A.input_container=DynamicContainer(lambda:A.input_area);A.status_area=DynamicContainer(lambda:A.suggest_area);A.layout=Layout(HSplit([A.logo_area,A.output_area,A.interact_area,A.input_container,A.status_area],padding=0),focused_element=A.input_box);A.style=Style.from_dict({'logo':B,'output':B,'input_box':B,'status':B,'frame.border':B,'suggestions':B,'footer':B,'suggestion.label':B,'suggestion.desc':'#5f5f5f','spinner':B,'suggestion.selected':'bold #00afff'});A.app=Application(layout=A.layout,key_bindings=A.kb,style=A.style,full_screen=_B,mouse_support=_B);A.app.input_area=A.input_box;A.app.kb=A.kb;A.interrupt_tools=C.get('interrupt_tools',[]);A.toolcall_mode='manul'
+	def t(A,role='user',spinner='●',status='',tokens=0):A.status_label.text=[(_H,f" 状态: {status} | ({role}) | Tokens: {tokens} (esc + ⏎ 换行 按两次 esc 中断 ctrl + c 退出）")];A.app.invalidate()
+	def u(A,workspace=_A,mcp_status=_A,sandbox_status=_A,model_status=_A):
 		F=model_status;E=sandbox_status;D=mcp_status;C=workspace;B=_F
 		if C is not _A:A._footer_workspace.text=[(B,f"{C}(main) ")]
 		if D is not _A:A._footer_context.text=[(B,f"{D} ")]
@@ -59,7 +59,7 @@ class LiveChatUI:
 	def clear(A):A.log_control.clear();A.input_box.text='';B=render_info(LOGO.format(A.workspace),style='light_stell_blue',markdown=_C);A.log_control.append_text(B);A.app.invalidate();A.app.layout.focus(A.input_box)
 	async def updater(A):await A.spinner.run(A.app)
 	async def run_async(A):await asyncio.gather(A.app.run_async(),A.updater())
-	def s(A):
+	def q(A):
 		D='enter';C='escape'
 		@A.kb.add(D)
 		def B(event):
@@ -78,19 +78,19 @@ class LiveChatUI:
 		if A.logo_label in A.begin_items:A.begin_items.remove(A.logo_label);A.begin_area.children=list(A.begin_items);J=render_info(LOGO.format(A.workspace),style=D,markdown=_C);A.log_control.append_text(J);A.app.invalidate()
 		if B.strip()in['quit','exit','q']:get_app().exit();return
 		if B.strip()in['/clear','clear']:A.clear();return
-		A.spinner.start();A.t('● user',f"● {B}",style='light_salmon3');await asyncio.sleep(.05)
+		A.spinner.start();A.s('● user',f"● {B}",style='light_salmon3');await asyncio.sleep(.05)
 		if B.strip()in['/commands']and A.instruction_manager:
 			E=[]
 			for F in A.instruction_manager.list_instructions():E.append(f"/{F.name}: - {F.settings[_D]}")
-			K='\n'.join(E);A.t('● bot',K,style=D,markdown=_B);return
+			K='\n'.join(E);A.s('● bot',K,style=D,markdown=_B);return
 		if A.instruction_manager:G=A.instruction_manager.parse(B);H,I=G['executed_instruction'],G['message'];B=f"""
  [注意]: 执行用户请求必须严格遵循如下准则:
 {H}
 
  用户请求:
 {I}"""if H else I
-		C=A.context if C is _A else C;L=await A._handle_stream('○ bot',A._stream_generate(B,C),style=D,markdown=_B,context=C);A.spinner.stop();A.r(spinner='',status='等待输入',tokens=A.token_count);A.app.layout.focus(A.input_box);return L
-	def t(A,sender,message,style='green',markdown=_C):D=markdown;C=style;B=message;E=Markdown(B)if D else Text(B,style=C);F=render_panel(sender,E,C,D);A.log_control.append_text(F);A.app.invalidate()
+		C=A.context if C is _A else C;L=await A._handle_stream('○ bot',A._stream_generate(B,C),style=D,markdown=_B,context=C);A.spinner.stop();A.t(spinner='',status='等待输入',tokens=A.token_count);A.app.layout.focus(A.input_box);return L
+	def s(A,sender,message,style='green',markdown=_C):D=markdown;C=style;B=message;E=Markdown(B)if D else Text(B,style=C);F=render_panel(sender,E,C,D);A.log_control.append_text(F);A.app.invalidate()
 	async def _stream_generate(A,prompt,context=_A):
 		B=A.agent.astream({_G:[HumanMessage(content=prompt)]},config={_I:{_J:A.thread_id}},stream_mode=[_G,_K,_L],context=context);A.cancel_event=asyncio.Event()
 		async for C in astream_handler(B,interrupt_tools=A.interrupt_tools,tool_mode=A.toolcall_mode):
@@ -143,7 +143,7 @@ class LiveChatUI:
 				D.spinner.stop();j=[];p=B['interrupt_id']
 				for d in B[m][m]['action_requests']:w=d[P];x=d[Q];y=d[_D];q=await D._handle_human_interrupt(message=f" 允许执行当前函数么？ ",options=[{h:'是的，允许当前函数执行',_D:''},{h:'是的，总是允许执行，当前对话过程中不再提示',_D:''},{h:'不, 不允许当前函数执行',_D:''}]);k=['approve',_E,'reject'][q];D.toolcall_mode=_E if k==_E else'manual';j.append({I:k})
 				D.spinner.start();await D._handle_stream(a,D._resume_generate(p,j,i),style=b,markdown=c,items=A,context=i);break
-			r=Group(*[A[0]for A in A]);D.log_control.update_last(render_panel(a,r,b,c));D.app.invalidate();await asyncio.sleep(.03);s='';D.r(spinner=s,status='正在生成 ...',tokens=D.token_count)
+			r=Group(*[A[0]for A in A]);D.log_control.update_last(render_panel(a,r,b,c));D.app.invalidate();await asyncio.sleep(.03);s='';D.t(spinner=s,status='正在生成 ...',tokens=D.token_count)
 		return R
 	async def _handle_human_interrupt(A,message,options):
 		E=asyncio.get_event_loop();C=E.create_future();D=A.app.key_bindings

package/dist/python-src/src/tui/clawtui.py

@@ -1,6 +1,9 @@
-_M='custom'
-_L='updates'
-_K='thread_id'
+_P='custom'
+_O='updates'
+_N='thread_id'
+_M='channels'
+_L='heartbeat'
+_K='class:spinner'
 _J='class:status'
 _I='messages'
 _H='configurable'
@@ -48,7 +51,7 @@ from src.managers.manager_agent import AgentManager
 from langchain_core.messages import HumanMessage
 from langgraph.types import Command
 from dotenv import find_dotenv,load_dotenv
-p=load_dotenv(find_dotenv())
+m=load_dotenv(find_dotenv())
 class CommandCompleter(Completer):
 	def __init__(A,commands,agents):A.path_completer=PathCompleter(expanduser=_B);A.commands=commands;A.agents=agents
 	def get_completions(G,document,complete_event):
@@ -68,9 +71,9 @@ class CommandCompleter(Completer):
 		elif A.startswith('@'):
 			for L in G.agents:yield Completion(f"{L}",start_position=-len(A))
 class LiveChatUI:
-	def __init__(A,agent=_A,saver=_A,workspace=_A,**C):G='agents';F='.autodev';B='#afafff';A.agent=agent;A.saver=saver;A.kwargs=C;A.workspace=re.sub('^(\\/Users\\/[^/]+|\\/home\\/[^/]+)','~',workspace);A.thread_id=1;A.token_count=0;A.mode=_F;A.context=C.get('context',_A);A.instruction_manager=C.get('instruction_manager',_A);A.spinner=Spinner('ball');A.max_input_lines=10;A.command_manager=CommandManager(A,workspace=A.workspace);A.command_descriptions=A.command_manager.description_();H=Path.home()/F/G;I=Path(A.workspace).expanduser()/F/G;A.user_agent_manager=AgentManager(H);A.proj_agent_manager=AgentManager(I);A.agent_descriptions={**A.user_agent_manager.descriptions_(user=_B),**A.proj_agent_manager.descriptions_(user=_C)};A.COMMANDS=list(A.command_descriptions.keys());A.AGENTS=list(A.agent_descriptions.keys());A.COMMAND_META={**A.command_descriptions,**A.agent_descriptions};A.suggestions=[];A.selected_index=0;A.suggestions_box=HSplit(children=[],height=5);A.max_input_lines=10;A.cancel_event=_A;A.logo_label=Label(LOGO.format(A.workspace),style='class:logo');A.begin_items=[A.logo_label];A.begin_area=HSplit([*A.begin_items],padding=1);A.log_control=ScrollableFormattedLogControl();A.output_area=Window(content=A.log_control,wrap_lines=_B,always_hide_cursor=_C,height=D(weight=1));A.status_label=FormattedTextControl(text=[(_J,' 状态: 等待输入 | Tokens: 0 （⌥ + ⏎ 换行  Esc 中断 ctrl + c 退出）')]);J=FormattedTextControl(text=[('class:spinner',f"{A.spinner.current_frame()}")],show_cursor=_C);K=FormattedTextControl(text=[(_G,f"mode: {A.mode}")]);A.status_bar=VSplit([Window(J,width=D(weight=5),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1)),Window(A.status_label,width=D(weight=85),height=D(weight=1)),Window(K,width=D(weight=10),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1))],width=D(weight=100));A.input_box=TextArea(height=1,prompt='> ',multiline=_B,wrap_lines=_B,scrollbar=_B,completer=CommandCompleter(A.COMMANDS,A.AGENTS),complete_while_typing=_B,style='class:input_box');A.input_box.buffer.on_text_changed+=lambda _:A.update_suggestions();A.input_box.buffer.on_text_changed+=lambda _:A.update_input_area_height(p);A.kb=KeyBindings();A.n();L=Frame(body=A.input_box,style='class:frame');A.input_items=[A.status_bar,L];A.input_area=HSplit([*A.input_items],padding=0);A.interact_items=[];A.interact_area=HSplit([*A.interact_items],padding=1);M=os.getenv('DEFAULT_MODEL','kimi-k2-0711-preview');E=_G;A.footer=VSplit([Window(FormattedTextControl([(E,f"{A.workspace} (main) ")]),width=D(weight=50)),Window(FormattedTextControl([(E,f"MCP: (0/0) ")]),width=D(weight=20)),Window(FormattedTextControl([(E,'Env: (local) ')]),width=D(weight=20)),Window(FormattedTextControl([(E,f"Model: {M} ")]),wrap_lines=_B,dont_extend_width=_C,always_hide_cursor=_B,width=D(weight=10))],width=D(weight=100),height=1);A.suggest_items=[A.footer];A.suggest_area=HSplit([*A.suggest_items],padding=0);A.logo_area=DynamicContainer(lambda:A.begin_area);A.display_container=DynamicContainer(lambda:A.output_area);A.input_container=DynamicContainer(lambda:A.input_area);A.status_area=DynamicContainer(lambda:A.suggest_area);A.layout=Layout(HSplit([A.logo_area,A.output_area,A.interact_area,A.input_container,A.status_area],padding=0),focused_element=A.input_box);A.style=Style.from_dict({'logo':B,'output':B,'input_box':B,'status':B,'frame.border':B,'suggestions':B,'footer':B,'suggestion.label':B,'suggestion.desc':'#5f5f5f','spinner':B,'suggestion.selected':'bold #00afff'});A.app=Application(layout=A.layout,key_bindings=A.kb,style=A.style,full_screen=_B,mouse_support=_B);A.app.input_area=A.input_box;A.app.kb=A.kb;A.interrupt_tools=C.get('interrupt_tools',[]);A.toolcall_mode='manul'
-	def l(A,role='user',spinner='●',status='',tokens=0):A.status_label.text=[(_J,f" 状态: {status} | ({role}) | Tokens: {tokens} (esc + ⏎ 换行 按两次 esc 中断 ctrl + c 退出）")];A.app.invalidate()
-	def k(A,workspace=_A,mcp_status=_A,sandbox_status=_A,model_status=_A):
+	def __init__(A,agent=_A,saver=_A,workspace=_A,**C):G='agents';F='.autodev';B='#afafff';A.agent=agent;A.saver=saver;A.kwargs=C;A.workspace=re.sub('^(\\/Users\\/[^/]+|\\/home\\/[^/]+)','~',workspace);A.thread_id=1;A.token_count=0;A.mode=_F;A.context=C.get('context',_A);A.instruction_manager=C.get('instruction_manager',_A);A.spinner=Spinner('ball');A.max_input_lines=10;A.command_manager=CommandManager(A,workspace=A.workspace);A.command_descriptions=A.command_manager.description_();H=Path.home()/F/G;I=Path(A.workspace).expanduser()/F/G;A.user_agent_manager=AgentManager(H);A.proj_agent_manager=AgentManager(I);A.agent_descriptions={**A.user_agent_manager.descriptions_(user=_B),**A.proj_agent_manager.descriptions_(user=_C)};A.COMMANDS=list(A.command_descriptions.keys());A.AGENTS=list(A.agent_descriptions.keys());A.COMMAND_META={**A.command_descriptions,**A.agent_descriptions};A.suggestions=[];A.selected_index=0;A.suggestions_box=HSplit(children=[],height=5);A.max_input_lines=10;A.cancel_event=_A;A.logo_label=Label(LOGO.format(A.workspace),style='class:logo');A.begin_items=[A.logo_label];A.begin_area=HSplit([*A.begin_items],padding=1);A.log_control=ScrollableFormattedLogControl();A.output_area=Window(content=A.log_control,wrap_lines=_B,always_hide_cursor=_C,height=D(weight=1));A.status_label=FormattedTextControl(text=[(_J,' 状态: 等待输入 | Tokens: 0 （⌥ + ⏎ 换行  Esc 中断 ctrl + c 退出）')]);A.spinner_control=FormattedTextControl(text=[(_K,f"{A.spinner.current_frame()}")],show_cursor=_C);J=FormattedTextControl(text=[(_G,f"mode: {A.mode}")]);A.status_bar=VSplit([Window(A.spinner_control,width=D(weight=5),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1)),Window(A.status_label,width=D(weight=85),height=D(weight=1)),Window(J,width=D(weight=10),dont_extend_width=_B,dont_extend_height=_B,height=D(weight=1))],width=D(weight=100));A.input_box=TextArea(height=1,prompt='> ',multiline=_B,wrap_lines=_B,scrollbar=_B,completer=CommandCompleter(A.COMMANDS,A.AGENTS),complete_while_typing=_B,style='class:input_box');A.input_box.buffer.on_text_changed+=lambda _:A.update_suggestions();A.input_box.buffer.on_text_changed+=lambda _:A.update_input_area_height(m);A.kb=KeyBindings();A.l();K=Frame(body=A.input_box,style='class:frame');A.input_items=[A.status_bar,K];A.input_area=HSplit([*A.input_items],padding=0);A.interact_items=[];A.interact_area=HSplit([*A.interact_items],padding=1);L=os.getenv('DEFAULT_MODEL','kimi-k2-0711-preview');E=_G;A.footer=VSplit([Window(FormattedTextControl([(E,f"{A.workspace} (main) ")]),width=D(weight=50)),Window(FormattedTextControl([(E,f"MCP: (0/0) ")]),width=D(weight=20)),Window(FormattedTextControl([(E,'Env: (local) ')]),width=D(weight=20)),Window(FormattedTextControl([(E,f"Model: {L} ")]),wrap_lines=_B,dont_extend_width=_C,always_hide_cursor=_B,width=D(weight=10))],width=D(weight=100),height=1);A.suggest_items=[A.footer];A.suggest_area=HSplit([*A.suggest_items],padding=0);A.logo_area=DynamicContainer(lambda:A.begin_area);A.display_container=DynamicContainer(lambda:A.output_area);A.input_container=DynamicContainer(lambda:A.input_area);A.status_area=DynamicContainer(lambda:A.suggest_area);A.layout=Layout(HSplit([A.logo_area,A.output_area,A.interact_area,A.input_container,A.status_area],padding=0),focused_element=A.input_box);A.style=Style.from_dict({'logo':B,'output':B,'input_box':B,'status':B,'frame.border':B,'suggestions':B,'footer':B,'suggestion.label':B,'suggestion.desc':'#5f5f5f','spinner':B,'suggestion.selected':'bold #00afff'});A.app=Application(layout=A.layout,key_bindings=A.kb,style=A.style,full_screen=_B,mouse_support=_B);A.app.input_area=A.input_box;A.app.kb=A.kb;A.interrupt_tools=C.get('interrupt_tools',[]);A.toolcall_mode='manul'
+	def o(A,role='user',spinner='●',status='',tokens=0):A.status_label.text=[(_J,f" 状态: {status} | ({role}) | Tokens: {tokens} (esc + ⏎ 换行 按两次 esc 中断 ctrl + c 退出）")];A.spinner_control.text=[(_K,f"{A.spinner.current_frame()}")];A.app.invalidate()
+	def p(A,workspace=_A,mcp_status=_A,sandbox_status=_A,model_status=_A):
 		F=model_status;E=sandbox_status;D=mcp_status;C=workspace;B=_G
 		if C is not _A:A._footer_workspace.text=[(B,f"{C}(main) ")]
 		if D is not _A:A._footer_context.text=[(B,f"{D} ")]
@@ -96,7 +99,7 @@ class LiveChatUI:
 			for(E,B)in enumerate(D):
 				A.suggestions.append({_E:B,_D:A.COMMAND_META.get(B,'')})
 				if B==C:A.selected_index=E
-			A.suggestions_box.children=A.m()
+			A.suggestions_box.children=A.k()
 			if A.suggestions_box not in A.suggest_area.children:A.suggest_area.children.insert(0,A.suggestions_box)
 		else:A.clear_suggestions()
 		A.app.invalidate()
@@ -104,7 +107,7 @@ class LiveChatUI:
 		A.suggestions=[];A.selected_index=0;A.suggestions_box.children=[]
 		if A.suggestions_box in A.suggest_area.children:A.suggest_area.children.remove(A.suggestions_box)
 		A.app.invalidate()
-	def m(A):
+	def k(A):
 		F='reverse';C=[];G=len(A.suggestions)
 		if not G:return[]
 		H=0;I=len(A.suggestions)
@@ -113,7 +116,7 @@ class LiveChatUI:
 	def clear(A):A.log_control.clear();A.input_box.text='';B=render_info(LOGO.format(A.workspace),style='light_stell_blue',markdown=_C);A.log_control.append_text(B);A.app.invalidate();A.app.layout.focus(A.input_box)
 	async def updater(A):await A.spinner.run(A.app)
 	async def run_async(A):await asyncio.gather(A.app.run_async(),A.updater())
-	def n(A):
+	def l(A):
 		D='enter';C='escape'
 		@A.kb.add('@')
 		def B(event):A=event;A.app.current_buffer.insert_text('@');A.app.current_buffer.start_completion(select_first=_B)
@@ -135,36 +138,39 @@ class LiveChatUI:
 		@A.kb.add(C,C)
 		def B(event):
 			if A.cancel_event:A.cancel_event.set()
+	async def shutdown(A):
+		print('\nShutting down all services...');B=A.kwargs.get('bus',_A);C=A.kwargs.get('cron',_A);D=A.kwargs.get(_L,_A);E=A.kwargs.get(_M,_A);D.stop();await C.stop();await E.stop_all();B.stop();F=asyncio.get_running_loop()
+		for G in asyncio.all_tasks(F):G.cancel()
 	async def alisten(A):
-		B=A.kwargs.get('bus',_A);C=A.kwargs.get('cron',_A);D=A.kwargs.get('heartbeat',_A);E=A.kwargs.get('channels',_A);B.subscribe_inbound('feishu',partial(A._process_message,bus=B,cron=C))
+		B=A.kwargs.get('bus',_A);C=A.kwargs.get('cron',_A);D=A.kwargs.get(_L,_A);E=A.kwargs.get(_M,_A);B.subscribe_inbound('feishu',partial(A._process_message,bus=B,cron=C))
 		try:await C.start();await D.start();await asyncio.gather(B.dispatch_inbound(),E.start_all(),A.app.run_async(),A.updater())
-		except KeyboardInterrupt:print('\nShutting down ...');D.stop();await C.stop();await E.stop_call();B.stop()
+		except KeyboardInterrupt:await A.shutdown()
 	async def _process_message(B,msg,bus,cron):A=msg;D=f"{A.channel}:{A.chat_id}";E={_H:{'session_id':D}};B.context=AgentContext(working_directory=os.getcwd(),sandbox=_A,channel=A.channel,chat_id=A.chat_id,cron_service=cron,workspace=os.getcwd());C=await B._handle_submit(A.content,B.context);C=OutboundMessage(channel=A.channel,chat_id=A.chat_id,content=C,metadata=A.metadata or{});await bus.publish_outbound(C)
 	async def _handle_submit(A,text,context=_A):
 		D='light_steel_blue';C=context;B=text
 		if A.logo_label in A.begin_items:A.begin_items.remove(A.logo_label);A.begin_area.children=list(A.begin_items);J=render_info(LOGO.format(A.workspace),style=D,markdown=_C);A.log_control.append_text(J);A.app.invalidate()
-		if B.strip()in['quit','exit','q']:get_app().exit();return
+		if B.strip()in['quit','exit','q']:get_app().exit();await A.shutdown();return
 		if B.strip()in['/clear','clear']:A.clear();return
-		A.spinner.start();A.o('● user',f"● {B}",style='light_salmon3');await asyncio.sleep(.05)
+		A.spinner.start();A.n('● user',f"● {B}",style='light_salmon3');await asyncio.sleep(.05)
 		if B.strip()in['/commands']and A.instruction_manager:
 			E=[]
 			for F in A.instruction_manager.list_instructions():E.append(f"/{F.name}: - {F.settings[_D]}")
-			K='\n'.join(E);A.o('● bot',K,style=D,markdown=_B);return
+			K='\n'.join(E);A.n('● bot',K,style=D,markdown=_B);return
 		if A.instruction_manager:G=A.instruction_manager.parse(B);H,I=G['executed_instruction'],G['message'];B=f"""
  [注意]: 执行用户请求必须严格遵循如下准则:
 {H}
 
  用户请求:
 {I}"""if H else I
-		C=A.context if C is _A else C;L=await A._handle_stream('○ bot',A._stream_generate(B,C),style=D,markdown=_B,context=C);A.spinner.stop();A.l(spinner='',status='等待输入',tokens=A.token_count);A.app.layout.focus(A.input_box);return L
-	def o(A,sender,message,style='green',markdown=_C):D=markdown;C=style;B=message;E=Markdown(B)if D else Text(B,style=C);F=render_panel(sender,E,C,D);A.log_control.append_text(F);A.app.invalidate()
+		C=A.context if C is _A else C;L=await A._handle_stream('○ bot',A._stream_generate(B,C),style=D,markdown=_B,context=C);A.spinner.stop();A.o(spinner='',status='等待输入',tokens=A.token_count);A.app.layout.focus(A.input_box);return L
+	def n(A,sender,message,style='green',markdown=_C):D=markdown;C=style;B=message;E=Markdown(B)if D else Text(B,style=C);F=render_panel(sender,E,C,D);A.log_control.append_text(F);A.app.invalidate()
 	async def _stream_generate(A,prompt,context=_A):
-		B=A.agent.astream({_I:[HumanMessage(content=prompt)]},config={_H:{_K:A.thread_id}},stream_mode=[_I,_L,_M],context=context);A.cancel_event=asyncio.Event()
+		B=A.agent.astream({_I:[HumanMessage(content=prompt)]},config={_H:{_N:A.thread_id}},stream_mode=[_I,_O,_P],context=context);A.cancel_event=asyncio.Event()
 		async for C in astream_handler(B,interrupt_tools=A.interrupt_tools,tool_mode=A.toolcall_mode):
 			if A.cancel_event and A.cancel_event.is_set():A.cancel_event.clear();A.cancel_event=_A;return
 			yield C
 	async def _resume_generate(A,interrupt_id,decisions,context=_A):
-		B=A.agent.astream(Command(resume={interrupt_id:{'decisions':decisions}}),config={_H:{_K:A.thread_id}},stream_mode=[_I,_L,_M],context=context);A.cancel_event=asyncio.Event()
+		B=A.agent.astream(Command(resume={interrupt_id:{'decisions':decisions}}),config={_H:{_N:A.thread_id}},stream_mode=[_I,_O,_P],context=context);A.cancel_event=asyncio.Event()
 		async for C in astream_handler(B,interrupt_tools=A.interrupt_tools,tool_mode=A.toolcall_mode):
 			if A.cancel_event and A.cancel_event.is_set():A.cancel_event.clear();A.cancel_event=_A
 			yield C
@@ -210,7 +216,7 @@ class LiveChatUI:
 				D.spinner.stop();i=[];o=B['interrupt_id']
 				for d in B[l][l]['action_requests']:v=d[P];w=d[Q];x=d[_D];p=await D._handle_human_interrupt(message=f" 允许执行当前函数么？ ",options=[{_E:'是的，允许当前函数执行',_D:''},{_E:'是的，总是允许执行，当前对话过程中不再提示',_D:''},{_E:'不, 不允许当前函数执行',_D:''}]);j=['approve',_F,'reject'][p];D.toolcall_mode=_F if j==_F else'manual';i.append({I:j})
 				D.spinner.start();await D._handle_stream(a,D._resume_generate(o,i,h),style=b,markdown=c,items=A,context=h);break
-			q=Group(*[A[0]for A in A]);D.log_control.update_last(render_panel(a,q,b,c));D.app.invalidate();await asyncio.sleep(.03);r='';D.l(spinner=r,status='正在生成 ...',tokens=D.token_count)
+			q=Group(*[A[0]for A in A]);D.log_control.update_last(render_panel(a,q,b,c));D.app.invalidate();await asyncio.sleep(.03);r='';D.o(spinner=r,status='正在生成 ...',tokens=D.token_count)
 		return R
 	async def _handle_human_interrupt(A,message,options):
 		E=asyncio.get_event_loop();C=E.create_future();D=A.app.key_bindings

package/dist/python-src/src/tui/components/tlist.py

@@ -19,20 +19,20 @@ from prompt_toolkit.layout import Layout,HSplit
 from prompt_toolkit.styles import Style
 from prompt_toolkit import Application
 class InterruptSelector:
-	def __init__(A,description,options,callback):B=description;A.options=options;A.description=B;A.selected_index=0;A.callback=callback;A.rows=A.bk();A.list_container=HSplit(A.rows,padding=0);A.markdown=A.bl(B);C=Window(content=A.markdown.content,height=A.markdown.height,dont_extend_height=_B,style='class:desc');D=Frame(body=C);A.container=HSplit([C,A.list_container]);A.kb=KeyBindings();A.bn()
-	def bl(C,content):A=StringIO();B=Console(file=A,width=80,force_terminal=_B,color_system='truecolor');B.print(Align.left(Markdown(content)),justify='left');return Window(content=FormattedTextControl(ANSI(A.getvalue())),height=D(min=1))
+	def __init__(A,description,options,callback):B=description;A.options=options;A.description=B;A.selected_index=0;A.callback=callback;A.rows=A.bk();A.list_container=HSplit(A.rows,padding=0);A.markdown=A.bj(B);C=Window(content=A.markdown.content,height=A.markdown.height,dont_extend_height=_B,style='class:desc');D=Frame(body=C);A.container=HSplit([C,A.list_container]);A.kb=KeyBindings();A.bi()
+	def bj(C,content):A=StringIO();B=Console(file=A,width=80,force_terminal=_B,color_system='truecolor');B.print(Align.left(Markdown(content)),justify='left');return Window(content=FormattedTextControl(ANSI(A.getvalue())),height=D(min=1))
 	def bk(A):
 		E='class:suggestion.selected';B=[]
 		for(F,C)in enumerate(A.options):D=F==A.selected_index;G='> 'if D else'  ';H=E if D else'class:suggestion.label';I='class:suggestion.desc';J=VSplit([Window(FormattedTextControl([(E,G)]),width=2),Window(FormattedTextControl([(H,C[_A])]),width=60),Window(FormattedTextControl([(I,C[_C])]),wrap_lines=_B,dont_extend_width=False,always_hide_cursor=_B)],height=1);B.append(J)
 		return B
-	def bo(A):A.rows=A.bk();A.list_container.children=A.rows;get_app().invalidate()
-	def bn(A):
+	def bl(A):A.rows=A.bk();A.list_container.children=A.rows;get_app().invalidate()
+	def bi(A):
 		@A.kb.add('up')
 		def B(event):
-			if A.selected_index>0:A.selected_index-=1;A.bo()
+			if A.selected_index>0:A.selected_index-=1;A.bl()
 		@A.kb.add('down')
 		def C(event):
-			if A.selected_index<len(A.options)-1:A.selected_index+=1;A.bo()
+			if A.selected_index<len(A.options)-1:A.selected_index+=1;A.bl()
 		@A.kb.add('enter')
 		def D(event):A.callback(A.selected_index)
 async def demo():

package/dist/python-src/src/tui/components/tscroll_panel.py

@@ -9,7 +9,7 @@ from prompt_toolkit.layout.controls import FormattedTextControl
 from prompt_toolkit.application.current import get_app
 from prompt_toolkit.formatted_text.ansi import ANSI
 class ScrollableFormattedLogControl(FormattedTextControl):
-	def __init__(A):A.lines=[];A.scroll_offset=0;A._height=0;A.last_count=0;A._selecting=_A;A._selection_start=_B;A._selection_end=_B;super().__init__(A.bq,focusable=_C,show_cursor=_A)
+	def __init__(A):A.lines=[];A.scroll_offset=0;A._height=0;A.last_count=0;A._selecting=_A;A._selection_start=_B;A._selection_end=_B;super().__init__(A.bs,focusable=_C,show_cursor=_A)
 	def clear(A):A.lines=[];A.scroll_offset=0;A.last_count=0;A._height=0
 	def append_text(A,ansi_text):B=ansi_text.splitlines();A.lines.extend(B);A.last_count=len(B);A.bt()
 	def update_last(A,ansi_text):
@@ -23,7 +23,7 @@ class ScrollableFormattedLogControl(FormattedTextControl):
 		if A._height:A.scroll_offset=max(0,len(A.lines)-A._height)
 	def br(A,amount):
 		if A._height:B=max(0,len(A.lines)-A._height);A.scroll_offset=max(0,min(A.scroll_offset+amount,B))
-	def bq(A):
+	def bs(A):
 		R=A._height or 100;S=A.lines[A.scroll_offset:A.scroll_offset+R];C=[]
 		for(T,U)in enumerate(S):
 			L=A.scroll_offset+T;M=ANSI(U).__pt_formatted_text__()
@@ -76,9 +76,9 @@ log_window=Window(content=log_control,wrap_lines=_A)
 frame=log_window
 kb=KeyBindings()
 @kb.add('c-c')
-def bs(event):log_control.copy_selection_to_clipboard()
+def bq(event):log_control.copy_selection_to_clipboard()
 @kb.add('q')
-def bs(event):event.app.exit()
+def bq(event):event.app.exit()
 style=Style.from_dict({'frame.border':'#888888','frame.title':'bold','log':'#ffffff','selection':'reverse'})
 layout=Layout(HSplit([frame]))
 app=Application(layout=layout,key_bindings=kb,mouse_support=_C,full_screen=_C,style=style)