shotgun-sh 0.1.16.dev2__py3-none-any.whl → 0.2.0__py3-none-any.whl

shotgun/utils/env_utils.py

@@ -1,5 +1,17 @@
 """Utilities for working with environment variables."""
 
+import os
+
+
+def is_shotgun_account_enabled() -> bool:
+    """Check if Shotgun Account feature is enabled via environment variable.
+
+    Returns:
+        True if SHOTGUN_ACCOUNT_ENABLED is set to a truthy value,
+        False otherwise
+    """
+    return is_truthy(os.environ.get("SHOTGUN_ACCOUNT_ENABLED"))
+
 
 def is_truthy(value: str | None) -> bool:
     """Check if a string value represents true.

{shotgun_sh-0.1.16.dev2.dist-info → shotgun_sh-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.1.16.dev2
+Version: 0.2.0
 Summary: AI-powered research, planning, and task management CLI tool
 Project-URL: Homepage, https://shotgun.sh/
 Project-URL: Repository, https://github.com/shotgun-sh/shotgun
@@ -23,7 +23,6 @@ Classifier: Topic :: Utilities
 Requires-Python: >=3.11
 Requires-Dist: anthropic>=0.39.0
 Requires-Dist: genai-prices>=0.0.27
-Requires-Dist: google-generativeai>=0.8.5
 Requires-Dist: httpx>=0.27.0
 Requires-Dist: jinja2>=3.1.0
 Requires-Dist: kuzu>=0.7.0
@@ -33,6 +32,7 @@ Requires-Dist: packaging>=23.0
 Requires-Dist: posthog>=3.0.0
 Requires-Dist: pydantic-ai>=0.0.14
 Requires-Dist: rich>=13.0.0
+Requires-Dist: sentencepiece>=0.2.0
 Requires-Dist: sentry-sdk[pure-eval]>=2.0.0
 Requires-Dist: textual-dev>=1.7.0
 Requires-Dist: textual>=6.1.0

{shotgun_sh-0.1.16.dev2.dist-info → shotgun_sh-0.2.0.dist-info}/RECORD

@@ -1,38 +1,45 @@
 shotgun/__init__.py,sha256=P40K0fnIsb7SKcQrFnXZ4aREjpWchVDhvM1HxI4cyIQ,104
-shotgun/build_constants.py,sha256=RXNxMz46HaB5jucgMVpw8a2yCJqjbhTOh0PddyEVMN8,713
+shotgun/build_constants.py,sha256=hDFr6eO0lwN0iCqHQ1A5s0D68txR8sYrTJLGa7tSi0o,654
 shotgun/logging_config.py,sha256=UKenihvgH8OA3W0b8ZFcItYaFJVe9MlsMYlcevyW1HY,7440
-shotgun/main.py,sha256=670RwzIwEIz9QRil37IbVoxWuX66YATqXFLSYSqKw-w,4955
-shotgun/posthog_telemetry.py,sha256=ZD_BjRej1v4Mxh7VN3AlXGKV4jIU9SC0uBrH94VQa6c,5885
+shotgun/main.py,sha256=RA3q1xPfqxCu43UmgI2ryZpA-IxPhJb_MJrbLqp9c_g,5140
+shotgun/posthog_telemetry.py,sha256=RGA_rgKxAnIofjkm-cLKzyuckAKz-861gpp97tk8IME,5949
 shotgun/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 shotgun/sentry_telemetry.py,sha256=L7jFMNAnDIENWVeQYSLpyul2nmIm2w3wnOp2kDP_cic,2902
 shotgun/telemetry.py,sha256=Ves6Ih3hshpKVNVAUUmwRdtW8NkTjFPg8hEqvFKZ0t0,3208
 shotgun/agents/__init__.py,sha256=8Jzv1YsDuLyNPFJyckSr_qI4ehTVeDyIMDW4omsfPGc,25
 shotgun/agents/agent_manager.py,sha256=xq8L0oAFgtFCpKVsyUoMtYJqUyz5XxjWLKNnxoe1zo4,26577
-shotgun/agents/common.py,sha256=vt7ECq1rT6GR5Rt63t0whH0R0cydrk7Mty2KyPL8mEg,19045
+shotgun/agents/common.py,sha256=Hr9HigsDopkI0Sr3FThGDv1f67NLemOjcYA6LV9v970,18963
 shotgun/agents/conversation_history.py,sha256=5J8_1yxdZiiWTq22aDio88DkBDZ4_Lh_p5Iy5_ENszc,3898
 shotgun/agents/conversation_manager.py,sha256=fxAvXbEl3Cl2ugJ4N9aWXaqZtkrnfj3QzwjWC4LFXwI,3514
 shotgun/agents/export.py,sha256=Zke952DbJ_lOBUmN-TPHw7qmjbfqsFu1uycBRQI_pkg,2969
+shotgun/agents/llm.py,sha256=hs8j1wwTczGtehzahL1Z_5D4qus5QUx4-h9-m5ZPzm4,2209
 shotgun/agents/messages.py,sha256=wNn0qC5AqASM8LMaSGFOerZEJPn5FsIOmaJs1bdosuU,1036
-shotgun/agents/models.py,sha256=ULN7wdOJebYlNxtbKFTm9nJ9uv8g1C8nAG46XOcscTY,8104
+shotgun/agents/models.py,sha256=IvwwjbJYi5wi9S-budg8g1ezi1VaO57Q-XtegkbTrXg,8096
 shotgun/agents/plan.py,sha256=s-WfILBOW4l8kY59RUOVtX5MJSuSzFm1nGp6b17If78,3030
 shotgun/agents/research.py,sha256=lYG7Rytcitop8mXs3isMI3XvYzzI3JH9u0VZz6K9zfo,3274
 shotgun/agents/specify.py,sha256=7MoMxfIn34G27mw6wrp_F0i2O5rid476L3kHFONDCd0,3137
 shotgun/agents/tasks.py,sha256=nk8zIl24o01hfzOGyWSbeVWeke6OGseO4Ppciurh13U,2999
 shotgun/agents/usage_manager.py,sha256=5d9JC4_cthXwhTSytMfMExMDAUYp8_nkPepTJZXk13w,5017
 shotgun/agents/config/__init__.py,sha256=Fl8K_81zBpm-OfOW27M_WWLSFdaHHek6lWz95iDREjQ,318
-shotgun/agents/config/constants.py,sha256=MogArrb2r5rFI6BBzc6NhPz1payGeM6K-t5oIFbJgxg,494
-shotgun/agents/config/manager.py,sha256=kwMbPjz0kEH_WCQAamESGjHdE8d_P-ztel4NL4FWNUw,10662
-shotgun/agents/config/models.py,sha256=vpVXrtiHsDt2D_h7BLyMiiQeT97vAz2L6lYKx2SEMjo,5909
-shotgun/agents/config/provider.py,sha256=pVWf_WM3MNWH0v2fU-peBCqx49X-nW81piQ_M-AKWRE,7249
+shotgun/agents/config/constants.py,sha256=I3f0ueoQaTg5HddXGCYimCYpj-U57z3IBQYIVJxVIhg,872
+shotgun/agents/config/manager.py,sha256=2QBpaWBV3WAqZR-wFHE9Ufwen05KHQbEkPzjSb3r_V0,15140
+shotgun/agents/config/models.py,sha256=ZojhfheNO337e1icy_cE2PpBXIl5oHkdajr4azzFF-U,5106
+shotgun/agents/config/provider.py,sha256=TwwZC_BtYSOpN2jdX6WZdor29EnAqfMoQK5GmNEYaPI,11012
 shotgun/agents/history/__init__.py,sha256=XFQj2a6fxDqVg0Q3juvN9RjV_RJbgvFZtQOCOjVJyp4,147
-shotgun/agents/history/compaction.py,sha256=Je8-7T2i78gv_QWzgZNe9mvscaHcaTMttQX9xS8M38Q,3509
+shotgun/agents/history/compaction.py,sha256=9RMpG0aY_7L4TecbgwHSOkGtbd9W5XZTg-MbzZmNl00,3515
 shotgun/agents/history/constants.py,sha256=yWY8rrTZarLA3flCCMB_hS2NMvUDRDTwP4D4j7MIh1w,446
 shotgun/agents/history/context_extraction.py,sha256=yVka1U6TqNVsORR4JlxpWi9yBt3Quip8g_u3x2Vi9Gs,3564
 shotgun/agents/history/history_building.py,sha256=6LFDZ60MTPDoGAcmu_mjlnjVYu8YYWdIi-cGbF3jm7A,3532
-shotgun/agents/history/history_processors.py,sha256=NbStr6CvszwK3DKO5LCiaVEJI-RAcVnH5dCpHTAHjQo,17731
+shotgun/agents/history/history_processors.py,sha256=D3z-hzrXHxE7OAZaVX4_YAKN_nyxSF5iYMIYO24V_CI,17943
 shotgun/agents/history/message_utils.py,sha256=aPusAl2RYKbjc7lBxPaNprRHmZEG6fe97q7DQUlhlzU,2918
-shotgun/agents/history/token_counting.py,sha256=RasWy84eNjbmqyQDTGAzj1Q1I9ml_G_9R-maWN7gr8s,13839
-shotgun/agents/history/token_estimation.py,sha256=iNqhDSqFzG0YYxGijMRzj54GALFglOp0qVMB6G59RhU,4690
+shotgun/agents/history/token_estimation.py,sha256=iRyKq-YDivEpJrULIbQgNpjhOuSC4nHVJYfsWEFV8sQ,4770
+shotgun/agents/history/token_counting/__init__.py,sha256=YZt5Lus--fkF6l1hdkIlp1e_oAIpACNwHOI0FRP4q8s,924
+shotgun/agents/history/token_counting/anthropic.py,sha256=b2LvwKM4dSILGhv_-W4mLMKMUCPLhe1ov9UGW_-iBsw,3011
+shotgun/agents/history/token_counting/base.py,sha256=TN4mzwSyWNQyTuOuCFaU-8AgLdAyquoX3af4qrmkxCs,1904
+shotgun/agents/history/token_counting/openai.py,sha256=XJ2z2HaUG6f3Cw9tCK_yaOsaMJGHpSFF1I30-d3soSI,2350
+shotgun/agents/history/token_counting/sentencepiece_counter.py,sha256=qj1bT7J5nCd5y6Mr42O9K1KTaele0rjdd09FeyyEA70,3987
+shotgun/agents/history/token_counting/tokenizer_cache.py,sha256=Y0V6KMtEwn42M5-zJGAc7YudM8X6m5-j2ekA6YGL5Xk,2868
+shotgun/agents/history/token_counting/utils.py,sha256=d124IDjtd0IYBYrr3gDJGWxSbdP10Vrc7ZistbUosMg,5002
 shotgun/agents/tools/__init__.py,sha256=QaN80IqWvB5qEcjHqri1-PYvYlO74vdhcwLugoEdblo,772
 shotgun/agents/tools/file_management.py,sha256=HYNe_QA4T3_bPzSWBYcFZcnWdj8eb4aQ3GB735-G8Nw,7138
 shotgun/agents/tools/user_interaction.py,sha256=b3ncEpvoD06Cz4hwsS-ppVbQajQj640iWnVfA5WBjAA,1236
@@ -43,16 +50,16 @@ shotgun/agents/tools/codebase/file_read.py,sha256=EGK5yNqiS4cbIEQfDtdKVoJSJYk20N
 shotgun/agents/tools/codebase/models.py,sha256=8eR3_8DQiBNgB2twu0aC_evIJbugN9KW3gtxMZdGYCE,10087
 shotgun/agents/tools/codebase/query_graph.py,sha256=vOeyN4-OZj-vpTSk3Z9W5TjraZAepJ-Qjk_zzvum3fU,2115
 shotgun/agents/tools/codebase/retrieve_code.py,sha256=2VjiqVKJMd9rPV-mGrL4C-N8fqGjYLW6ZInFGbcTxOM,2878
-shotgun/agents/tools/web_search/__init__.py,sha256=Sj1tVokrCsJiLRWWTq0zrAolMHEGntRIYnqiyFi8L2E,1840
-shotgun/agents/tools/web_search/anthropic.py,sha256=NDhj8MrdxLsmGwHp7uM0IQeJVW2poY58GCUTJEM9dew,4827
-shotgun/agents/tools/web_search/gemini.py,sha256=hXjWUF-aTX3B9ViaKe5aF2aHXlaoBA5am40cgilinGE,2981
-shotgun/agents/tools/web_search/openai.py,sha256=V8GeqwUAi5wrbRuU41Y38schpXRdyeIfw85-CT5rAhY,3415
+shotgun/agents/tools/web_search/__init__.py,sha256=_9rgs_gv41-wfPvwfWM_Qfq-zvboyQ_srfyneGsxgM4,3182
+shotgun/agents/tools/web_search/anthropic.py,sha256=GelAhAmb-b4o87-3sgxNFfw-G2LXDEjfdZ7XfF0bQD0,4983
+shotgun/agents/tools/web_search/gemini.py,sha256=-fI_deaBT4-_61A7KlKtz8tmKXW50fVx_97WAJTUg4w,3468
+shotgun/agents/tools/web_search/openai.py,sha256=pnIcTV3vwXJQuxPs4I7gQNX18XzM7D7FqeNxnn1E7yw,3437
 shotgun/agents/tools/web_search/utils.py,sha256=GLJ5QV9bT2ubFMuFN7caMN7tK9OTJ0R3GD57B-tCMF0,532
 shotgun/cli/__init__.py,sha256=_F1uW2g87y4bGFxz8Gp8u7mq2voHp8vQIUtCmm8Tojo,40
-shotgun/cli/config.py,sha256=LbjxDNPdetYJiwlcyOYLnqwzALfgU-m54cfstUshbrs,8715
+shotgun/cli/config.py,sha256=Lrcqxm7W7I6g6iP_K5-yK7QFOgcYt5KIc8A6Wit1Ksg,7835
 shotgun/cli/export.py,sha256=3hIwK2_OM1MFYSTfzBxsGuuBGm5fo0XdxASfQ5Uqb3Y,2471
 shotgun/cli/feedback.py,sha256=Me1dQQgkYwP4AIFwYgfHcPXxFdJ6CzFbCBttKcFd2Q0,1238
-shotgun/cli/models.py,sha256=LoajeEK7MEDUSnZXb1Li-dbhXqne812YZglx-LcVpiQ,181
+shotgun/cli/models.py,sha256=kwZEldQWUheNsqF_ezgDzRBc6h0Y0JxFw1VMQjZlvPE,182
 shotgun/cli/plan.py,sha256=T-eu-I9z-dSoKqJ-KI8X5i5Mm0VL1BfornxRiUjTgnk,2324
 shotgun/cli/research.py,sha256=qvBBtX3Wyn6pDZlJpcEvbeK-0iTOXegi71tm8HKVYaE,2490
 shotgun/cli/specify.py,sha256=ErRQ72Zc75fmxopZbKy0vvnLPuYBLsGynpjj1X6-BwI,2166
@@ -63,7 +70,7 @@ shotgun/cli/codebase/__init__.py,sha256=rKdvx33p0i_BYbNkz5_4DCFgEMwzOOqLi9f5p7XT
 shotgun/cli/codebase/commands.py,sha256=1N2yOGmok0ZarqXPIpWGcsQrwm_ZJcyWiMxy6tm0j70,8711
 shotgun/cli/codebase/models.py,sha256=B9vs-d-Bq0aS6FZKebhHT-9tw90Y5f6k_t71VlZpL8k,374
 shotgun/codebase/__init__.py,sha256=QBgFE2Abd5Vl7_NdYOglF9S6d-vIjkb3C0cpIYoHZEU,309
-shotgun/codebase/models.py,sha256=1AAipm6KrGOHmYBBavugnyeOsVkzX-YXAD6dDsSVRWg,5299
+shotgun/codebase/models.py,sha256=5e_7zaPL032n_ghcvs01Uug3BH4jyKiQ3S3U5w21BSM,5296
 shotgun/codebase/service.py,sha256=nyggapfHKdwkKXyuT9oA0tJ9qf4RNVsOxfY8lC5pHro,8006
 shotgun/codebase/core/__init__.py,sha256=GWWhJEqChiDXAF4omYCgzgoZmJjwsAf6P1aZ5Bl8OE0,1170
 shotgun/codebase/core/change_detector.py,sha256=kWCYLWzRzb3IGGOj71KBn7UOCOKMpINJbOBDf98aMxE,12409
@@ -74,6 +81,9 @@ shotgun/codebase/core/language_config.py,sha256=vsqHyuFnumRPRBV1lMOxWKNOIiClO6Fy
 shotgun/codebase/core/manager.py,sha256=USGLBdDUoFtq6fMFWRtUu2HBC_FI8d6lWcAV4l6fcvk,66000
 shotgun/codebase/core/nl_query.py,sha256=kPoSJXBlm5rLhzOofZhqPVMJ_Lj3rV2H6sld6BwtMdg,16115
 shotgun/codebase/core/parser_loader.py,sha256=LZRrDS8Sp518jIu3tQW-BxdwJ86lnsTteI478ER9Td8,4278
+shotgun/llm_proxy/__init__.py,sha256=BLD9NnVzdD0H7gFb65Ajud-Q7SiCymegLRaGx8UkC-Y,435
+shotgun/llm_proxy/clients.py,sha256=wP4UlgtCdrNwWsZLZ9inE3fEIDa-i1j7gsr9oXQf1o4,1037
+shotgun/llm_proxy/constants.py,sha256=E8sqL-8GZzl989T3OS7E1hImSZPj2vqmp3lbM6zGiQU,309
 shotgun/prompts/__init__.py,sha256=RswUm0HMdfm2m2YKUwUsEdRIwoczdbI7zlucoEvHYRo,132
 shotgun/prompts/loader.py,sha256=jy24-E02pCSmz2651aCT2NgHfRrHAGMYvKrD6gs0Er8,4424
 shotgun/prompts/agents/__init__.py,sha256=YRIJMbzpArojNX1BP5gfxxois334z_GQga8T-xyWMbY,39
@@ -104,7 +114,7 @@ shotgun/sdk/exceptions.py,sha256=qBcQv0v7ZTwP7CMcxZST4GqCsfOWtOUjSzGBo0-heqo,412
 shotgun/sdk/models.py,sha256=X9nOTUHH0cdkQW1NfnMEDu-QgK9oUsEISh1Jtwr5Am4,5496
 shotgun/sdk/services.py,sha256=J4PJFSxCQ6--u7rb3Ta-9eYtlYcxcbnzrMP6ThyCnw4,705
 shotgun/tui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-shotgun/tui/app.py,sha256=Ca7pAF2GOR6RvJdK1_P_aqF7OosTm5WRUhxF1M-Ihcs,4994
+shotgun/tui/app.py,sha256=sBPviBs-3niD8rDoD0wC27lfBsKRbFYMUPttOwmDzOM,5139
 shotgun/tui/filtered_codebase_service.py,sha256=lJ8gTMhIveTatmvmGLP299msWWTkVYKwvY_2FhuL2s4,1687
 shotgun/tui/styles.tcss,sha256=ETyyw1bpMBOqTi5RLcAJUScdPWTvAWEqE9YcT0kVs_E,121
 shotgun/tui/commands/__init__.py,sha256=8D5lvtpqMW5-fF7Bg3oJtUzU75cKOv6aUaHYYszydU8,2518
@@ -112,25 +122,26 @@ shotgun/tui/components/prompt_input.py,sha256=Ss-htqraHZAPaehGE4x86ij0veMjc4Ugad
 shotgun/tui/components/spinner.py,sha256=ovTDeaJ6FD6chZx_Aepia6R3UkPOVJ77EKHfRmn39MY,2427
 shotgun/tui/components/splash.py,sha256=vppy9vEIEvywuUKRXn2y11HwXSRkQZHLYoVjhDVdJeU,1267
 shotgun/tui/components/vertical_tail.py,sha256=kROwTaRjUwVB7H35dtmNcUVPQqNYvvfq7K2tXBKEb6c,638
-shotgun/tui/screens/chat.py,sha256=H3uAE4sQ7iJO33tJhdoW_OkzaNxXgW6BUP-SxoNiohw,30465
+shotgun/tui/screens/chat.py,sha256=CqAv_x6R4zl-MGbtg8KgZWt8OhpBJYpx5gGBQ3oxqgw,30313
 shotgun/tui/screens/chat.tcss,sha256=2Yq3E23jxsySYsgZf4G1AYrYVcpX0UDW6kNNI0tDmtM,437
 shotgun/tui/screens/directory_setup.py,sha256=lIZ1J4A6g5Q2ZBX8epW7BhR96Dmdcg22CyiM5S-I5WU,3237
 shotgun/tui/screens/feedback.py,sha256=cYtmuM3qqKwevstu8gJ9mmk7lkIKZvfAyDEBUOLh-yI,5660
-shotgun/tui/screens/provider_config.py,sha256=KIsI9bCOzk6wf_WIDHoaWOcrAaXgT4gF6hnX-1ArEn8,7487
+shotgun/tui/screens/model_picker.py,sha256=G-EvalpxgHKk0W3FgHMcxIr817VwZyEgh_ZadSQiRwo,11831
+shotgun/tui/screens/provider_config.py,sha256=RWH7ksf9dp7eD7mz0g_o2Q-O6HTAyzIv7ZkOYrIOAI4,8686
 shotgun/tui/screens/splash.py,sha256=E2MsJihi3c9NY1L28o_MstDxGwrCnnV7zdq00MrGAsw,706
 shotgun/tui/screens/chat_screen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-shotgun/tui/screens/chat_screen/command_providers.py,sha256=A3vOs6J9QXg-JVi47knQdYM2_vA_IygsDJst0xpDibg,8923
+shotgun/tui/screens/chat_screen/command_providers.py,sha256=7Xnxd4k30bpLOMZSX32bcugU4IgpqU4Y8f6eHWKXd4o,12694
 shotgun/tui/screens/chat_screen/hint_message.py,sha256=WOpbk8q7qt7eOHTyyHvh_IQIaublVDeJGaLpsxEk9FA,933
-shotgun/tui/screens/chat_screen/history.py,sha256=NVLA3_tERTyB4vkH71w8ef_M5CszfkwbQOuMb100Fzc,12272
+shotgun/tui/screens/chat_screen/history.py,sha256=Go859iEjw0s5aELKpF42MjLXy7UFQ52XnJMTIkV3aLo,12406
 shotgun/tui/utils/__init__.py,sha256=cFjDfoXTRBq29wgP7TGRWUu1eFfiIG-LLOzjIGfadgI,150
 shotgun/tui/utils/mode_progress.py,sha256=lseRRo7kMWLkBzI3cU5vqJmS2ZcCjyRYf9Zwtvc-v58,10931
 shotgun/utils/__init__.py,sha256=WinIEp9oL2iMrWaDkXz2QX4nYVPAm8C9aBSKTeEwLtE,198
-shotgun/utils/env_utils.py,sha256=8QK5aw_f_V2AVTleQQlcL0RnD4sPJWXlDG46fsHu0d8,1057
+shotgun/utils/env_utils.py,sha256=5spVCdeqVKtlWoKocPhz_5j_iRN30neqcGUzUuiWmfc,1365
 shotgun/utils/file_system_utils.py,sha256=l-0p1bEHF34OU19MahnRFdClHufThfGAjQ431teAIp0,1004
 shotgun/utils/source_detection.py,sha256=Co6Q03R3fT771TF3RzB-70stfjNP2S4F_ArZKibwzm8,454
 shotgun/utils/update_checker.py,sha256=IgzPHRhS1ETH7PnJR_dIx6lxgr1qHpCkMTgzUxvGjhI,7586
-shotgun_sh-0.1.16.dev2.dist-info/METADATA,sha256=593zGac-zT_lBaqzrAXKl_N0WxPigfiXnLSWmY92Z14,11233
-shotgun_sh-0.1.16.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-shotgun_sh-0.1.16.dev2.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
-shotgun_sh-0.1.16.dev2.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
-shotgun_sh-0.1.16.dev2.dist-info/RECORD,,
+shotgun_sh-0.2.0.dist-info/METADATA,sha256=GMbem4VmalbvjFnNaBI1-XbnlghM_YVUMlrtNkKKqXA,11221
+shotgun_sh-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+shotgun_sh-0.2.0.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
+shotgun_sh-0.2.0.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
+shotgun_sh-0.2.0.dist-info/RECORD,,

shotgun/agents/history/token_counting.py

@@ -1,429 +0,0 @@
-"""Real token counting for all supported providers.
-
-This module provides accurate token counting using each provider's official
-APIs and libraries, eliminating the need for rough character-based estimation.
-"""
-
-from abc import ABC, abstractmethod
-from typing import TYPE_CHECKING
-
-from pydantic_ai.messages import ModelMessage
-
-from shotgun.agents.config.models import ModelConfig, ProviderType
-from shotgun.logging_config import get_logger
-
-if TYPE_CHECKING:
-    pass
-
-logger = get_logger(__name__)
-
-# Global cache for token counter instances (singleton pattern)
-_token_counter_cache: dict[tuple[str, str, str], "TokenCounter"] = {}
-
-
-class TokenCounter(ABC):
-    """Abstract base class for provider-specific token counting."""
-
-    @abstractmethod
-    def count_tokens(self, text: str) -> int:
-        """Count tokens in text using provider-specific method.
-
-        Args:
-            text: Text to count tokens for
-
-        Returns:
-            Exact token count as determined by the provider
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-
-    @abstractmethod
-    def count_message_tokens(self, messages: list[ModelMessage]) -> int:
-        """Count tokens in PydanticAI message structures.
-
-        Args:
-            messages: List of messages to count tokens for
-
-        Returns:
-            Total token count across all messages
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-
-
-class OpenAITokenCounter(TokenCounter):
-    """Token counter for OpenAI models using tiktoken."""
-
-    # Official encoding mappings for OpenAI models
-    ENCODING_MAP = {
-        "gpt-5": "o200k_base",
-        "gpt-4o": "o200k_base",
-        "gpt-4": "cl100k_base",
-        "gpt-3.5-turbo": "cl100k_base",
-    }
-
-    def __init__(self, model_name: str):
-        """Initialize OpenAI token counter.
-
-        Args:
-            model_name: OpenAI model name to get correct encoding for
-
-        Raises:
-            RuntimeError: If encoding initialization fails
-        """
-        self.model_name = model_name
-
-        import tiktoken
-
-        try:
-            # Get the appropriate encoding for this model
-            encoding_name = self.ENCODING_MAP.get(model_name, "o200k_base")
-            self.encoding = tiktoken.get_encoding(encoding_name)
-            logger.debug(
-                f"Initialized OpenAI token counter with {encoding_name} encoding"
-            )
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to initialize tiktoken encoding for {model_name}"
-            ) from e
-
-    def count_tokens(self, text: str) -> int:
-        """Count tokens using tiktoken.
-
-        Args:
-            text: Text to count tokens for
-
-        Returns:
-            Exact token count using tiktoken
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-        try:
-            return len(self.encoding.encode(text))
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to count tokens for OpenAI model {self.model_name}"
-            ) from e
-
-    def count_message_tokens(self, messages: list[ModelMessage]) -> int:
-        """Count tokens across all messages using tiktoken.
-
-        Args:
-            messages: List of PydanticAI messages
-
-        Returns:
-            Total token count for all messages
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-        total_text = self._extract_text_from_messages(messages)
-        return self.count_tokens(total_text)
-
-    def _extract_text_from_messages(self, messages: list[ModelMessage]) -> str:
-        """Extract all text content from messages for token counting."""
-        text_parts = []
-
-        for message in messages:
-            if hasattr(message, "parts"):
-                for part in message.parts:
-                    if hasattr(part, "content") and isinstance(part.content, str):
-                        text_parts.append(part.content)
-                    else:
-                        # Handle non-text parts (tool calls, etc.)
-                        text_parts.append(str(part))
-            else:
-                # Handle messages without parts
-                text_parts.append(str(message))
-
-        return "\n".join(text_parts)
-
-
-class AnthropicTokenCounter(TokenCounter):
-    """Token counter for Anthropic models using official client."""
-
-    def __init__(self, model_name: str, api_key: str):
-        """Initialize Anthropic token counter.
-
-        Args:
-            model_name: Anthropic model name for token counting
-            api_key: Anthropic API key
-
-        Raises:
-            RuntimeError: If client initialization fails
-        """
-        self.model_name = model_name
-        import anthropic
-
-        try:
-            self.client = anthropic.Anthropic(api_key=api_key)
-            logger.debug(f"Initialized Anthropic token counter for {model_name}")
-        except Exception as e:
-            raise RuntimeError("Failed to initialize Anthropic client") from e
-
-    def count_tokens(self, text: str) -> int:
-        """Count tokens using Anthropic's official API.
-
-        Args:
-            text: Text to count tokens for
-
-        Returns:
-            Exact token count from Anthropic API
-
-        Raises:
-            RuntimeError: If API call fails
-        """
-        try:
-            # Anthropic API expects messages format and model parameter
-            result = self.client.messages.count_tokens(
-                messages=[{"role": "user", "content": text}], model=self.model_name
-            )
-            return result.input_tokens
-        except Exception as e:
-            raise RuntimeError(
-                f"Anthropic token counting API failed for {self.model_name}"
-            ) from e
-
-    def count_message_tokens(self, messages: list[ModelMessage]) -> int:
-        """Count tokens across all messages using Anthropic API.
-
-        Args:
-            messages: List of PydanticAI messages
-
-        Returns:
-            Total token count for all messages
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-        total_text = self._extract_text_from_messages(messages)
-        return self.count_tokens(total_text)
-
-    def _extract_text_from_messages(self, messages: list[ModelMessage]) -> str:
-        """Extract all text content from messages for token counting."""
-        text_parts = []
-
-        for message in messages:
-            if hasattr(message, "parts"):
-                for part in message.parts:
-                    if hasattr(part, "content") and isinstance(part.content, str):
-                        text_parts.append(part.content)
-                    else:
-                        # Handle non-text parts (tool calls, etc.)
-                        text_parts.append(str(part))
-            else:
-                # Handle messages without parts
-                text_parts.append(str(message))
-
-        return "\n".join(text_parts)
-
-
-class GoogleTokenCounter(TokenCounter):
-    """Token counter for Google models using genai API."""
-
-    def __init__(self, model_name: str, api_key: str):
-        """Initialize Google token counter.
-
-        Args:
-            model_name: Google model name
-            api_key: Google API key
-
-        Raises:
-            RuntimeError: If configuration fails
-        """
-        self.model_name = model_name
-
-        import google.generativeai as genai
-
-        try:
-            genai.configure(api_key=api_key)  # type: ignore[attr-defined]
-            self.model = genai.GenerativeModel(model_name)  # type: ignore[attr-defined]
-            logger.debug(f"Initialized Google token counter for {model_name}")
-        except Exception as e:
-            raise RuntimeError(
-                f"Failed to configure Google genai client for {model_name}"
-            ) from e
-
-    def count_tokens(self, text: str) -> int:
-        """Count tokens using Google's genai API.
-
-        Args:
-            text: Text to count tokens for
-
-        Returns:
-            Exact token count from Google API
-
-        Raises:
-            RuntimeError: If API call fails
-        """
-        try:
-            result = self.model.count_tokens(text)
-            return result.total_tokens
-        except Exception as e:
-            raise RuntimeError(
-                f"Google token counting API failed for {self.model_name}"
-            ) from e
-
-    def count_message_tokens(self, messages: list[ModelMessage]) -> int:
-        """Count tokens across all messages using Google API.
-
-        Args:
-            messages: List of PydanticAI messages
-
-        Returns:
-            Total token count for all messages
-
-        Raises:
-            RuntimeError: If token counting fails
-        """
-        total_text = self._extract_text_from_messages(messages)
-        return self.count_tokens(total_text)
-
-    def _extract_text_from_messages(self, messages: list[ModelMessage]) -> str:
-        """Extract all text content from messages for token counting."""
-        text_parts = []
-
-        for message in messages:
-            if hasattr(message, "parts"):
-                for part in message.parts:
-                    if hasattr(part, "content") and isinstance(part.content, str):
-                        text_parts.append(part.content)
-                    else:
-                        # Handle non-text parts (tool calls, etc.)
-                        text_parts.append(str(part))
-            else:
-                # Handle messages without parts
-                text_parts.append(str(message))
-
-        return "\n".join(text_parts)
-
-
-def get_token_counter(model_config: ModelConfig) -> TokenCounter:
-    """Get appropriate token counter for the model provider (cached singleton).
-
-    This function ensures that every provider has a proper token counting
-    implementation without any fallbacks to estimation. Token counters are
-    cached to avoid repeated initialization overhead.
-
-    Args:
-        model_config: Model configuration with provider and credentials
-
-    Returns:
-        Cached provider-specific token counter
-
-    Raises:
-        ValueError: If provider is not supported for token counting
-        RuntimeError: If token counter initialization fails
-    """
-    # Create cache key from provider, model name, and API key
-    cache_key = (
-        model_config.provider.value,
-        model_config.name,
-        model_config.api_key[:10]
-        if model_config.api_key
-        else "no-key",  # Partial key for cache
-    )
-
-    # Return cached instance if available
-    if cache_key in _token_counter_cache:
-        logger.debug(
-            f"Reusing cached token counter for {model_config.provider.value}:{model_config.name}"
-        )
-        return _token_counter_cache[cache_key]
-
-    # Create new instance and cache it
-    logger.debug(
-        f"Creating new token counter for {model_config.provider.value}:{model_config.name}"
-    )
-
-    counter: TokenCounter
-    if model_config.provider == ProviderType.OPENAI:
-        counter = OpenAITokenCounter(model_config.name)
-    elif model_config.provider == ProviderType.ANTHROPIC:
-        counter = AnthropicTokenCounter(model_config.name, model_config.api_key)
-    elif model_config.provider == ProviderType.GOOGLE:
-        counter = GoogleTokenCounter(model_config.name, model_config.api_key)
-    else:
-        raise ValueError(
-            f"Unsupported provider for token counting: {model_config.provider}. "
-            f"Supported providers: {[p.value for p in ProviderType]}"
-        )
-
-    # Cache the instance
-    _token_counter_cache[cache_key] = counter
-    logger.debug(
-        f"Cached token counter for {model_config.provider.value}:{model_config.name}"
-    )
-
-    return counter
-
-
-def count_tokens_from_messages(
-    messages: list[ModelMessage], model_config: ModelConfig
-) -> int:
-    """Count actual tokens from messages using provider-specific methods.
-
-    This replaces the old estimation approach with accurate token counting
-    using each provider's official APIs and libraries.
-
-    Args:
-        messages: List of messages to count tokens for
-        model_config: Model configuration with provider info
-
-    Returns:
-        Exact token count for the messages
-
-    Raises:
-        ValueError: If provider is not supported
-        RuntimeError: If token counting fails
-    """
-    counter = get_token_counter(model_config)
-    return counter.count_message_tokens(messages)
-
-
-def count_post_summary_tokens(
-    messages: list[ModelMessage], summary_index: int, model_config: ModelConfig
-) -> int:
-    """Count actual tokens from summary onwards for incremental compaction decisions.
-
-    Args:
-        messages: Full message history
-        summary_index: Index of the last summary message
-        model_config: Model configuration with provider info
-
-    Returns:
-        Exact token count from summary onwards
-
-    Raises:
-        ValueError: If provider is not supported
-        RuntimeError: If token counting fails
-    """
-    if summary_index >= len(messages):
-        return 0
-
-    post_summary_messages = messages[summary_index:]
-    return count_tokens_from_messages(post_summary_messages, model_config)
-
-
-def count_tokens_from_message_parts(
-    messages: list[ModelMessage], model_config: ModelConfig
-) -> int:
-    """Count actual tokens from message parts for summarization requests.
-
-    Args:
-        messages: List of messages to count tokens for
-        model_config: Model configuration with provider info
-
-    Returns:
-        Exact token count from message parts
-
-    Raises:
-        ValueError: If provider is not supported
-        RuntimeError: If token counting fails
-    """
-    # For now, use the same logic as count_tokens_from_messages
-    # This can be optimized later if needed for different counting strategies
-    return count_tokens_from_messages(messages, model_config)