shotgun-sh 0.1.16.dev2__py3-none-any.whl → 0.2.1.dev2__py3-none-any.whl

shotgun/tui/screens/model_picker.py

@@ -0,0 +1,215 @@
+"""Screen for selecting AI model."""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, cast
+
+from textual import on
+from textual.app import ComposeResult
+from textual.containers import Horizontal, Vertical
+from textual.reactive import reactive
+from textual.screen import Screen
+from textual.widgets import Button, Label, ListItem, ListView, Static
+
+from shotgun.agents.config import ConfigManager
+from shotgun.agents.config.models import MODEL_SPECS, ModelName
+
+if TYPE_CHECKING:
+    from ..app import ShotgunApp
+
+
+# Available models for selection
+AVAILABLE_MODELS = list(ModelName)
+
+
+def _sanitize_model_name_for_id(model_name: ModelName) -> str:
+    """Convert model name to valid Textual ID by replacing dots with hyphens."""
+    return model_name.value.replace(".", "-")
+
+
+class ModelPickerScreen(Screen[None]):
+    """Select AI model to use."""
+
+    CSS = """
+        ModelPicker {
+            layout: vertical;
+        }
+
+        ModelPicker > * {
+            height: auto;
+        }
+
+        #titlebox {
+            height: auto;
+            margin: 2 0;
+            padding: 1;
+            border: hkey $border;
+            content-align: center middle;
+
+            & > * {
+                text-align: center;
+            }
+        }
+
+        #model-picker-title {
+            padding: 1 0;
+            text-style: bold;
+            color: $text-accent;
+        }
+
+        #model-list {
+            margin: 2 0;
+            height: auto;
+            & > * {
+            padding: 1 0;
+            }
+        }
+        #model-actions {
+            padding: 1;
+        }
+        #model-actions > * {
+        margin-right: 2;
+        }
+        #model-list {
+            padding: 1;
+        }
+    """
+
+    BINDINGS = [
+        ("escape", "done", "Back"),
+    ]
+
+    selected_model: reactive[ModelName] = reactive(ModelName.GPT_5)
+
+    def compose(self) -> ComposeResult:
+        with Vertical(id="titlebox"):
+            yield Static("Model selection", id="model-picker-title")
+            yield Static(
+                "Select the AI model you want to use for your tasks.",
+                id="model-picker-summary",
+            )
+        yield ListView(*self._build_model_items(), id="model-list")
+        with Horizontal(id="model-actions"):
+            yield Button("Select \\[ENTER]", variant="primary", id="select")
+            yield Button("Done \\[ESC]", id="done")
+
+    def on_mount(self) -> None:
+        # Load current selection
+        config_manager = self.config_manager
+        config = config_manager.load()
+        current_model = config.selected_model or ModelName.CLAUDE_OPUS_4_1
+        self.selected_model = current_model
+
+        # Find and highlight current selection
+        list_view = self.query_one(ListView)
+        if list_view.children:
+            for i, model_name in enumerate(AVAILABLE_MODELS):
+                if model_name == current_model:
+                    list_view.index = i
+                    break
+        self.refresh_model_labels()
+
+    def action_done(self) -> None:
+        self.dismiss()
+
+    @on(ListView.Highlighted)
+    def _on_model_highlighted(self, event: ListView.Highlighted) -> None:
+        model_name = self._model_from_item(event.item)
+        if model_name:
+            self.selected_model = model_name
+
+    @on(ListView.Selected)
+    def _on_model_selected(self, event: ListView.Selected) -> None:
+        model_name = self._model_from_item(event.item)
+        if model_name:
+            self.selected_model = model_name
+            self._select_model()
+
+    @on(Button.Pressed, "#select")
+    def _on_select_pressed(self) -> None:
+        self._select_model()
+
+    @on(Button.Pressed, "#done")
+    def _on_done_pressed(self) -> None:
+        self.action_done()
+
+    @property
+    def config_manager(self) -> ConfigManager:
+        app = cast("ShotgunApp", self.app)
+        return app.config_manager
+
+    def refresh_model_labels(self) -> None:
+        """Update the list view entries to reflect current selection."""
+        current_model = (
+            self.config_manager.load().selected_model or ModelName.CLAUDE_OPUS_4_1
+        )
+        for model_name in AVAILABLE_MODELS:
+            label = self.query_one(
+                f"#label-{_sanitize_model_name_for_id(model_name)}", Label
+            )
+            label.update(
+                self._model_label(model_name, is_current=model_name == current_model)
+            )
+
+    def _build_model_items(self) -> list[ListItem]:
+        items: list[ListItem] = []
+        current_model = self.selected_model
+        for model_name in AVAILABLE_MODELS:
+            label = Label(
+                self._model_label(model_name, is_current=model_name == current_model),
+                id=f"label-{_sanitize_model_name_for_id(model_name)}",
+            )
+            items.append(
+                ListItem(label, id=f"model-{_sanitize_model_name_for_id(model_name)}")
+            )
+        return items
+
+    def _model_from_item(self, item: ListItem | None) -> ModelName | None:
+        if item is None or item.id is None:
+            return None
+        sanitized_id = item.id.removeprefix("model-")
+        # Find the original model name by comparing sanitized versions
+        for model_name in AVAILABLE_MODELS:
+            if _sanitize_model_name_for_id(model_name) == sanitized_id:
+                return model_name
+        return None
+
+    def _model_label(self, model_name: ModelName, is_current: bool) -> str:
+        """Generate label for model with specs and current indicator."""
+        if model_name not in MODEL_SPECS:
+            return model_name.value
+
+        spec = MODEL_SPECS[model_name]
+        display_name = self._model_display_name(model_name)
+
+        # Format context/output tokens in readable format
+        input_k = spec.max_input_tokens // 1000
+        output_k = spec.max_output_tokens // 1000
+
+        label = f"{display_name} · {input_k}K context · {output_k}K output"
+
+        if is_current:
+            label += " · Current"
+
+        return label
+
+    def _model_display_name(self, model_name: ModelName) -> str:
+        """Get human-readable model name."""
+        names = {
+            ModelName.GPT_5: "GPT-5 (OpenAI)",
+            ModelName.CLAUDE_OPUS_4_1: "Claude Opus 4.1 (Anthropic)",
+            ModelName.CLAUDE_SONNET_4_5: "Claude Sonnet 4.5 (Anthropic)",
+            ModelName.GEMINI_2_5_PRO: "Gemini 2.5 Pro (Google)",
+        }
+        return names.get(model_name, model_name.value)
+
+    def _select_model(self) -> None:
+        """Save the selected model."""
+        try:
+            self.config_manager.update_selected_model(self.selected_model)
+            self.refresh_model_labels()
+            self.notify(
+                f"Selected model: {self._model_display_name(self.selected_model)}"
+            )
+        except Exception as exc:  # pragma: no cover - defensive; textual path
+            self.notify(f"Failed to select model: {exc}", severity="error")

shotgun/tui/screens/provider_config.py

@@ -16,6 +16,9 @@ from shotgun.agents.config import ConfigManager, ProviderType
 if TYPE_CHECKING:
     from ..app import ShotgunApp
 
+# Provider identifiers for configuration screen (LLM providers + Shotgun Account)
+CONFIGURABLE_PROVIDERS = ["openai", "anthropic", "google", "shotgun"]
+
 
 class ProviderConfigScreen(Screen[None]):
     """Collect API keys for available providers."""
@@ -73,7 +76,7 @@ class ProviderConfigScreen(Screen[None]):
         ("escape", "done", "Back"),
     ]
 
-    selected_provider: reactive[ProviderType] = reactive(ProviderType.OPENAI)
+    selected_provider: reactive[str] = reactive("openai")
 
     def compose(self) -> ComposeResult:
         with Vertical(id="titlebox"):
@@ -102,7 +105,7 @@ class ProviderConfigScreen(Screen[None]):
         list_view = self.query_one(ListView)
         if list_view.children:
             list_view.index = 0
-        self.selected_provider = ProviderType.OPENAI
+        self.selected_provider = "openai"
         self.set_focus(self.query_one("#api-key", Input))
 
     def action_done(self) -> None:
@@ -152,45 +155,55 @@ class ProviderConfigScreen(Screen[None]):
 
     def refresh_provider_status(self) -> None:
         """Update the list view entries to reflect configured providers."""
-        for provider in ProviderType:
-            label = self.query_one(f"#label-{provider.value}", Label)
-            label.update(self._provider_label(provider))
+        for provider_id in CONFIGURABLE_PROVIDERS:
+            label = self.query_one(f"#label-{provider_id}", Label)
+            label.update(self._provider_label(provider_id))
 
     def _build_provider_items(self) -> list[ListItem]:
         items: list[ListItem] = []
-        for provider in ProviderType:
-            label = Label(self._provider_label(provider), id=f"label-{provider.value}")
-            items.append(ListItem(label, id=f"provider-{provider.value}"))
+        for provider_id in CONFIGURABLE_PROVIDERS:
+            label = Label(self._provider_label(provider_id), id=f"label-{provider_id}")
+            items.append(ListItem(label, id=f"provider-{provider_id}"))
         return items
 
-    def _provider_from_item(self, item: ListItem | None) -> ProviderType | None:
+    def _provider_from_item(self, item: ListItem | None) -> str | None:
         if item is None or item.id is None:
             return None
         provider_id = item.id.removeprefix("provider-")
-        try:
-            return ProviderType(provider_id)
-        except ValueError:
-            return None
+        return provider_id if provider_id in CONFIGURABLE_PROVIDERS else None
 
-    def _provider_label(self, provider: ProviderType) -> str:
-        display = self._provider_display_name(provider)
+    def _provider_label(self, provider_id: str) -> str:
+        display = self._provider_display_name(provider_id)
         status = (
-            "Configured"
-            if self.config_manager.has_provider_key(provider)
-            else "Not configured"
+            "Configured" if self._has_provider_key(provider_id) else "Not configured"
         )
         return f"{display} · {status}"
 
-    def _provider_display_name(self, provider: ProviderType) -> str:
+    def _provider_display_name(self, provider_id: str) -> str:
         names = {
-            ProviderType.OPENAI: "OpenAI",
-            ProviderType.ANTHROPIC: "Anthropic",
-            ProviderType.GOOGLE: "Google Gemini",
+            "openai": "OpenAI",
+            "anthropic": "Anthropic",
+            "google": "Google Gemini",
+            "shotgun": "Shotgun Account",
         }
-        return names.get(provider, provider.value.title())
-
-    def _input_placeholder(self, provider: ProviderType) -> str:
-        return f"{self._provider_display_name(provider)} API key"
+        return names.get(provider_id, provider_id.title())
+
+    def _input_placeholder(self, provider_id: str) -> str:
+        return f"{self._provider_display_name(provider_id)} API key"
+
+    def _has_provider_key(self, provider_id: str) -> bool:
+        """Check if provider has a configured API key."""
+        if provider_id == "shotgun":
+            # Check shotgun key directly
+            config = self.config_manager.load()
+            return self.config_manager._provider_has_api_key(config.shotgun)
+        else:
+            # Check LLM provider key
+            try:
+                provider = ProviderType(provider_id)
+                return self.config_manager.has_provider_key(provider)
+            except ValueError:
+                return False
 
     def _save_api_key(self) -> None:
         input_widget = self.query_one("#api-key", Input)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.1.16.dev2
+Version: 0.2.1.dev2
 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.1.dev2.dist-info}/RECORD

@@ -1,38 +1,45 @@
 shotgun/__init__.py,sha256=P40K0fnIsb7SKcQrFnXZ4aREjpWchVDhvM1HxI4cyIQ,104
 shotgun/build_constants.py,sha256=RXNxMz46HaB5jucgMVpw8a2yCJqjbhTOh0PddyEVMN8,713
 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=WKVWx77XrD_Jra3-6e-bSUYR4bUYMAsRuQNcy-MpoLY,14499
+shotgun/agents/config/models.py,sha256=ZojhfheNO337e1icy_cE2PpBXIl5oHkdajr4azzFF-U,5106
+shotgun/agents/config/provider.py,sha256=zaQoD8JYEKgTMEJ5etZDSyLhPzD9-EHSOYNfS70BemQ,10931
 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=wQhuz8zE2K-fjwxeK2yZt-NkV13kEZAitBqrwPLUvgg,7466
 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=DrULqhxPPDZzz1WQNjXDY_cls4kXJNexYlI1ISrNTKs,5089
 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
@@ -116,10 +126,11 @@ shotgun/tui/screens/chat.py,sha256=H3uAE4sQ7iJO33tJhdoW_OkzaNxXgW6BUP-SxoNiohw,3
 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=Lbmsh-wCwFpJ9pEANi6uKpSatKJ2kHVocJ9-wCYMd58,6960
+shotgun/tui/screens/provider_config.py,sha256=nJ7mwhabCo9AGTqi03YWhic9uHkRQstN5UCgg4bdTX0,8126
 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=AlBegXO-meob5TfNqjbs8txPXhScxxbcy7600FW_228,9520
 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/utils/__init__.py,sha256=cFjDfoXTRBq29wgP7TGRWUu1eFfiIG-LLOzjIGfadgI,150
@@ -129,8 +140,8 @@ shotgun/utils/env_utils.py,sha256=8QK5aw_f_V2AVTleQQlcL0RnD4sPJWXlDG46fsHu0d8,10
 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.1.dev2.dist-info/METADATA,sha256=21gMEduUKH5hRYdTHdBTg0vQnEX_7H-gHJU5elXs1t0,11226
+shotgun_sh-0.2.1.dev2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+shotgun_sh-0.2.1.dev2.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
+shotgun_sh-0.2.1.dev2.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
+shotgun_sh-0.2.1.dev2.dist-info/RECORD,,