code-puppy 0.0.154__py3-none-any.whl → 0.0.156__py3-none-any.whl

{code_puppy-0.0.154.dist-info → code_puppy-0.0.156.dist-info}/RECORD

@@ -1,43 +1,42 @@
 code_puppy/__init__.py,sha256=ehbM1-wMjNmOXk_DBhhJECFyBv2dRHwwo7ucjHeM68E,107
 code_puppy/__main__.py,sha256=pDVssJOWP8A83iFkxMLY9YteHYat0EyWDQqMkKHpWp4,203
-code_puppy/agent.py,sha256=rWqJ2849HSrWPX-IzI31v2sH6_j5pnBRfuum3yxUNt8,7256
+code_puppy/agent.py,sha256=hZOrJJJDwtLtb82b6Zyt5CZb8RbzDtvHMIBUPVYsgOM,8133
 code_puppy/callbacks.py,sha256=6wYB6K_fGSCkKKEFaYOYkJT45WaV5W_NhUIzcvVH_nU,5060
-code_puppy/config.py,sha256=oJCVEp6cK4u4ynKe-1TGlpB_TW1Q4pp14oFhtGEWhbk,16165
+code_puppy/config.py,sha256=nD-1Ftwzd1LgZIi5Pi0P4_8DvPkZiOwFXQ5UuYFMO6E,16207
 code_puppy/http_utils.py,sha256=BAvt4hed7fVMXglA7eS9gOb08h2YTuOyai6VmQq09fg,3432
-code_puppy/main.py,sha256=Vv5HSJnkgZhCvvOoXrJ2zqM5P-i47-RcYAU00Z1Pfx0,21733
+code_puppy/main.py,sha256=tYLfhUjPTJ-4S1r-pr-jSbn6kIU1iYvt2Z8lxI7zDFY,22220
 code_puppy/message_history_processor.py,sha256=aV-vcRcOQJPZPlrokB4CaLMxEU3Y4nDiabb9Ov_sJeU,15933
-code_puppy/model_factory.py,sha256=-_0kJ7qYmRTDk0v3mfp-UrYf0IaDLyG8IMBweDECwJg,11917
-code_puppy/models.json,sha256=dAfpMMI2EEeOMv0ynHSmMuJAYDLcZrs5gCLX3voC4-A,3252
+code_puppy/model_factory.py,sha256=0tJWwmKHSqNGEiakk1aGdUrlZyQy0QM4yShxHqxmXpA,11758
+code_puppy/models.json,sha256=jNLDAGhbASNAhul64H4SMS5V4Y6QKW09WcNej4I_4oc,4034
 code_puppy/reopenable_async_client.py,sha256=4UJRaMp5np8cbef9F0zKQ7TPKOfyf5U-Kv-0zYUWDho,8274
-code_puppy/round_robin_model.py,sha256=VWGCbK0-0rLpnrdxlmND5JsKXiNZZWFMIjF4_OL3STE,5594
+code_puppy/round_robin_model.py,sha256=SEN3VSwTgC5wHjx2sZsHQLPWOycf4jGwzB-EydgqkdY,5643
 code_puppy/state_management.py,sha256=o4mNBCPblRyVrNBH-992-1YqffgH6AKHU7iZRqgP1LI,5925
 code_puppy/status_display.py,sha256=F6eEAkGePDp4StM2BWj-uLLQTDGtJrf0IufzCeP1rRg,8336
-code_puppy/summarization_agent.py,sha256=-e6yUGZ22ahSaF0y7QhgVcQBfx5ktNUkPxBIWQfPaA4,3275
-code_puppy/token_utils.py,sha256=inLo-S2YERGA-JmV-nrSFN7KMswSfHxpawAuK6YiDHc,1982
+code_puppy/summarization_agent.py,sha256=kos4_YK-l_YjYRq4Fs4X5YoTUbmAcDhhPqefL-rdenI,3197
 code_puppy/version_checker.py,sha256=bjLDmgGPrl7XnYwX1u13O8uFlsfikV90PK6nbA9Z9QU,1150
 code_puppy/agents/__init__.py,sha256=SwtHGNG1GIgDBv7y3EGIXOXEWrG_Ou7fEknNgFbrHv8,594
 code_puppy/agents/agent_code_puppy.py,sha256=sbuQxLzlkMbPOyLbILbXOAHecRsxbFdQt13HJ_GEqQo,7972
-code_puppy/agents/agent_creator_agent.py,sha256=IUPLJDhhqIQ8NAjNTFQ_3Z5kZYYxVAUE0RPkcz90reY,20450
+code_puppy/agents/agent_creator_agent.py,sha256=jyOfqCa4d3Tqlv0ZODjK1J6cRwJjBxKyNa9RQSdsB44,23256
 code_puppy/agents/agent_manager.py,sha256=nXvro6fpX8KA-NedRoVJuhJW966trrePOrH4eAnqq40,17034
 code_puppy/agents/agent_orchestrator.json,sha256=Iqnc0p6ICoAlUTMkEsi1XXMXJi4pdxVnWZUMaih6s5o,1267
 code_puppy/agents/base_agent.py,sha256=XYSff6IQX9Z6C7hPVbN_YXC2xfjwd268e2jtG3ZGnVk,3450
-code_puppy/agents/json_agent.py,sha256=0j6_P1ppje7TsjaZIbxKn8meiuvoBngvjVLNdtCkGwc,4272
-code_puppy/agents/runtime_manager.py,sha256=t8F37mYX3txUqVoOrGwt0VR4Yc-M8AKIxSo22P0TOmg,9921
+code_puppy/agents/json_agent.py,sha256=y6AYE3Fx9LhmemcPzt46d7359MNnkGIjU83YBGNer2g,4533
+code_puppy/agents/runtime_manager.py,sha256=fUOBpmETo3wTyLc5wWBfGKSX1HFRQWSpuwxYAOyA-_8,10059
 code_puppy/command_line/__init__.py,sha256=y7WeRemfYppk8KVbCGeAIiTuiOszIURCDjOMZv_YRmU,45
-code_puppy/command_line/command_handler.py,sha256=1o9tKAGycpHFDBldYRAAvY5HJ6QAfikLPrXTEkfw37o,21137
+code_puppy/command_line/command_handler.py,sha256=KVZySqUMJwrRadKSUCcWa62FZ_lge5XjxD0Hebzr2EE,24520
 code_puppy/command_line/file_path_completion.py,sha256=gw8NpIxa6GOpczUJRyh7VNZwoXKKn-yvCqit7h2y6Gg,2931
 code_puppy/command_line/load_context_completion.py,sha256=6eZxV6Bs-EFwZjN93V8ZDZUC-6RaWxvtZk-04Wtikyw,2240
 code_puppy/command_line/meta_command_handler.py,sha256=80aK5JQOaqjt149qBmSsM02uy2Cikkee8zaQnu5u2KQ,5712
 code_puppy/command_line/model_picker_completion.py,sha256=adxp3NZaDV67YqaGv0SG7WVvOTXN0UwkkLqxTsknAvs,4126
 code_puppy/command_line/motd.py,sha256=PEdkp3ZnydVfvd7mNJylm8YyFNUKg9jmY6uwkA1em8c,2152
-code_puppy/command_line/prompt_toolkit_completion.py,sha256=vmsA0F4TfXZ3gjVzCfSNM3TIY-w3z_fSteTCcia2zAU,9379
+code_puppy/command_line/prompt_toolkit_completion.py,sha256=saq1ZRjcxK8s2M-SKy-7I62xGC4Cpxdw02bvxbMwTxA,10010
 code_puppy/command_line/utils.py,sha256=7eyxDHjPjPB9wGDJQQcXV_zOsGdYsFgI0SGCetVmTqE,1251
 code_puppy/command_line/mcp/__init__.py,sha256=0-OQuwjq_pLiTVJ1_NrirVwdRerghyKs_MTZkwPC7YY,315
 code_puppy/command_line/mcp/add_command.py,sha256=RghGmOCjwwYJfrbLiV5UZojMA4-FNPlM04LKci0rpAw,6415
 code_puppy/command_line/mcp/base.py,sha256=nIDv0S5LmGuYtYjVAa9yw3NuSvldxDB17YEEvhbEL1s,889
 code_puppy/command_line/mcp/handler.py,sha256=ZPWNfJEwGurNQh4KUCRRRsHXuB0kHlaGG4oLN1sFhBg,4412
 code_puppy/command_line/mcp/help_command.py,sha256=Z55-ObpUQFMdqMWAkSJkRi_v2uZhDFVxg6DcIdjVY6Q,5250
-code_puppy/command_line/mcp/install_command.py,sha256=vVrl8JO4HK_y_rwnEfXbbpSFK7i2pUa8RyJwCR3bHME,6678
+code_puppy/command_line/mcp/install_command.py,sha256=AgrEp-MynkbLsrOZL54dod5jUsc0Zw_VbHcS3EFz8N4,8826
 code_puppy/command_line/mcp/list_command.py,sha256=9ggXgoqMyzCp9_XGMLbA0WOdIDTynRfJC0VRxnjG7uk,3178
 code_puppy/command_line/mcp/logs_command.py,sha256=x_QsVGPpI5XY7RCZtiNFegc6R3csiwF_IEB_Rh2575w,4453
 code_puppy/command_line/mcp/remove_command.py,sha256=MrWmXQ9jZTq1wrohFDO3ls0a1faTHCqRZocN-ynTzh8,2753
@@ -50,13 +49,13 @@ code_puppy/command_line/mcp/stop_all_command.py,sha256=-HbIE-WXzfWEMFVZhFtxf7pfe
 code_puppy/command_line/mcp/stop_command.py,sha256=zuZO8pIP2Ix11LATf-cwwX3fPFIGAbRDaZ6qnfPTqLY,2594
 code_puppy/command_line/mcp/test_command.py,sha256=Pjod77DYRcA9WDZcxaCe6AUgpEXKJr42QRkxMhSjXhM,3689
 code_puppy/command_line/mcp/utils.py,sha256=FqaeJMPEp39suMq_R3_xb-e8zcNpw1Rcluwvdgum3Uo,3624
-code_puppy/command_line/mcp/wizard_utils.py,sha256=yzsZFpBzPHTGaHKwgUHVWRpPa96aTFdgI7nvHbV5m4I,8156
-code_puppy/mcp/__init__.py,sha256=xnGRFDDtu1BwKz-lRsyevkFdD9qNHAcxrtzW3icQaWM,1446
+code_puppy/command_line/mcp/wizard_utils.py,sha256=HabnT3y6EWLP6w1llRwVO_P25Wi15bYFddTM_2kxAq8,11073
+code_puppy/mcp/__init__.py,sha256=UZ6ZYEIN8At3Jq88ZTkrVQHswNMxHyIBz4hbOn71osk,1444
 code_puppy/mcp/async_lifecycle.py,sha256=pTQrwQCVgjad8EJeRTSHtIeSQRgT_8r5BeLv-1SgKog,7772
 code_puppy/mcp/blocking_startup.py,sha256=bgYln3j12kuHoBsoCUzCjVcTdcoSIAx8kSaOJVwUaRg,13484
 code_puppy/mcp/captured_stdio_server.py,sha256=ubINgtKdd5-oOLR7MRuavUqd40iQFJk2YCNtTFYeI5Q,8948
 code_puppy/mcp/circuit_breaker.py,sha256=a83YwXux9h4R6zBWBUrCIqtp2ffyl7JZEoK2tErG_0I,8601
-code_puppy/mcp/config_wizard.py,sha256=lRVbVbPSnS6I_FSnjZ9D4XBrSVb68lFOeRhRvASqw3Y,16482
+code_puppy/mcp/config_wizard.py,sha256=SIsm8uhDInfRYy_2W8wPIZcmoxk24680ikhMT93VuvQ,16481
 code_puppy/mcp/dashboard.py,sha256=y6t6trrBZU-mr8W1_29VN5DeZI8VYvOsKNz1EXxlvUg,9022
 code_puppy/mcp/error_isolation.py,sha256=mpPBiH17zTXPsOEAn9WmkbwQwnt4gmgiaWv87JBJbUo,12426
 code_puppy/mcp/health_monitor.py,sha256=n5R6EeYOYbUucUFe74qGWCU3g6Mep5UEQbLF0wbT0dU,19688
@@ -73,20 +72,19 @@ code_puppy/messaging/message_queue.py,sha256=CDVpstLee_RbCBeJWv2eON3c3qhlEISf2i9
 code_puppy/messaging/queue_console.py,sha256=L4QFUsR_emAFsIRan3-rnL-F1LODF4DG6je-IlUTxNc,10864
 code_puppy/messaging/renderers.py,sha256=9VOpVmu7emyyg1CXgm17u4IzMNcLHvueBl7G14pLQho,16123
 code_puppy/messaging/spinner/__init__.py,sha256=9mkXPYojafydBOAMh9ZUrB4X6uH5Iqz_-E-Obpd72ko,1365
-code_puppy/messaging/spinner/console_spinner.py,sha256=cuOXQH99dJ1cq0l_rpCLVCGNsH-iVcUWtE6fC3kjZCg,6931
+code_puppy/messaging/spinner/console_spinner.py,sha256=DhEXjD37_FuJwcNNfIP2D0y1ruGlFmuCCuaUW4DFrNg,6932
 code_puppy/messaging/spinner/spinner_base.py,sha256=474qMrTYpNfWcprFzmhaOJEOC-2rRHpTFCLsnl54bXA,1689
 code_puppy/messaging/spinner/textual_spinner.py,sha256=Omx9A-FSPkxYDMYgBXgYMBQnK-DMlyqLOgkFVG8cmo4,3465
 code_puppy/plugins/__init__.py,sha256=fksDqMUiXPJ5WNuMsYsVR8ulueQRCXPlvECEyicHPtQ,1312
 code_puppy/tools/__init__.py,sha256=YiiXRqxU1BEJ5t0Oe163lSqOneI9sKtwDW0swCPgBt4,2119
-code_puppy/tools/agent_tools.py,sha256=tG11PMmjuU-pYG1MFCgqsYiC1Q8C-zPsitAYXxl3mRA,9726
+code_puppy/tools/agent_tools.py,sha256=bHMrFIbYRhuubR41G_XdLsk3cUKWfIPl2O4bVzo2pE0,5591
 code_puppy/tools/command_runner.py,sha256=GVNsgwjTFD9tkNlycgMNmMoVPdmMkZkbAcHH5y6iMww,26070
 code_puppy/tools/common.py,sha256=pL-9xcRs3rxU7Fl9X9EUgbDp2-csh2LLJ5DHH_KAHKY,10596
 code_puppy/tools/file_modifications.py,sha256=oeNEQItqwMhGOeEN2TzGR7TjmgLsfFFdPaVMzWbfXIQ,30398
-code_puppy/tools/file_operations.py,sha256=IxXEzNsotz3DHUV2YfslNf1_gSJrrOMUPEpI2b3TFw0,31093
-code_puppy/tools/token_check.py,sha256=cNrGOOKahXsnWsvh5xnMkL1NS9FjYur9QIRZGQFW-pE,1189
+code_puppy/tools/file_operations.py,sha256=dfG1MDmpvbxsdapSAyo-FJoctC1XM6_AO8HM1PlxUIE,30894
 code_puppy/tools/tools_content.py,sha256=bsBqW-ppd1XNAS_g50B3UHDQBWEALC1UneH6-afz1zo,2365
 code_puppy/tui/__init__.py,sha256=XesAxIn32zLPOmvpR2wIDxDAnnJr81a5pBJB4cZp1Xs,321
-code_puppy/tui/app.py,sha256=10FDM1suc5gDiTD3rj9z3crh2wMYb7W1VUa5QHCs3ss,39373
+code_puppy/tui/app.py,sha256=nPOzwlusjdWzBfu__EbC3Q0etkPrqRq-2g-mk4IcfG4,39378
 code_puppy/tui/messages.py,sha256=zQoToWI0eWdT36NEsY6RdCFzcDfAmfvoPlHv8jiCbgo,720
 code_puppy/tui/components/__init__.py,sha256=uj5pnk3s6SEN3SbFI0ZnzaA2KK1NNg8TfUj6U-Z732U,455
 code_puppy/tui/components/chat_view.py,sha256=wKdrCWgoo8p_f9rL9lB6PDuW4eVdMSZe88TVo--sGF0,18547
@@ -106,29 +104,9 @@ code_puppy/tui/screens/help.py,sha256=eJuPaOOCp7ZSUlecearqsuX6caxWv7NQszUh0tZJjB
 code_puppy/tui/screens/mcp_install_wizard.py,sha256=xqwN5omltMkfxWZwXj3D2PbXbtrxUi1dT0XT77oxOKk,27685
 code_puppy/tui/screens/settings.py,sha256=GMpv-qa08rorAE9mj3AjmqjZFPhmeJ_GWd-DBHG6iAA,10671
 code_puppy/tui/screens/tools.py,sha256=3pr2Xkpa9Js6Yhf1A3_wQVRzFOui-KDB82LwrsdBtyk,1715
-code_puppy/tui/tests/__init__.py,sha256=Fzb4un4eeKfaKsIa5tqI5pTuwfpS8qD7Z6W7KeqWe84,23
-code_puppy/tui/tests/test_agent_command.py,sha256=zTnL-FManVQWq5xSz0FVEEpWDOK8FkPJVQyhWHXn7GA,3012
-code_puppy/tui/tests/test_chat_message.py,sha256=uA3eZBzpTkSLEFVMp6k97JALVlzBP4_1YHHTXQCxV0I,765
-code_puppy/tui/tests/test_chat_view.py,sha256=LKdcQjM0lS-duHuM2fgDoEWY5o2pLSt7vZ2DqDNeitE,2884
-code_puppy/tui/tests/test_command_history.py,sha256=o5i7PvfTDCzRpZQ_napxYxawHiwM1clMSwZw7SVp6C8,3282
-code_puppy/tui/tests/test_copy_button.py,sha256=U1-PwzQQMLQrbWXQrZClwaX-ZUQZtHs5DcJDdfeqVnw,6892
-code_puppy/tui/tests/test_custom_widgets.py,sha256=kRsbnWDe0o8yz-3bWYlvX13fzlQeVp5zt4SgwIjag9k,640
-code_puppy/tui/tests/test_disclaimer.py,sha256=x83pkEldP-oCtoNgq8Tq-9hmPymxJETRwbIMoU1bxtg,921
-code_puppy/tui/tests/test_enums.py,sha256=UJ-F-ssqLiBHdvTMM6I7eOwiY8bthr7_wZPpmGjSvyo,430
-code_puppy/tui/tests/test_file_browser.py,sha256=MxFZXPV6wX5Zu22k78i4cO2_3Y86jDxdpWtIJjD4uYY,2006
-code_puppy/tui/tests/test_help.py,sha256=FeuhSOKkrZ41e6oM6T3d7IBFLxY8H4CXI-sf-EZ3hr4,1116
-code_puppy/tui/tests/test_history_file_reader.py,sha256=xALYU9BKH4ZpsLT2Vt1nn5uj6cF05GRq4swKkoO62SI,3610
-code_puppy/tui/tests/test_input_area.py,sha256=Loic1TDs-0ae7V551UHCxAxkMI4ltIiXGs6dKucEFHs,938
-code_puppy/tui/tests/test_settings.py,sha256=uP7mGRy2vslQqAQCeG66eDiCVivG0wzKN5fwkG2Y3zw,1330
-code_puppy/tui/tests/test_sidebar.py,sha256=GrnslL6V9e-q5Vm3Kq4ajp3YFtGehLY0i6Xa1KjjKdE,909
-code_puppy/tui/tests/test_sidebar_history.py,sha256=Va57Ww2nwMXgHrsnZAObMyLPU5a8-WbRjMqhzi_UgMc,5875
-code_puppy/tui/tests/test_sidebar_history_navigation.py,sha256=JGiyua8A2B8dLfwiEzyJjl7AWuzajVmStFIZe3OlW50,5157
-code_puppy/tui/tests/test_status_bar.py,sha256=nYT_FZGdmqnnbn6o0ZuOkLtNUtJzLSmtX8P72liQ5Vo,1797
-code_puppy/tui/tests/test_timestamped_history.py,sha256=nVXt9hExZZ_8MFP-AZj4L4bB_1Eo_mc-ZhVICzTuw3I,1799
-code_puppy/tui/tests/test_tools.py,sha256=kgzzAkK4r0DPzQwHHD4cePpVNgrHor6cFr05Pg6DBWg,2687
-code_puppy-0.0.154.data/data/code_puppy/models.json,sha256=dAfpMMI2EEeOMv0ynHSmMuJAYDLcZrs5gCLX3voC4-A,3252
-code_puppy-0.0.154.dist-info/METADATA,sha256=r4J6tLEqJwVVG7j5eMrQRI1zMENfwZKdNo2KrOI_pBQ,19516
-code_puppy-0.0.154.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-code_puppy-0.0.154.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
-code_puppy-0.0.154.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
-code_puppy-0.0.154.dist-info/RECORD,,
+code_puppy-0.0.156.data/data/code_puppy/models.json,sha256=jNLDAGhbASNAhul64H4SMS5V4Y6QKW09WcNej4I_4oc,4034
+code_puppy-0.0.156.dist-info/METADATA,sha256=-z0wFe40Q3FDRWzi1Z5vv-2VmPBu2fwa_57kwZHYhJg,19567
+code_puppy-0.0.156.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+code_puppy-0.0.156.dist-info/entry_points.txt,sha256=d8YkBvIUxF-dHNJAj-x4fPEqizbY5d_TwvYpc01U5kw,58
+code_puppy-0.0.156.dist-info/licenses/LICENSE,sha256=31u8x0SPgdOq3izJX41kgFazWsM43zPEF9eskzqbJMY,1075
+code_puppy-0.0.156.dist-info/RECORD,,

code_puppy/token_utils.py

@@ -1,67 +0,0 @@
-import json
-
-import pydantic
-from pydantic_ai.messages import ModelMessage
-
-
-def estimate_token_count(text: str) -> int:
-    """
-    Simple token estimation using len(message) - 4.
-    This replaces tiktoken with a much simpler approach.
-    """
-    return max(1, len(text) - 4)
-
-
-def stringify_message_part(part) -> str:
-    """
-    Convert a message part to a string representation for token estimation or other uses.
-
-    Args:
-        part: A message part that may contain content or be a tool call
-
-    Returns:
-        String representation of the message part
-    """
-    result = ""
-    if hasattr(part, "part_kind"):
-        result += part.part_kind + ": "
-    else:
-        result += str(type(part)) + ": "
-
-    # Handle content
-    if hasattr(part, "content") and part.content:
-        # Handle different content types
-        if isinstance(part.content, str):
-            result = part.content
-        elif isinstance(part.content, pydantic.BaseModel):
-            result = json.dumps(part.content.model_dump())
-        elif isinstance(part.content, dict):
-            result = json.dumps(part.content)
-        else:
-            result = str(part.content)
-
-    # Handle tool calls which may have additional token costs
-    # If part also has content, we'll process tool calls separately
-    if hasattr(part, "tool_name") and part.tool_name:
-        # Estimate tokens for tool name and parameters
-        tool_text = part.tool_name
-        if hasattr(part, "args"):
-            tool_text += f" {str(part.args)}"
-        result += tool_text
-
-    return result
-
-
-def estimate_tokens_for_message(message: ModelMessage) -> int:
-    """
-    Estimate the number of tokens in a message using len(message) - 4.
-    Simple and fast replacement for tiktoken.
-    """
-    total_tokens = 0
-
-    for part in message.parts:
-        part_str = stringify_message_part(part)
-        if part_str:
-            total_tokens += estimate_token_count(part_str)
-
-    return max(1, total_tokens)

code_puppy/tools/token_check.py

@@ -1,32 +0,0 @@
-try:
-    from code_puppy.token_utils import estimate_tokens_for_message
-    from code_puppy.tools.common import get_model_context_length
-except ImportError:
-    # Fallback if these modules aren't available in the internal version
-    def get_model_context_length():
-        return 128000  # Default context length
-
-    def estimate_tokens_for_message(msg):
-        # Simple fallback estimation
-        return len(str(msg)) // 4  # Rough estimate: 4 chars per token
-
-
-def token_guard(num_tokens: int):
-    try:
-        from code_puppy import state_management
-
-        current_history = state_management.get_message_history()
-        message_hist_tokens = sum(
-            estimate_tokens_for_message(msg) for msg in current_history
-        )
-
-        if message_hist_tokens + num_tokens > (get_model_context_length() * 0.9):
-            raise ValueError(
-                "Tokens produced by this tool call would exceed model capacity"
-            )
-    except ImportError:
-        # Fallback: simple check against a reasonable limit
-        if num_tokens > 10000:
-            raise ValueError(
-                f"Token count {num_tokens} exceeds safety limit of 10,000 tokens"
-            )

code_puppy/tui/tests/__init__.py

@@ -1 +0,0 @@
-# Test package for tui

code_puppy/tui/tests/test_agent_command.py

@@ -1,79 +0,0 @@
-"""Tests for the /agent command handling in TUI mode."""
-
-from unittest.mock import MagicMock, patch
-
-from code_puppy.tui.app import CodePuppyTUI
-
-
-class TestTUIAgentCommand:
-    """Test the TUI's handling of /agent commands."""
-
-    @patch("code_puppy.tui.app.get_runtime_agent_manager")
-    @patch("code_puppy.tui.app.handle_command")
-    def test_tui_handles_agent_command(self, mock_handle_command, mock_get_manager):
-        """Test that TUI properly delegates /agent commands to command handler."""
-        # Create a TUI app instance
-        app = CodePuppyTUI()
-
-        # Mock the agent manager and agent
-        mock_agent_instance = MagicMock()
-        mock_manager = MagicMock()
-        mock_manager.get_agent.return_value = mock_agent_instance
-        mock_get_manager.return_value = mock_manager
-
-        # Mock handle_command to simulate successful processing
-        mock_handle_command.return_value = True
-
-        # Simulate processing an /agent command
-        message = "/agent code-puppy"
-        app.agent = mock_agent_instance
-
-        # Call the method that processes messages
-        # We'll need to mock some UI elements to avoid complex setup
-        with (
-            patch.object(app, "add_user_message"),
-            patch.object(app, "_update_submit_cancel_button"),
-            patch.object(app, "start_agent_progress"),
-            patch.object(app, "stop_agent_progress"),
-            patch.object(app, "refresh_history_display"),
-        ):
-            import asyncio
-
-            # Create an event loop for the async test
-            loop = asyncio.get_event_loop()
-            loop.run_until_complete(app.process_message(message))
-
-        # Verify that handle_command was called with the correct argument
-        mock_handle_command.assert_called_once_with(message)
-
-        # Verify that agent manager's get_agent was called to refresh the agent instance
-        mock_manager.get_agent.assert_called()
-
-    @patch("code_puppy.tui.app.get_runtime_agent_manager")
-    def test_tui_refreshes_agent_after_command(self, mock_get_manager):
-        """Test that TUI refreshes its agent instance after processing /agent command."""
-        # Create a TUI app instance
-        app = CodePuppyTUI()
-
-        # Mock the agent manager
-        mock_manager = MagicMock()
-        initial_agent = MagicMock()
-        new_agent = MagicMock()
-
-        # Set initial agent
-        app.agent = initial_agent
-        app.agent_manager = mock_manager
-
-        # Mock manager to return a new agent instance
-        mock_manager.get_agent.return_value = new_agent
-        mock_get_manager.return_value = mock_manager
-
-        # Simulate that an /agent command was processed
-        with patch("code_puppy.tui.app.handle_command"):
-            import asyncio
-
-            loop = asyncio.get_event_loop()
-            loop.run_until_complete(app.process_message("/agent code-puppy"))
-
-        # Verify that the agent was refreshed through the manager
-        mock_manager.get_agent.assert_called()

code_puppy/tui/tests/test_chat_message.py

@@ -1,28 +0,0 @@
-import unittest
-from datetime import datetime
-
-from code_puppy.tui.models.chat_message import ChatMessage
-from code_puppy.tui.models.enums import MessageType
-
-
-class TestChatMessage(unittest.TestCase):
-    def test_chat_message_defaults(self):
-        msg = ChatMessage(
-            id="1", type=MessageType.USER, content="hi", timestamp=datetime.now()
-        )
-        self.assertEqual(msg.metadata, {})
-
-    def test_chat_message_with_metadata(self):
-        meta = {"foo": "bar"}
-        msg = ChatMessage(
-            id="2",
-            type=MessageType.AGENT,
-            content="hello",
-            timestamp=datetime.now(),
-            metadata=meta,
-        )
-        self.assertEqual(msg.metadata, meta)
-
-
-if __name__ == "__main__":
-    unittest.main()

code_puppy/tui/tests/test_chat_view.py

@@ -1,88 +0,0 @@
-import unittest
-from datetime import datetime
-from unittest.mock import patch
-
-from code_puppy.tui.components.chat_view import ChatView
-from code_puppy.tui.models.chat_message import ChatMessage
-from code_puppy.tui.models.enums import MessageType
-
-
-class TestChatView(unittest.TestCase):
-    def setUp(self):
-        self.chat_view = ChatView()
-
-    @patch.object(ChatView, "mount")
-    def test_add_message_user(self, mock_mount):
-        msg = ChatMessage(
-            id="test-user-1",
-            type=MessageType.USER,
-            content="Hello",
-            timestamp=datetime.now(),
-        )
-        self.chat_view.add_message(msg)
-        self.assertIn(msg, self.chat_view.messages)
-        mock_mount.assert_called_once()
-
-    @patch.object(ChatView, "mount")
-    def test_add_message_agent(self, mock_mount):
-        msg = ChatMessage(
-            id="test-agent-1",
-            type=MessageType.AGENT,
-            content="Hi there!",
-            timestamp=datetime.now(),
-        )
-        self.chat_view.add_message(msg)
-        self.assertIn(msg, self.chat_view.messages)
-        mock_mount.assert_called_once()
-
-    @patch.object(ChatView, "mount")
-    def test_add_message_system(self, mock_mount):
-        msg = ChatMessage(
-            id="test-system-1",
-            type=MessageType.SYSTEM,
-            content="System message",
-            timestamp=datetime.now(),
-        )
-        self.chat_view.add_message(msg)
-        self.assertIn(msg, self.chat_view.messages)
-        mock_mount.assert_called_once()
-
-    @patch.object(ChatView, "mount")
-    def test_add_message_error(self, mock_mount):
-        msg = ChatMessage(
-            id="test-error-1",
-            type=MessageType.ERROR,
-            content="Error occurred",
-            timestamp=datetime.now(),
-        )
-        self.chat_view.add_message(msg)
-        self.assertIn(msg, self.chat_view.messages)
-        mock_mount.assert_called_once()
-
-    @patch.object(ChatView, "mount")
-    @patch.object(ChatView, "query")
-    def test_clear_messages(self, mock_query, mock_mount):
-        # Mock the query method to return empty iterables
-        mock_query.return_value = []
-
-        msg = ChatMessage(
-            id="test-clear-1",
-            type=MessageType.USER,
-            content="Hello",
-            timestamp=datetime.now(),
-        )
-        self.chat_view.add_message(msg)
-        self.chat_view.clear_messages()
-        self.assertEqual(len(self.chat_view.messages), 0)
-        # Verify that query was called to find widgets to remove
-        self.assertTrue(mock_query.called)
-
-    def test_render_agent_message_with_syntax(self):
-        prefix = "Agent: "
-        content = "Some text\n```python\nprint('hi')\n```"
-        group = self.chat_view._render_agent_message_with_syntax(prefix, content)
-        self.assertIsNotNone(group)
-
-
-if __name__ == "__main__":
-    unittest.main()

code_puppy/tui/tests/test_command_history.py

@@ -1,89 +0,0 @@
-import re
-import unittest
-from unittest.mock import MagicMock, patch
-
-from code_puppy.config import COMMAND_HISTORY_FILE
-from code_puppy.tui.app import CodePuppyTUI
-from code_puppy.tui.components.custom_widgets import CustomTextArea
-
-
-class TestCommandHistory(unittest.TestCase):
-    def setUp(self):
-        self.app = CodePuppyTUI()
-
-    @patch("builtins.open", new_callable=unittest.mock.mock_open)
-    def test_action_send_message_saves_to_history(self, mock_open):
-        # Setup test mocks
-        self.app.query_one = MagicMock()
-        input_field_mock = MagicMock(spec=CustomTextArea)
-        input_field_mock.text = "test command"
-        self.app.query_one.return_value = input_field_mock
-
-        # Mock other methods to prevent full execution
-        self.app.add_user_message = MagicMock()
-        self.app._update_submit_cancel_button = MagicMock()
-        self.app.run_worker = MagicMock()
-
-        # Execute
-        self.app.action_send_message()
-
-        # Assertions
-        mock_open.assert_called_once_with(COMMAND_HISTORY_FILE, "a")
-        # Check that write was called with timestamped format
-        write_calls = mock_open().write.call_args_list
-        self.assertEqual(len(write_calls), 1)
-        written_content = write_calls[0][0][0]
-        # Should match pattern: \n# YYYY-MM-DDTHH:MM:SS\ntest command\n
-        self.assertTrue(
-            re.match(
-                r"^\n# \d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}\ntest command\n$",
-                written_content,
-            )
-        )
-        self.app.add_user_message.assert_called_once_with("test command")
-
-    @patch("builtins.open", new_callable=unittest.mock.mock_open)
-    def test_action_send_message_empty_command(self, mock_open):
-        # Setup test mocks
-        self.app.query_one = MagicMock()
-        input_field_mock = MagicMock(spec=CustomTextArea)
-        input_field_mock.text = "   "  # Empty or whitespace-only command
-        self.app.query_one.return_value = input_field_mock
-
-        # Mock other methods
-        self.app.add_user_message = MagicMock()
-
-        # Execute
-        self.app.action_send_message()
-
-        # Assertions - nothing should happen with empty commands
-        mock_open.assert_not_called()
-        self.app.add_user_message.assert_not_called()
-
-    @patch("builtins.open")
-    def test_action_send_message_handles_error(self, mock_open):
-        # Setup test mocks
-        self.app.query_one = MagicMock()
-        input_field_mock = MagicMock(spec=CustomTextArea)
-        input_field_mock.text = "test command"
-        self.app.query_one.return_value = input_field_mock
-
-        # Mock other methods to prevent full execution
-        self.app.add_user_message = MagicMock()
-        self.app._update_submit_cancel_button = MagicMock()
-        self.app.run_worker = MagicMock()
-        self.app.add_error_message = MagicMock()
-
-        # Make open throw an exception
-        mock_open.side_effect = Exception("File error")
-
-        # Execute
-        self.app.action_send_message()
-
-        # Assertions - error is printed to stdout, not added to UI
-        # Message should still be processed despite error saving to history
-        self.app.add_user_message.assert_called_once_with("test command")
-
-
-if __name__ == "__main__":
-    unittest.main()