shotgun-sh 0.2.3.dev2__py3-none-any.whl → 0.2.11.dev1__py3-none-any.whl

shotgun/tui/containers.py

@@ -0,0 +1,93 @@
+"""Dependency injection container for TUI components."""
+
+from typing import TYPE_CHECKING
+
+from dependency_injector import containers, providers
+from pydantic_ai import RunContext
+
+from shotgun.agents.agent_manager import AgentManager
+from shotgun.agents.config import get_provider_model
+from shotgun.agents.conversation_manager import ConversationManager
+from shotgun.agents.models import AgentDeps, AgentType
+from shotgun.sdk.codebase import CodebaseSDK
+from shotgun.tui.commands import CommandHandler
+from shotgun.tui.filtered_codebase_service import FilteredCodebaseService
+from shotgun.tui.services.conversation_service import ConversationService
+from shotgun.tui.state.processing_state import ProcessingStateManager
+from shotgun.tui.utils.mode_progress import PlaceholderHints
+from shotgun.tui.widgets.widget_coordinator import WidgetCoordinator
+from shotgun.utils import get_shotgun_home
+
+if TYPE_CHECKING:
+    pass
+
+
+# Placeholder system prompt function (agents provide their own)
+# Using Object provider to pass the function itself, not call it
+def _placeholder_system_prompt(ctx: "RunContext[AgentDeps]") -> str:
+    raise RuntimeError(
+        "This should not be called - agents provide their own system_prompt_fn"
+    )
+
+
+class TUIContainer(containers.DeclarativeContainer):
+    """Dependency injection container for TUI components.
+
+    This container manages the lifecycle and dependencies of all TUI components,
+    ensuring consistent configuration and facilitating testing.
+    """
+
+    # Configuration
+    config = providers.Configuration()
+
+    # Core dependencies
+    model_config = providers.Singleton(get_provider_model)
+
+    storage_dir = providers.Singleton(lambda: get_shotgun_home() / "codebases")
+
+    codebase_service = providers.Singleton(
+        FilteredCodebaseService, storage_dir=storage_dir
+    )
+
+    system_prompt_fn = providers.Object(_placeholder_system_prompt)
+
+    # AgentDeps singleton
+    agent_deps = providers.Singleton(
+        AgentDeps,
+        interactive_mode=True,
+        is_tui_context=True,
+        llm_model=model_config,
+        codebase_service=codebase_service,
+        system_prompt_fn=system_prompt_fn,
+    )
+
+    # Service singletons
+    codebase_sdk = providers.Singleton(CodebaseSDK)
+
+    command_handler = providers.Singleton(CommandHandler)
+
+    placeholder_hints = providers.Singleton(PlaceholderHints)
+
+    conversation_manager = providers.Singleton(ConversationManager)
+
+    conversation_service = providers.Factory(
+        ConversationService, conversation_manager=conversation_manager
+    )
+
+    # Factory for AgentManager (needs agent_type parameter)
+    agent_manager_factory = providers.Factory(
+        AgentManager, deps=agent_deps, initial_type=providers.Object(AgentType.RESEARCH)
+    )
+
+    # Factory for ProcessingStateManager (needs ChatScreen reference)
+    processing_state_factory = providers.Factory(
+        ProcessingStateManager,
+        screen=providers.Object(None),  # Will be overridden when creating ChatScreen
+        telemetry_context=providers.Object({}),  # Will be overridden when creating
+    )
+
+    # Factory for WidgetCoordinator (needs ChatScreen reference)
+    widget_coordinator_factory = providers.Factory(
+        WidgetCoordinator,
+        screen=providers.Object(None),  # Will be overridden when creating ChatScreen
+    )

shotgun/tui/dependencies.py

@@ -0,0 +1,39 @@
+"""Dependency creation utilities for TUI components."""
+
+from pydantic_ai import RunContext
+
+from shotgun.agents.config import get_provider_model
+from shotgun.agents.models import AgentDeps
+from shotgun.tui.filtered_codebase_service import FilteredCodebaseService
+from shotgun.utils import get_shotgun_home
+
+
+def create_default_tui_deps() -> AgentDeps:
+    """Create default AgentDeps for TUI components.
+
+    This creates a standard AgentDeps configuration suitable for interactive
+    TUI usage with:
+    - Interactive mode enabled
+    - TUI context flag set
+    - Filtered codebase service (restricted to CWD)
+    - Placeholder system prompt (agents provide their own)
+
+    Returns:
+        Configured AgentDeps instance ready for TUI use.
+    """
+    model_config = get_provider_model()
+    storage_dir = get_shotgun_home() / "codebases"
+    codebase_service = FilteredCodebaseService(storage_dir)
+
+    def _placeholder_system_prompt_fn(ctx: RunContext[AgentDeps]) -> str:
+        raise RuntimeError(
+            "This should not be called - agents provide their own system_prompt_fn"
+        )
+
+    return AgentDeps(
+        interactive_mode=True,
+        is_tui_context=True,
+        llm_model=model_config,
+        codebase_service=codebase_service,
+        system_prompt_fn=_placeholder_system_prompt_fn,
+    )

shotgun/tui/protocols.py

@@ -0,0 +1,45 @@
+"""Protocol definitions for TUI components.
+
+These protocols define interfaces that components can depend on without
+creating circular imports. Screens like ChatScreen can satisfy these
+protocols without explicitly implementing them.
+"""
+
+from typing import Protocol, runtime_checkable
+
+
+@runtime_checkable
+class QAStateProvider(Protocol):
+    """Protocol for screens that provide Q&A mode state.
+
+    This protocol allows components to check if they're on a screen with
+    Q&A mode without importing the concrete ChatScreen class, eliminating
+    circular dependencies.
+    """
+
+    @property
+    def qa_mode(self) -> bool:
+        """Whether Q&A mode is currently active.
+
+        Returns:
+            True if Q&A mode is active, False otherwise.
+        """
+        ...
+
+
+@runtime_checkable
+class ProcessingStateProvider(Protocol):
+    """Protocol for screens that provide processing state.
+
+    This protocol allows components to check if they're on a screen with
+    an active agent processing without importing the concrete ChatScreen class.
+    """
+
+    @property
+    def working(self) -> bool:
+        """Whether an agent is currently working.
+
+        Returns:
+            True if an agent is processing, False otherwise.
+        """
+        ...

shotgun/tui/screens/chat/__init__.py

@@ -0,0 +1,5 @@
+"""Chat screen module."""
+
+from shotgun.tui.screens.chat.chat_screen import ChatScreen
+
+__all__ = ["ChatScreen"]

shotgun/tui/screens/chat/chat.tcss

@@ -0,0 +1,54 @@
+ChatHistory {
+  height: auto;
+}
+
+PromptInput {
+  min-height: 3;
+  max-height: 7;
+  height: auto;
+}
+
+StatusBar {
+  height: auto;
+}
+
+ModeIndicator {
+  height: auto;
+}
+
+#footer {
+  dock: bottom;
+  height: auto;
+  padding: 1 1 1 2;
+  max-height: 14;
+}
+
+#window {
+  align: left bottom;
+}
+
+.hidden {
+  display: none;
+}
+
+#footer > Grid {
+  height: auto;
+  grid-columns: 1fr auto;
+  grid-size: 2;
+}
+
+
+#right-footer-indicators {
+  width: auto;
+  height: auto;
+  layout: vertical;
+}
+
+#context-indicator {
+  text-align: end;
+  height: 1;
+}
+
+#indexing-job-display {
+  text-align: end;
+}