@brandon_9527/tcode 1.0.6 → 1.0.7

package/dist/python-src/src/managers/manager_skill.py

@@ -0,0 +1,121 @@
+_C='managed'
+_B='user'
+_A='project'
+from typing import Optional,Dict,Any,List,Tuple
+from dataclasses import dataclass,field
+from pathlib import Path
+import logging,yaml,re,os
+logger=logging.getLogger(__name__)
+@dataclass
+class Skill:
+	name:str;description:str;content:str;path:Path;license:Optional[str]=None;allowed_tools:List[str]=field(default_factory=list);metadata:Dict[str,Any]=field(default_factory=dict);location:str=_A
+	@classmethod
+	def from_skill_md(C,skill_md_path,location=_A):
+		B=skill_md_path
+		if not B.exists():return
+		F=B.read_text(encoding='utf-8');A,G=C.br(F)
+		if not A:return
+		D=A.get('name');E=A.get('description')
+		if not D or not E:return
+		return C(name=D,description=E,content=G.strip(),path=B.parent,license=A.get('license'),allowed_tools=A.get('allowed-tools',[])or[],metadata=A.get('metadata',{})or{},location=location)
+	@staticmethod
+	def br(content):
+		A=content;C='^---\\s*\\n(.*?)\\n---\\s*\\n(.*)$';B=re.match(C,A,re.DOTALL)
+		if not B:return{},A
+		D=B.group(1);E=B.group(2)
+		try:F=yaml.safe_load(D)or{}
+		except yaml.YAMLError:return{},A
+		return F,E
+	def get_prompt(A):B=f"Loading: {A.name} \nBase directory: {A.path}\n        \n        ";return B+A.content
+	def to_xml(A):return f"<skill>\n<name>{A.name}</name>\n<description>{A.description}</description>\n<location>{A.location}</location>\n</skill>"
+	def __repr__(A):return f"Skill(name={A.name!r}, location={A.location!r})"
+class SkillManager:
+	SKILL_DIRS=['.claude/skills','.autodev/skills'];SKILL_FILE='SKILL.md'
+	def __init__(A,project_root=None,home_path=Path.home()):C=home_path;B=project_root;A._skills={};A._skills_by_location={_A:[],_B:[],_C:[]};A.project_root=Path(B)if B else Path.cwd();A.home_dir=Path(C)if C else Path.home()
+	def register(B,skill):
+		A=skill;C=B._skills.get(A.name)
+		if C:
+			D={_A:0,_B:1,_C:2};E=D.get(C.location,99);F=D.get(A.location,99)
+			if F>=E:return False
+		B._skills[A.name]=A;B._skills_by_location[A.location].append(A);return True
+	def get(A,name):return A._skills.get(name)
+	def exists(A,name):return name in A._skills
+	def list_all(A):return list(A._skills.values())
+	def list_by_location(A,location):return A._skills_by_location.get(location,[])
+	def clear(A):
+		A._skills.clear()
+		for B in A._skills_by_location:A._skills_by_location[B].clear()
+	def generate_skills_prompt(E,char_budget=10000):
+		C=E.list_all()
+		if not C:return'Execute a skill within the main conversation.\n\nNo skills are currently available. Skills can be added to:\n- .claude/skills/ (project-level)\n- ~/.claude/skills/ (user-level)\n- .minion/skills/ (project-level)\n- ~/.minion/skills/ (user-level) \n    '
+		A=[];D=0
+		for F in C:
+			B=F.to_xml()
+			if D+len(B)>char_budget:break
+			A.append(B);D+=len(B)
+		if not A:return'Execute a skill within the main conversation.\n\nNo skills are currently available. Skills can be added to:\n- .claude/skills/ (project-level)\n- ~/.claude/skills/ (user-level)\n- .minion/skills/ (project-level)\n- ~/.minion/skills/ (user-level) \n'
+		G='\n'.join(A);return f'''Execute a skill within the main conversation.
+
+<skills_instructions>
+When users ask you to perform tasks, check if any of the available skills below can help complete the task more effectively. Skills provide specialized capabilities and domain knowledge.
+
+How to use skills:
+- Invoke skills using skill tool with the skill name only (no arguments)
+- When you invoke a skill, you will see <command-message>The "{{name}}" skill is loading</command-message>
+- The skill\'s prompt will expand and provide detailed instructions on how to complete the task
+- Base directory provided in output for resolving bundled resources (references/, scripts/, assets/)
+
+Important:
+- Only use skills listed in <available_skills> below
+- Do not invoke a skill that is already running
+</skills_instructions>
+
+<available_skills>
+{G}
+</available_skills>'''
+	def __len__(A):return len(A._skills)
+	def __contains__(A,name):return name in A._skills
+	def __iter__(A):return iter(A._skills.values())
+	def get_search_paths(A):
+		B=[]
+		for C in A.SKILL_DIRS:D=A.project_root/C;B.append((D,_A))
+		for C in A.SKILL_DIRS:E=A.home_dir/C;B.append((E,_B))
+		return B
+	def discover_skills(D,skills_dir):
+		A=skills_dir
+		if not A.exists()or not A.is_dir():return[]
+		B=[]
+		for C in A.iterdir():
+			if C.is_dir():
+				E=C/D.SKILL_FILE
+				if E.exists():B.append(E)
+				else:
+					for F in C.iterdir():
+						if F.is_dir():
+							G=F/D.SKILL_FILE
+							if G.exists():B.append(G)
+		return B
+	def load_skill(D,skill_md_path,location):
+		A=skill_md_path
+		try:
+			B=Skill.from_skill_md(A,location)
+			if B:logger.debug(f"Loaded skill: {B.name} from {A}")
+			else:logger.warning(f"Failed to parse skill: {A}")
+			return B
+		except Exception as C:logger.error(f"Error loading skill from {A}: {C}");return
+	def load_all(A):
+		for(D,C)in A.get_search_paths():
+			E=A.discover_skills(D)
+			for F in E:
+				B=A.load_skill(F,C)
+				if B:
+					G=A.register(B)
+					if G:logger.info(f"Registered skill: {B.name} ({C})")
+					else:logger.debug(f"Skipped skill {B.name} - already registered from high priority location")
+		return A
+	def reload(A):A.clear();return A.load_all()
+def main():
+	A=SkillManager(project_root=Path.cwd());A.load_all();C=A.list_all();print(f"Loaded {len(C)} skills");B=A.get('arxiv-search')
+	if B:print(B.get_prompt())
+	D=A.generate_skills_prompt(char_budget=5000);print(D);A.reload()
+if __name__=='__main__':main()

package/dist/python-src/src/managers/sandbox.py

@@ -21,14 +21,14 @@ sys.path.append(PROJ_PATH)
 logger=logging.getLogger(__file__)
 class Container:
 	def __init__(A,user_id=_C,workspace_path=_B,dockerfile_path='.',image_name=_J,container_basename='sandbox'):A.client=docker.from_env();A.container_basename=container_basename;A.workspace_path=workspace_path;A.dockerfile_path=dockerfile_path;A.image_name=image_name;A.sock=_B;A.queue=_B;A.container=_B;A.user_id=user_id;A.images={_G:'ubuntu:22.04',_H:'python:3.12-slim','node':'node:20-bullseye-slim',_C:_J}
-	def br(A):
+	def bs(A):
 		try:
 			B=A.client.images.get(A.image_name)
 			if B:logger.info(f"Image {A.image_name} exists.");return _A
 			else:logger.info(f"Image {A.image_name} not exists.");return _E
 		except docker.errors.ImageNotFound:logger.error(f"Image {A.image_name} not exists.");return _E
 		except Exception as C:logger.error(f"检查镜像时出错: {str(C)}");return _E
-	def bs(B):
+	def bt(B):
 		F='arm64';print('🔧 Building Docker image...')
 		try:
 			C=platform.machine()
@@ -44,7 +44,7 @@ class Container:
 		except Exception as D:logger.error(f"An unexpected error occurred: {D}");raise
 	def get_or_create_container(A,container_name,image_type=_C):
 		E='running';C=image_type;B=container_name;D=_F;F=A.workspace_path;B=f"{A.container_basename}_{A.user_id}"
-		if C==_C and not A.br():A.bs()
+		if C==_C and not A.bs():A.bt()
 		try:
 			A.container=A.client.containers.get(B)
 			if A.container.status==E:logger.info(f"容器 {B} 已存在且运行中，直接复用")

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

@@ -19,11 +19,11 @@ logger=logging.getLogger(__name__)
 class DynamicContentState(AgentState):review_contents:NotRequired[Annotated[dict[str,str],PrivateStateAttr]];security_contents:NotRequired[Annotated[dict[str,str],PrivateStateAttr]]
 class DynamicContentUpdate(TypedDict):review_contents:dict[str,str];security_contents:dict[str,str]
 REVIEW_SYSTEM_PROMPT=' \n<review_rules>\n  {review_rules}\n</review_rules>\n\n<review_guidelines>\n  如果用户要求对项目进行 Review ，请严格依照 review_rules 标签中的规则 对项目进行 Review 。\n</review_guidelines>\n'
-SECURITY_SYSTEM_PROMPT=' \n<security_rules>\n  {security_rules}\n</security_rules>\n\n<<security_guidelines>\n  如果用户要求对项目进行 安全检查 ，请严格依照 security_rules 标签中的规则 对项目进行安全检查 。\n</security_guidelines>\n'
+SECURITY_SYSTEM_PROMPT=' \n<security_rules>\n  {security_rules}\n</security_rules>\n\n<security_guidelines>\n  如果用户要求对项目进行 安全检查 ，请严格依照 security_rules 标签中的规则 对项目进行安全检查 。\n</security_guidelines>\n'
 class DynamicContentMiddleware(AgentMiddleware[DynamicContentState,ContextT,ResponseT]):
 	state_schema=DynamicContentState
 	def __init__(A,*,home_path=os.path.expanduser('~'),project_path=os.getcwd()):A.home_path=home_path;A.project_path=project_path
-	def cu(G,prompt,content_type,contents):
+	def cv(G,prompt,content_type,contents):
 		E='( No rules loaded)';C=contents;B=content_type;A=prompt
 		if not C:return A.format(**{B:E})
 		D=[f"{A}\n{C[A]}"for A in C]
@@ -58,7 +58,7 @@ class DynamicContentMiddleware(AgentMiddleware[DynamicContentState,ContextT,Resp
 			with open(H,_A,encoding=_B)as B:F[H]=B.read()
 		return DynamicContentUpdate(review_contents=C,security_contents=F)
 	def modify_request(C,request):
-		B=request;F=B.state.get(_C,{});G=B.state.get(_D,{});D=C.cu(REVIEW_SYSTEM_PROMPT,'review_rules',F);E=C.cu(SECURITY_SYSTEM_PROMPT,'security_rules',G);A=B.system_message
+		B=request;F=B.state.get(_C,{});G=B.state.get(_D,{});D=C.cv(REVIEW_SYSTEM_PROMPT,'review_rules',F);E=C.cv(SECURITY_SYSTEM_PROMPT,'security_rules',G);A=B.system_message
 		if D:A=append_to_system_message(A,D)
 		if E:A=append_to_system_message(A,E)
 		return B.override(system_message=A)

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

@@ -21,12 +21,12 @@ class ReviewConfig(TypedDict):action_name:str;allowed_decisions:list[DecisionTyp
 class HITLRequest(TypedDict):action_requests:list[ActionRequest];review_configs:list[ReviewConfig]
 class HumanInTheLoopMiddleware(AgentMiddleware[StateT,ContextT]):
 	def __init__(A,interrupt_on=[],*,description_perfix='Tool execution requires approval'):super().__init__();A.interrupt_on=interrupt_on;A.description_prefix=description_perfix
-	def cd(D,tool_call):A=tool_call;B=A[_E];C=A[_F];E=A['id'];F=f"{D.description_prefix}\n\nTool: {B}\nArgs: {C}";G=ActionRequest(name=B,id=E,args=C,description=F);H=ReviewConfig(action_name=B,allowed_decisions=[_B,_A,_C]);return G,H
+	def ce(D,tool_call):A=tool_call;B=A[_E];C=A[_F];E=A['id'];F=f"{D.description_prefix}\n\nTool: {B}\nArgs: {C}";G=ActionRequest(name=B,id=E,args=C,description=F);H=ReviewConfig(action_name=B,allowed_decisions=[_B,_A,_C]);return G,H
 	async def awrap_tool_call(E,request,handler):
 		C=handler;A=request;F=A.tool_call['id'];D=A.tool_call[_E];R=A.tool_call[_F];G=A.runtime;H=G.context;I=H.tool_mode
 		if I==_A:return await C(A)
 		if len(E.interrupt_on)>0 and not D.strip().lower()in E.interrupt_on:return await C(A)
-		J,K=E.cd(A.tool_call);L=HITLRequest(action_requests=[J],review_configs=[K]);M=interrupt(L);B=M[_G][0][_D]
+		J,K=E.ce(A.tool_call);L=HITLRequest(action_requests=[J],review_configs=[K]);M=interrupt(L);B=M[_G][0][_D]
 		if B==_B:return await C(A)
 		if B==_A:return await C(A)
 		if B==_C:N=f"User rejected the tool call for `{D}` with id {F}";O=ToolMessage(content=N,name=D,tool_call_id=F,status='error');return O
@@ -35,7 +35,7 @@ class HumanInTheLoopMiddleware(AgentMiddleware[StateT,ContextT]):
 		C=handler;A=request;F=A.tool_call['id'];D=A.tool_call[_E];R=A.tool_call[_F];H=A.runtime;I=H.context;J=I.tool_mode
 		if J==_A:return C(A)
 		if len(E.interrupt_on)>0 and not D.strip().lower()in E.interrupt_on:return C(A)
-		K,L=E.cd(A.tool_call);M=HITLRequest(action_requests=[K],review_configs=[L]);G=interrupt(M);print(f"[HITL] -> Human decision: {G}");B=G[_G][0][_D]
+		K,L=E.ce(A.tool_call);M=HITLRequest(action_requests=[K],review_configs=[L]);G=interrupt(M);print(f"[HITL] -> Human decision: {G}");B=G[_G][0][_D]
 		if B==_B:return C(A)
 		if B==_A:return C(A)
 		if B==_C:N=f"User rejected the tool call for `{D}` with id {F}";O=ToolMessage(content=N,name=D,tool_call_id=F,status='error');return O

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

@@ -14,7 +14,7 @@ MEMORY_SYSTEM_PROMPT=' \n<agent_memory>\n{agent_memory}\n</agent_memory>\n\n<mem
 class MemoryMiddleware(AgentMiddleware[MemoryState,ContextT,ResponseT]):
 	state_schema=MemoryState
 	def __init__(A,*,home_path=os.path.expanduser('~'),project_path=os.getcwd()):A.home_path=home_path;A.project_path=project_path
-	def ce(D,contents):
+	def cf(D,contents):
 		C='(No memory loaded)';A=contents
 		if not A:return MEMORY_SYSTEM_PROMPT.format(agent_memory=C)
 		B=[f"{B}\n{A[B]}"for B in D.sources if A.get(B)]
@@ -39,6 +39,6 @@ class MemoryMiddleware(AgentMiddleware[MemoryState,ContextT,ResponseT]):
 				H=f"Failed to download {C}: {A.error}";raise ValueError(H)
 			if A.content is not None:E[C]=A.content.decode(_B);logger.debug('Loaded memory from: %s',C)
 		return MemoryStateUpdate(memory_contents=E)
-	def modify_request(B,request):A=request;C=A.state.get(_A,{});D=B.ce(C);E=append_to_system_message(A.system_message,D);return A.override(system_message=E)
+	def modify_request(B,request):A=request;C=A.state.get(_A,{});D=B.cf(C);E=append_to_system_message(A.system_message,D);return A.override(system_message=E)
 	def wrap_model_call(A,request,handler):B=A.modify_request(request);return handler(B)
 	async def awrap_model_call(A,request,handler):B=A.modify_request(request);return await handler(B)

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

@@ -0,0 +1,27 @@
+from collections.abc import Awaitable,Callable
+from langchain.agents.middleware.types import AgentMiddleware,ModelRequest,ModelResponse
+from langchain.tools import BaseTool,ToolRuntime
+from langchain_core.language_models import BaseChatModel
+from langchain_core.messages import HumanMessage,ToolMessage
+from langchain_core.runnables import Runnable
+from langchain_core.tools import StructuredTool
+from langgraph.types import Command
+from langchain_core.tools import tool
+from src.managers.manager_skill import SkillManager
+def generate_skill_tool(manager):
+	B=manager;A=' Execute a skill within the main conversation.\n\nSkills are folders of instructions, scripts, and resources that Claude loads\ndynamically to improve performance on specialized tasks.\n\nUsage:\n- Invoke skills using this tool with the skill name only (no arguments)\n- When you invoke a skill, its prompt will expand and provide detailed instructions\n- Only use skills listed in <available_skills> in the system prompt\n\nImportant:\n- Only use skills that are listed as available\n- Do not use skills that are not listed as available\n'
+	@tool(description=A)
+	def C(skill):
+		D='success';C=skill;A=B.get(C)
+		if A is None:E=[A.name for A in B.list_all()];return{D:False,'error':f"Unknown skill: {C}",'available_skills':E[:10],'hint':'Use one of the available skills listed above'}
+		F=A.get_prompt();return{D:True,'skill_name':A.name,'skill_description':A.description,'skill_location':A.location,'skill_path':str(A.path),'prompt':F,'message':f'The "{A.name}" skill is loading','allowed_tools':A.allowed_tools}
+	return C
+class SkillMiddleware(AgentMiddleware):
+	def __init__(A,*,home_path,workspace):B=home_path;super().__init__();A.workspace=workspace;A.home_path=B;A.manager=SkillManager(A.workspace,B);A.manager.load_all();C=generate_skill_tool(A.manager);A.tools=[C]
+	def modify_request(D,request):
+		A=request;B=D.manager.generate_skills_prompt(5000)
+		if A.system_prompt:C=A.system_prompt+'\n\n'+B
+		else:C=B
+		return A.override(system_prompt=C)
+	def wrap_model_call(A,request,handler):B=A.modify_request(request);return handler(B)
+	async def awrap_model_call(A,request,handler):B=A.modify_request(request);return await handler(B)

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

@@ -79,7 +79,7 @@ async def astream_handler(stream,runtime):
 						if isinstance(A,AIMessage)and len(A.tool_calls)==0:C=A
 						if isinstance(A,ToolMessage):G=A.name;N=A.content;H=A.tool_call_id;I={_E:'tool_result',_G:D.tool_call_id,_H:H,_I:'',_C:G,_D:{},_J:N,_K:_O};J(json.dumps(I,indent=2,ensure_ascii=_L))
 	O={_B:[C]if C and isinstance(C,AIMessage)else[]};return O
-def bz(*,default_model,default_tools,default_middleware,default_interrupt_on,subagents,general_purpose_agent,task_description=_A):
+def cd(*,default_model,default_tools,default_middleware,default_interrupt_on,subagents,general_purpose_agent,task_description=_A):
 	G='Tool call ID is required for subagent invocation';F='custom';A=task_description;B,H=cc(default_model=default_model,default_tools=default_tools,default_middleware=default_middleware,default_interrupt_on=default_interrupt_on,subagents=subagents,general_purpose_agent=general_purpose_agent);C='\n'.join(H)
 	def D(result,tool_call_id):A=result;B={A:B for(A,B)in A.items()if A not in _EXCLUDED_STATE_KEYS};C=A[_B][-1];D=I(C);return Command(update={**B,_B:[ToolMessage(D,tool_call_id=tool_call_id)]})
 	def I(msg):
@@ -110,7 +110,7 @@ def bz(*,default_model,default_tools,default_middleware,default_interrupt_on,sub
 		return D(L,A.tool_call_id)
 	return StructuredTool.from_function(name='task',func=J,coroutine=K,description=A)
 class SubAgentMiddleware(AgentMiddleware):
-	def __init__(A,*,default_model,default_tools=_A,default_middleware=_A,default_interrupt_on=_A,subagents=_A,system_prompt=TASK_SYSTEM_PROMPT,general_purpose_agent=True,task_description=_A):super().__init__();A.system_prompt=system_prompt;B=bz(default_model=default_model,default_tools=default_tools or[],default_middleware=default_middleware,default_interrupt_on=default_interrupt_on,subagents=subagents or[],general_purpose_agent=general_purpose_agent,task_description=task_description);A.tools=[B]
+	def __init__(A,*,default_model,default_tools=_A,default_middleware=_A,default_interrupt_on=_A,subagents=_A,system_prompt=TASK_SYSTEM_PROMPT,general_purpose_agent=True,task_description=_A):super().__init__();A.system_prompt=system_prompt;B=cd(default_model=default_model,default_tools=default_tools or[],default_middleware=default_middleware,default_interrupt_on=default_interrupt_on,subagents=subagents or[],general_purpose_agent=general_purpose_agent,task_description=task_description);A.tools=[B]
 	def wrap_model_call(B,request,handler):
 		C=handler;A=request
 		if B.system_prompt is not _A:D=A.system_prompt+'\n\n'+B.system_prompt if A.system_prompt else B.system_prompt;return C(A.override(system_prompt=D))

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 ck(model):
+def cm(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,61 +44,61 @@ 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.ch(B,N)for B in B];A.trigger=I;G=I
-		else:J=A.ch(B,N);A.trigger=J;G=[J]
-		A._trigger_conditions=G;A.keep=A.ch(F,'keep')
-		if H is count_tokens_approximately:A.token_counter=ck(A.model)
+		elif isinstance(B,list):I=[A.cq(B,N)for B in B];A.trigger=I;G=I
+		else:J=A.cq(B,N);A.trigger=J;G=[J]
+		A._trigger_conditions=G;A.keep=A.cq(F,'keep')
+		if H is count_tokens_approximately:A.token_counter=cm(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.cp()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 cl(B,state):A=state.get(_G,_F);return bool(A)
+		if K and A.cj()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 ct(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.cq(B);E=A.token_counter(B);F=A.cl(C);G=A.co(B,E)
+		C=state;A=self;B=C[_B];A.co(B);E=A.token_counter(B);F=A.ct(C);G=A.ch(B,E)
 		if not G and not F:return
-		D=A.cn(B)
+		D=A.cu(B)
 		if D<=0:return
-		H,I=A.cg(B,D);J=A.cm(H);K=A.cj(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
+		H,I=A.cn(B,D);J=A.ck(H);K=A.cl(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.cq(B);E=A.token_counter(B);F=A.cl(C);G=A.co(B,E)
+		C=state;A=self;B=C[_B];A.co(B);E=A.token_counter(B);F=A.ct(C);G=A.ch(B,E)
 		if not G and not F:return
-		D=A.cn(B)
+		D=A.cu(B)
 		if D<=0:return
-		H,I=A.cg(B,D);J=await A._acreate_summary(H);K=A.cj(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
-	def cs(B,messages,threshold):
+		H,I=A.cn(B,D);J=await A._acreate_summary(H);K=A.cl(J);return{_B:[RemoveMessage(id=REMOVE_ALL_MESSAGES),*K,*I],_G:_F}
+	def ci(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 co(A,messages,total_tokens):
+	def ch(A,messages,total_tokens):
 		F=total_tokens;E=messages
 		if not A._trigger_conditions:return _F
 		for(B,C)in A._trigger_conditions:
 			if B==_B and len(E)>=C:return _C
 			if B==_E and F>=C:return _C
-			if B==_E and A.cs(E,C):return _C
+			if B==_E and A.ci(E,C):return _C
 			if B==_D:
-				G=A.cp()
+				G=A.cj()
 				if G is _A:continue
 				D=int(G*C)
 				if D<=0:D=1
 				if F>=D:return _C
-				if A.cs(E,D):return _C
+				if A.ci(E,D):return _C
 		return _F
-	def cn(A,messages):
+	def cu(A,messages):
 		B=messages;D,E=A.keep
 		if D in{_E,_D}:
-			C=A.cf(B)
+			C=A.cs(B)
 			if C is not _A:return C
-			return A.cr(B,_DEFAULT_MESSAGES_TO_KEEP)
-		return A.cr(B,cast('int',E))
-	def cf(C,messages):
+			return A.cp(B,_DEFAULT_MESSAGES_TO_KEEP)
+		return A.cp(B,cast('int',E))
+	def cs(C,messages):
 		A=messages
 		if not A:return 0
 		H,I=C.keep
 		if H==_D:
-			J=C.cp()
+			J=C.cj()
 			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.ct(A,B)
-	def cp(C):
+		return C.cr(A,B)
+	def cj(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 ch(context,parameter_name):
+	def cq(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 cj(summary):return[HumanMessage(content=f"Here is a summary of the conversation to date:\n\n{summary}",additional_kwargs={'lc_source':'summarization'})]
+	def cl(summary):return[HumanMessage(content=f"Here is a summary of the conversation to date:\n\n{summary}",additional_kwargs={'lc_source':'summarization'})]
 	@staticmethod
-	def cq(messages):
+	def co(messages):
 		for A in messages:
 			if A.id is _A:A.id=str(uuid.uuid4())
 	@staticmethod
-	def cg(conversation_messages,cutoff_index):B=cutoff_index;A=conversation_messages;C=A[:B];D=A[B:];return C,D
-	def cr(C,messages,messages_to_keep):
+	def cn(conversation_messages,cutoff_index):B=cutoff_index;A=conversation_messages;C=A[:B];D=A[B:];return C,D
+	def cp(C,messages,messages_to_keep):
 		B=messages_to_keep;A=messages
 		if len(A)<=B:return 0
-		D=len(A)-B;return C.ct(A,D)
+		D=len(A)-B;return C.cr(A,D)
 	@staticmethod
-	def ct(messages,cutoff_index):
+	def cr(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 cm(A,messages_to_sumarize):
+	def ck(A,messages_to_sumarize):
 		B=messages_to_sumarize
 		if not B:return _H
-		C=A.ci(B)
+		C=A.cg(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.ci(B)
+		C=A.cg(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 ci(A,messages):
+	def cg(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.bi(A)
-			if B.bd(C):return ContentType.JSON
+			C=B.bh(A)
+			if B.bi(C):return ContentType.JSON
 			return ContentType.SUCCESS
 		if A.startswith(FAILURE_PREFIX):return ContentType.ERROR
-		if B.bd(A):return ContentType.JSON
-		if B.bg(A):return ContentType.ERROR
-		if B.bc(A):return ContentType.MARKDOWN
+		if B.bi(A):return ContentType.JSON
+		if B.bd(A):return ContentType.ERROR
+		if B.bg(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.bf,ContentType.ERROR:A.bh,ContentType.JSON:A.bj,ContentType.MARKDOWN:A.bb,ContentType.TEXT:A.be};F=E.get(C,A.be);G=F(name,B,max_length);return FormattedResult(content_type=C,elements=G,success=D)
-	def bi(B,content):A=content.split('\n',2);return A[2].strip()if len(A)>2 else''
-	def bd(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.be,ContentType.ERROR:A.bc,ContentType.JSON:A.ba,ContentType.MARKDOWN:A.bj,ContentType.TEXT:A.bb};F=E.get(C,A.bb);G=F(name,B,max_length);return FormattedResult(content_type=C,elements=G,success=D)
+	def bh(B,content):A=content.split('\n',2);return A[2].strip()if len(A)>2 else''
+	def bi(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 bg(B,content):A=['Traceback (most recent call last)','Exception:','Error:'];return any(A in content for A in A)
-	def bc(C,content):A=content;B=['```','**','##','- **'];return A.startswith('#')or any(B in A for B in B)
-	def bf(B,name,content,max_length):A='green';C=B.ba(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✓",border_style=A)]
-	def bh(B,name,content,max_length):A='red';C=B.ba(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✗",border_style=A)]
-	def bj(B,name,content,max_length):
+	def bd(B,content):A=['Traceback (most recent call last)','Exception:','Error:'];return any(A in content for A in A)
+	def bg(C,content):A=content;B=['```','**','##','- **'];return A.startswith('#')or any(B in A for B in B)
+	def be(B,name,content,max_length):A='green';C=B.bf(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✓",border_style=A)]
+	def bc(B,name,content,max_length):A='red';C=B.bf(content,max_length);return[Panel(Text(C,style=A),title=f"📤 {name} ✗",border_style=A)]
+	def ba(B,name,content,max_length):
 		D=max_length;A=content;E=A
-		if A.startswith(SUCCESS_PREFIX):E=B.bi(A)
-		try:F=json.loads(E);C=json.dumps(F,indent=2,ensure_ascii=_A);C=B.ba(C,D);return[Text(f"📤 {name} ✓",style=_B),Syntax(C,'json',theme='monokai',line_numbers=_A)]
-		except(json.JSONDecodeError,ValueError):return B.be(name,A,D)
-	def bb(A,name,content,max_length):B=A.ba(content,max_length);return[Panel(Markdown(B),title=f"📤 {name}",border_style='cyan dim')]
-	def be(A,name,content,max_length):B=A.ba(content,max_length);return[Text(f"📤 {name}:",style=_B),Text(f"   {B}",style='dim')]
-	def ba(A,content,max_length):return truncate(content,max_length)
+		if A.startswith(SUCCESS_PREFIX):E=B.bh(A)
+		try:F=json.loads(E);C=json.dumps(F,indent=2,ensure_ascii=_A);C=B.bf(C,D);return[Text(f"📤 {name} ✓",style=_B),Syntax(C,'json',theme='monokai',line_numbers=_A)]
+		except(json.JSONDecodeError,ValueError):return B.bb(name,A,D)
+	def bj(A,name,content,max_length):B=A.bf(content,max_length);return[Panel(Markdown(B),title=f"📤 {name}",border_style='cyan dim')]
+	def bb(A,name,content,max_length):B=A.bf(content,max_length);return[Text(f"📤 {name}:",style=_B),Text(f"   {B}",style='dim')]
+	def bf(A,content,max_length):return truncate(content,max_length)

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

@@ -0,0 +1,45 @@
+from __future__ import annotations
+_A=None
+import argparse,json,sys
+from datetime import datetime,timedelta,timezone
+from pathlib import Path
+from.pricing import ModelPricing
+from.tracker import CostTracker,UsageRecord,BudgetGuard
+from.report import CostReport
+def parse_since(s):
+	A=datetime.now(timezone.utc);s=s.strip().lower()
+	if s.endswith('d'):return A-timedelta(days=int(s[:-1]))
+	if s.endswith('w'):return A-timedelta(weeks=int(s[:-1]))
+	if s.endswith('m'):return A-timedelta(days=30*int(s[:-1]))
+	if s.endswith('h'):return A-timedelta(hours=int(s[:-1]))
+	return datetime.fromisoformat(s)
+def cmd_add(args,tracker):
+	B=tracker;A=args;D=B.add_call(model=A.model,input_tokens=A.input,output_tokens=A.output,tag=A.tag);E=ModelPricing()
+	try:F=E.comp(A.model,A.input,A.output);C=f"${F:.6f}"
+	except KeyError:C='unknown (model not in pricing DB)'
+	print(f"✓ Logged: {A.model}  {A.input:,} in / {A.output:,} out  cost={C}  tag={A.tag}");print(f"  record_id={D.record_id}  db={B.db_path}")
+def cmd_report(args,tracker):
+	A=args;B=parse_since(A.since)if A.since else _A;C=CostReport(tracker)
+	if A.format=='json':print(C.to_json(since=B,period=A.period))
+	else:print(C.to_text(since=B,period=A.period))
+def cmd_models(args):
+	C=ModelPricing();B=C.list_models(provider=args.provider)
+	if not B:print('No models found.');return
+	print(f"\n{'Model':<45} {'Provider':<12} {'Input $/1M':>12} {'Output $/1M':>12}");print('─'*85)
+	for A in B:print(f"  {A.model:<43} {A.provider:<12} ${A.input_per_1m:>10.3f}  ${A.output_per_1m:>10.3f}")
+	print()
+def cmd_budget(args,tracker):
+	C=tracker;A=args
+	if A.budget_cmd=='set':D=BudgetGuard(tag=A.tag,limit_usd=A.limit);C.set_budget(D);print(f"✓ Budget set: tag={A.tag}  limit=${A.limit:.2f}");B=C.total_cost(tag=A.tag);E=B/A.limit*100 if A.limit>0 else 0;print(f"  Current spend: ${B:.4f} ({E:.1f}% of budget)")
+	elif A.budget_cmd=='check':B=C.total_cost(tag=A.tag);print(f"  Tag '{A.tag}': ${B:.4f} spent")
+	else:print('Usage: llm-cost budget set --tag TAG --limit AMOUNT')
+def main():
+	P='budget';O='models';N='daily';M='text';L='report';K='add';H='--tag';B=True;C=argparse.ArgumentParser(prog='llm-cost',description='LLM cost tracker — log API calls, report spending, guard budgets.');C.add_argument('--db',type=Path,default=_A,help='Path to usage DB (default: ~/.config/llm-cost-tracker/usage.jsonl)');E=C.add_subparsers(dest='command');D=E.add_parser(K,help='Log an LLM API call');D.add_argument('--model',required=B,help='Model name, e.g. gpt-4o');D.add_argument('--input',type=int,required=B,help='Input token count');D.add_argument('--output',type=int,required=B,help='Output token count');D.add_argument(H,default='default',help='Project/tag label');D.add_argument('--timestamp',default=_A,help='ISO timestamp (default: now)');F=E.add_parser(L,help='Show cost report');F.add_argument('--since',default='30d',help='e.g. 7d, 1w, 1m, or ISO date');F.add_argument('--format',choices=[M,'json'],default=M);F.add_argument('--period',choices=[N,'weekly','monthly'],default=N);Q=E.add_parser(O,help='List pricing database');Q.add_argument('--provider',default=_A,help='Filter by provider');R=E.add_parser(P,help='Budget guard commands');I=R.add_subparsers(dest='budget_cmd');J=I.add_parser('set',help='Set a budget for a tag');J.add_argument(H,required=B);J.add_argument('--limit',type=float,required=B,help='Budget in USD');S=I.add_parser('check',help='Check spending for a tag');S.add_argument(H,required=B);A=C.parse_args()
+	if not A.command:C.print_help();sys.exit(0)
+	G=CostTracker(db_path=A.db)
+	if A.command==K:cmd_add(A,G)
+	elif A.command==L:cmd_report(A,G)
+	elif A.command==O:cmd_models(A)
+	elif A.command==P:cmd_budget(A,G)
+	else:C.print_help()
+if __name__=='__main__':main()

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

@@ -0,0 +1,39 @@
+_I='qwen'
+_H='mistral'
+_G='avg hosted price'
+_F='meta'
+_E='cohere'
+_D=None
+_C='google'
+_B='anthropic'
+_A='openai'
+from dataclasses import dataclass
+from typing import Optional
+@dataclass(frozen=True)
+class ModelRate:provider:str;model:str;input_per_1m:float;output_per_1m:float;notes:str=''
+_PRICING={}
+def cy(*C):
+	for A in C:
+		_PRICING[A.model]=A;B=A.model.split('/')[-1]if'/'in A.model else _D
+		if B and B not in _PRICING:_PRICING[B]=A
+cy(ModelRate(_A,'gpt-4o',2.5,1e1),ModelRate(_A,'gpt-4o-mini',.15,.6),ModelRate(_A,'gpt-4o-2024-11-20',2.5,1e1),ModelRate(_A,'gpt-4-turbo',1e1,3e1),ModelRate(_A,'gpt-4',3e1,6e1),ModelRate(_A,'gpt-3.5-turbo',.5,1.5),ModelRate(_A,'o1',15.,6e1),ModelRate(_A,'o1-mini',3.,12.),ModelRate(_A,'o3-mini',1.1,4.4),ModelRate(_A,'o3',1e1,4e1),ModelRate(_B,'claude-3-5-sonnet-20241022',3.,15.),ModelRate(_B,'claude-3-5-haiku-20241022',.8,4.),ModelRate(_B,'claude-3-opus-20240229',15.,75.),ModelRate(_B,'claude-3-sonnet-20240229',3.,15.),ModelRate(_B,'claude-3-haiku-20240307',.25,1.25),ModelRate(_B,'claude-sonnet-4-5',3.,15.),ModelRate(_B,'claude-sonnet-4-6',3.,15.),ModelRate(_B,'claude-opus-4',15.,75.),ModelRate(_C,'gemini-1.5-pro',3.5,10.5,'up to 128k ctx'),ModelRate(_C,'gemini-1.5-flash',.075,.3),ModelRate(_C,'gemini-2.0-flash',.1,.4),ModelRate(_C,'gemini-2.0-flash-lite',.075,.3),ModelRate(_C,'gemini-2.0-pro',3.5,10.5),ModelRate(_E,'command-r-plus',2.5,1e1),ModelRate(_E,'command-r',.15,.6),ModelRate(_E,'command-r7b',.0375,.15),ModelRate(_F,'llama-3.3-70b-instruct',.23,.4,_G),ModelRate(_F,'llama-3.1-405b-instruct',3.,3.,_G),ModelRate(_F,'llama-3.1-70b-instruct',.2,.2,_G),ModelRate(_H,'mistral-large-2411',2.,6.),ModelRate(_H,'mistral-small-2501',.1,.3),ModelRate(_H,'codestral-2501',.1,.3),ModelRate(_I,'qwen-plus',2.,6.),ModelRate(_I,'qwen-max-3',2.,6.),ModelRate(_I,'qwen-max-3.5',2.,6.))
+class ModelPricing:
+	def __init__(A,custom_rates=_D):
+		B=custom_rates;A._db=dict(_PRICING)
+		if B:A._db.update(B)
+	def get_rate(A,model):
+		C=model;B=C.lower().strip()
+		if B in A._db:return A._db[B]
+		for(D,E)in A._db.items():
+			if B in D or D in B:return E
+		raise KeyError(f"Unknown model: {C!r}. Known models: {sorted(A._db)}")
+	def compute_cost(B,model,input_tokens,output_tokens):A=B.get_rate(model);return A.input_per_1m*input_tokens/1000000+A.output_per_1m*output_tokens/1000000
+	def list_models(F,provider=_D):
+		B=provider;C=set();D=[]
+		for A in F._db.values():
+			E=A.provider,A.model
+			if E in C:continue
+			C.add(E)
+			if B is _D or A.provider.lower()==B.lower():D.append(A)
+		return sorted(D,key=lambda r:(r.provider,r.model))
+	def add_custom(A,rate):A._db[rate.model]=rate