agentic-cli 0.4.2__tar.gz → 0.4.4__tar.gz

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/.gitignore

@@ -43,10 +43,11 @@ Thumbs.db
 
 #CLAUDE.md
 .claude/
+.research_demo/
 plans/
 docs/
 changes/
-.research_demo/
+findings/
 
 # Git worktrees
 .worktrees/

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/CHANGELOG.md

@@ -5,6 +5,58 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.4.4] - 2026-03-11
+
+### Added
+- **Session Save/Resume**: Persistent conversations across CLI exits with `/save` and `/resume`
+- **Sandbox Executor**: Stateful multi-turn Python code execution with sandboxing and default deps
+- **Context Window Management**: Native ADK and LangGraph context trimming with source-level detection
+- **Context Window Visibility**: Trim detection, status bar token display, and `/status` breakdown
+- **Dynamic Model Registry**: Replace static model lists with live API discovery from Google/Anthropic
+- **Task Progress Display**: Rich-colored task progress in thinking box, persisted across turns
+
+### Changed
+- **Architecture**: Separated workflow+tools layer from UI/CLI layer; composable settings mixins
+- **DRY/SOLID Cleanup** (PR #59, #60): Deduplicated config paths, ArXiv parsing, session persistence, artifact loading; extracted `drain_trim_events()`, `format_detail_rows()`, `format_task_checklist()`; removed dead code (`get_help()`, `FinanceResearchState`, unused settings introspection); replaced dynamic `type()` with `_SessionEvent` class; moved token-drop heuristic into `UsageTracker`
+- **MessageProcessor**: Removed dead code, unused params, cleaner state init
+- **HITL**: Removed dead Future-based machinery, simplified callback path
+- **Ripgrep**: Cache availability check with `lru_cache`
+
+### Fixed
+- Executor import validation blocking allowed submodules
+- Settings not persisting (replaced `exclude_defaults` with explicit exclusion)
+- `verbose_thinking` not persisting across restarts
+- Thread safety, path traversal, and TOCTOU race conditions in persistence
+- VectorStore crash safety: write mappings before FAISS index
+- Embeddings: update in-memory state after `ingest_document`
+- `open_document`: extension allowlist, DANGEROUS permission, audit logging
+
+### Removed
+- `FinanceResearchState` (domain-specific code in shared library)
+- Dead `get_help()`, `get_ui_section`, `is_ui_hidden` methods
+- Dead `generate_tool_summary`, `clear_context`/`unbind_context`, model constant re-exports
+- Dead `cli/settings.py` re-export shim
+
+## [0.4.3] - 2026-02-12
+
+### Changed
+- **Overengineering Cleanup**: Removed ~550 lines of dead code — StateSnapshot, SearchSourceRegistry, unused settings/tool fields, dead ToolRegistry methods, dead command APIs
+- **Inlined Resolvers**: ModelResolver and PathResolver inlined into BaseSettings (`resolvers.py` now contains only constants)
+- **HITL Consolidation**: Merged `hitl/` package into single `tools/hitl_tools.py`; removed `create_checkpoint` tool
+- **Persistence Hardening**: Extracted `sanitize_filename` utility, hardened atomic writes in `persistence/_utils.py`
+- **Test Mocks Relocated**: MockEmbeddingService and MockVectorStore moved to `knowledge_base/_mocks.py`
+
+### Fixed
+- Broken example demos: updated stale imports in arxiv_demo, fileops_demo, shell_demo, websearch_demo
+- Disruptive `logger.error` in background init bypassing CLI UI
+- Research demo: removed stale README, trimmed prompt bloat
+
+### Removed
+- `create_checkpoint` tool (HITL simplified to `request_approval` only)
+- `SearchSourceRegistry` class (unused abstraction)
+- `StateSnapshot`, `ToolCallRecord` data classes (unused)
+- 3 unused conftest test fixtures
+
 ## [0.4.2] - 2026-02-08
 
 ### Added

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/CLAUDE.md

@@ -7,7 +7,7 @@ Agentic CLI is a shared library providing the core infrastructure for building d
 ## Tech Stack
 
 - **Language**: Python 3.12+
-- **CLI UI**: `thinking_prompt` - enhanced CLI with thinking boxes and markdown
+- **CLI UI**: `thinking-prompt` - enhanced CLI with thinking boxes and markdown
 - **Workflow**: Google ADK + LangGraph - dual orchestration backends (selectable via settings)
 - **Config**: `pydantic-settings` - type-safe configuration
 - **Logging**: `structlog` - structured logging
@@ -20,7 +20,6 @@ agentic-cli/
 │   ├── __init__.py           # Package exports, lazy imports
 │   ├── config.py             # BaseSettings (pydantic-settings)
 │   ├── constants.py          # Shared constants, truncate()
-│   ├── resolvers.py          # ModelResolver, PathResolver
 │   ├── settings_persistence.py
 │   ├── logging.py
 │   ├── cli/
@@ -63,24 +62,21 @@ agentic-cli/
 │   │   ├── memory_tools.py   # save_memory, search_memory + MemoryStore
 │   │   ├── planning_tools.py # save_plan, get_plan + PlanStore
 │   │   ├── task_tools.py     # save_tasks, get_tasks + TaskStore
-│   │   ├── hitl_tools.py     # request_approval, create_checkpoint
+│   │   ├── hitl_tools.py     # request_approval + ApprovalManager, HITLConfig
 │   │   ├── shell/            # 8-layer shell security
 │   │   └── webfetch/         # Fetcher, converter, validator, robots
-│   ├── hitl/
-│   │   ├── config.py         # ApprovalRule, HITLConfig
-│   │   ├── approval.py       # ApprovalManager
-│   │   └── checkpoints.py    # CheckpointManager
 │   ├── knowledge_base/
 │   │   ├── models.py         # Document, SearchResult
 │   │   ├── embeddings.py     # EmbeddingService
 │   │   ├── vector_store.py   # VectorStore (FAISS)
+│   │   ├── _mocks.py         # MockEmbeddingService, MockVectorStore
 │   │   └── manager.py        # KnowledgeBaseManager
 │   └── persistence/
 │       ├── session.py        # SessionPersistence
 │       ├── artifacts.py      # ArtifactManager
 │       └── _utils.py         # Atomic write utilities
 ├── tests/
-│   ├── conftest.py           # MockContext, MockVectorStore, MockEmbeddingService
+│   ├── conftest.py           # MockContext, shared fixtures
 │   ├── test_*.py             # Unit tests
 │   ├── tools/                # Tool-specific tests
 │   └── integration/          # ADK & LangGraph pipeline tests
@@ -110,10 +106,12 @@ conda run -n agenticcli python -c "from agentic_cli import BaseCLIApp; print(Bas
 - **main**: Stable branch, matches latest release. Only updated via merges from `develop` when releasing.
 - **develop**: Integration branch for ongoing work. Small fixes can be committed directly here.
 - **feature/\***: Feature branches for larger changes. Branch from `develop`, merge back to `develop`.
+- **fix/\***: Fix branches for fixing issues. Branch from `develop`, merge back to `develop`.
+- **refactor/\***: For larger refactoring changes. Branch from `develop`, merge back to `develop`.
 
 Workflow:
 1. For small fixes: commit directly to `develop`
-2. For features: create `feature/<name>` from `develop`, work there, merge back to `develop`
+2. For features: create `feature/<name>` (or `fix/<name>` or `refactor/<name>`) from `develop`, work there, merge back to `develop`
 3. When ready to release: merge `develop` → `main` and tag the release
 
 ## Development Principles
@@ -133,7 +131,7 @@ Workflow:
 ### Key Design Patterns
 - **Tool error handling**: All tools return `{"success": bool, ...}` dicts. Never raise `ToolError`.
 - **Tool registration**: Use `@register_tool(category=..., permission_level=..., description=...)` decorator. Tools are auto-discovered via the global `ToolRegistry`.
-- **Store consolidation**: Stores (MemoryStore, PlanStore, TaskStore) live inside their tool files (e.g. `memory_tools.py`), not in separate packages.
+- **Store consolidation**: Stores and managers (MemoryStore, PlanStore, TaskStore, ApprovalManager) live inside their tool files (e.g. `memory_tools.py`, `hitl_tools.py`), not in separate packages.
 - **Context access**: Tools use `get_context_*()` functions from `workflow.context` to access managers and stores via ContextVars.
 - **Atomic writes**: Use `atomic_write_json`/`atomic_write_text` from `persistence/_utils.py` for file persistence.
 
@@ -153,6 +151,6 @@ Available session methods:
 
 - **Framework**: pytest with `asyncio_mode = "auto"`
 - **MockContext**: From `tests/conftest.py` — provides isolated settings and temp dirs for all tests
-- **MockVectorStore** and **MockEmbeddingService**: In-tree mocks for testing without ML dependencies
+- **MockVectorStore** and **MockEmbeddingService**: In `knowledge_base/_mocks.py` for testing without ML dependencies
 - **FAISS tests**: Guard with `pytest.importorskip("faiss")` since FAISS is not installed in dev env
 - **Integration tests**: `tests/integration/` covers ADK and LangGraph pipeline tests

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentic-cli
-Version: 0.4.2
+Version: 0.4.4
 Summary: A framework for building domain-specific agentic CLI applications
 Project-URL: Homepage, https://github.com/shoom1/agentic-cli
 Project-URL: Repository, https://github.com/shoom1/agentic-cli
@@ -17,6 +17,8 @@ Requires-Python: >=3.12
 Requires-Dist: feedparser>=6.0.0
 Requires-Dist: google-adk[genai]>=0.4.0
 Requires-Dist: httpx>=0.27.0
+Requires-Dist: ipykernel>=6.0.0
+Requires-Dist: jupyter-client>=8.0.0
 Requires-Dist: numpy>=1.26.0
 Requires-Dist: prompt-toolkit>=3.0.0
 Requires-Dist: pydantic-settings>=2.0.0
@@ -24,7 +26,7 @@ Requires-Dist: pydantic>=2.0.0
 Requires-Dist: pypdf>=4.0.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: structlog>=24.0.0
-Requires-Dist: thinking-prompt>=0.2.0
+Requires-Dist: thinking-prompt>=0.2.5
 Provides-Extra: dev
 Requires-Dist: pytest-asyncio>=0.24.0; extra == 'dev'
 Requires-Dist: pytest-cov>=6.0.0; extra == 'dev'
@@ -384,7 +386,7 @@ agent = AgentConfig(
 )
 
 # Or call directly
-results = web_search("Python async programming", max_results=5)
+results = await web_search("Python async programming", max_results=5)
 # Returns: {"results": [{"title": "...", "url": "...", "snippet": "..."}], ...}
 ```
 
@@ -397,7 +399,7 @@ Fetch web content and summarize with LLM:
 ```python
 from agentic_cli.tools import web_fetch
 
-result = web_fetch(
+result = await web_fetch(
     url="https://example.com/article",
     prompt="Extract the main points from this article",
 )
@@ -411,16 +413,13 @@ Features: URL validation, robots.txt compliance, SSRF protection, content cachin
 Search and analyze academic papers:
 
 ```python
-from agentic_cli.tools import search_arxiv, fetch_arxiv_paper, analyze_arxiv_paper
+from agentic_cli.tools import search_arxiv, fetch_arxiv_paper
 
 # Search papers
 results = search_arxiv("transformer attention", max_results=10, categories=["cs.CL"])
 
 # Fetch paper details
 paper = fetch_arxiv_paper("1706.03762")  # "Attention Is All You Need"
-
-# Analyze with LLM
-analysis = await analyze_arxiv_paper("1706.03762", "Summarize the key contributions")
 ```
 
 #### File Operations
@@ -449,8 +448,8 @@ result = list_dir("src/", include_hidden=False)
 # Returns: {"success": True, "entries": [...]}
 
 # Compare files or text
-result = diff_compare(source1="old.txt", source2="new.txt")
-# Returns: {"success": True, "diff": "...", "has_changes": True}
+result = diff_compare(source_a="old.txt", source_b="new.txt")
+# Returns: {"success": True, "diff": "...", "similarity": 0.85}
 ```
 
 **WRITE Tools (Caution)**
@@ -538,8 +537,9 @@ class MyCommand(Command):
 
 # In your app
 class MyApp(BaseCLIApp):
-    def get_custom_commands(self) -> list[Command]:
-        return [MyCommand()]
+    def register_commands(self) -> None:
+        super().register_commands()
+        self.command_registry.register(MyCommand())
 ```
 
 ## Events
@@ -602,7 +602,7 @@ See the `examples/` directory for complete working examples:
 **Feature Demos**
 - **arxiv_demo.py** - ArXiv paper search and analysis
 - **fileops_demo.py** - File operation tools (read, write, grep, glob)
-- **memory_demo.py** - Working and long-term memory management
+- **memory_demo.py** - Memory persistence system
 - **planning_demo.py** - Task graph and planning tools
 - **shell_demo.py** - Shell security pattern detection
 - **webfetch_demo.py** - Web fetching and summarization
@@ -652,27 +652,43 @@ agentic-cli/
 │   ├── __init__.py           # Package exports
 │   ├── config.py             # BaseSettings, SettingsContext
 │   ├── constants.py          # Shared constants (truncation, limits)
+│   ├── resolvers.py          # Model/path constants (GOOGLE_MODELS, etc.)
+│   ├── settings_persistence.py
 │   ├── logging.py            # Structlog configuration
 │   ├── cli/
 │   │   ├── app.py            # BaseCLIApp
 │   │   ├── commands.py       # Command, CommandRegistry
 │   │   ├── builtin_commands.py
 │   │   ├── workflow_controller.py  # Workflow orchestration
-│   │   └── message_processor.py    # Event stream processing
+│   │   ├── message_processor.py    # Event stream processing
+│   │   └── settings*.py      # Settings UI (introspection, dialog)
 │   ├── workflow/
 │   │   ├── base_manager.py   # BaseWorkflowManager (abstract)
 │   │   ├── events.py         # WorkflowEvent, EventType
 │   │   ├── config.py         # AgentConfig
 │   │   ├── context.py        # Context variables for tools
 │   │   ├── thinking.py       # ThinkingDetector
-│   │   ├── adk_manager.py    # GoogleADKWorkflowManager
+│   │   ├── task_progress.py  # Task progress events
+│   │   ├── tool_summaries.py # Tool result summaries
+│   │   ├── settings.py       # WorkflowSettingsMixin
+│   │   ├── adk/              # ADK orchestrator
+│   │   │   ├── manager.py    # GoogleADKWorkflowManager
+│   │   │   ├── event_processor.py  # ADK event processing
+│   │   │   └── llm_event_logger.py # LLM traffic logging
 │   │   └── langgraph/        # LangGraph submodule
 │   │       ├── manager.py    # LangGraphWorkflowManager
-│   │       ├── state.py      # AgentState, CheckpointData
-│   │       └── persistence/  # Checkpointers and stores
+│   │       ├── graph_builder.py # Graph + LLM factory
+│   │       ├── state.py      # AgentState
+│   │       ├── persistence/  # Checkpointers and stores
+│   │       └── tools/        # LangChain-compatible wrappers
 │   ├── tools/
 │   │   ├── registry.py       # ToolRegistry, ToolCategory, PermissionLevel
 │   │   ├── executor.py       # SafePythonExecutor
+│   │   ├── arxiv_tools.py    # search_arxiv, fetch_arxiv_paper
+│   │   ├── arxiv_source.py   # ArxivSearchSource
+│   │   ├── execution_tools.py # execute_python
+│   │   ├── interaction_tools.py # ask_clarification
+│   │   ├── knowledge_tools.py # search/ingest_to_knowledge_base
 │   │   ├── file_read.py      # read_file, diff_compare
 │   │   ├── file_write.py     # write_file, edit_file
 │   │   ├── grep_tool.py      # grep (pattern search)
@@ -682,8 +698,7 @@ agentic-cli/
 │   │   ├── memory_tools.py   # MemoryStore, save/search_memory
 │   │   ├── planning_tools.py # PlanStore, save/get_plan
 │   │   ├── task_tools.py     # TaskStore, save/get_tasks
-│   │   ├── hitl_tools.py     # request_approval, create_checkpoint
-│   │   ├── standard.py       # ArXiv tools, ask_clarification
+│   │   ├── hitl_tools.py     # request_approval, ApprovalManager
 │   │   └── shell/            # Shell executor with security
 │   │       ├── executor.py   # Main entry point (disabled by default)
 │   │       ├── tokenizer.py  # Command parsing
@@ -695,15 +710,12 @@ agentic-cli/
 │   │   ├── models.py         # Document, SearchResult
 │   │   ├── embeddings.py     # EmbeddingService
 │   │   ├── vector_store.py   # VectorStore
-│   │   └── sources.py        # ArxivSearchSource, SearchSourceRegistry
-│   ├── persistence/
-│   │   ├── session.py        # SessionPersistence
-│   │   ├── artifacts.py      # ArtifactManager
-│   │   └── _utils.py         # Atomic write utilities
-│   └── hitl/
-│       ├── approval.py       # ApprovalManager
-│       ├── checkpoints.py    # CheckpointManager
-│       └── config.py         # HITLConfig
+│   │   ├── _mocks.py         # MockEmbeddingService, MockVectorStore
+│   │   └── sources.py        # ArxivSearchSource
+│   └── persistence/
+│       ├── session.py        # SessionPersistence
+│       ├── artifacts.py      # ArtifactManager
+│       └── _utils.py         # Atomic write utilities
 ├── examples/
 │   ├── hello_agent.py        # Basic ADK example
 │   ├── hello_langgraph.py    # Basic LangGraph example

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/README.md

@@ -344,7 +344,7 @@ agent = AgentConfig(
 )
 
 # Or call directly
-results = web_search("Python async programming", max_results=5)
+results = await web_search("Python async programming", max_results=5)
 # Returns: {"results": [{"title": "...", "url": "...", "snippet": "..."}], ...}
 ```
 
@@ -357,7 +357,7 @@ Fetch web content and summarize with LLM:
 ```python
 from agentic_cli.tools import web_fetch
 
-result = web_fetch(
+result = await web_fetch(
     url="https://example.com/article",
     prompt="Extract the main points from this article",
 )
@@ -371,16 +371,13 @@ Features: URL validation, robots.txt compliance, SSRF protection, content cachin
 Search and analyze academic papers:
 
 ```python
-from agentic_cli.tools import search_arxiv, fetch_arxiv_paper, analyze_arxiv_paper
+from agentic_cli.tools import search_arxiv, fetch_arxiv_paper
 
 # Search papers
 results = search_arxiv("transformer attention", max_results=10, categories=["cs.CL"])
 
 # Fetch paper details
 paper = fetch_arxiv_paper("1706.03762")  # "Attention Is All You Need"
-
-# Analyze with LLM
-analysis = await analyze_arxiv_paper("1706.03762", "Summarize the key contributions")
 ```
 
 #### File Operations
@@ -409,8 +406,8 @@ result = list_dir("src/", include_hidden=False)
 # Returns: {"success": True, "entries": [...]}
 
 # Compare files or text
-result = diff_compare(source1="old.txt", source2="new.txt")
-# Returns: {"success": True, "diff": "...", "has_changes": True}
+result = diff_compare(source_a="old.txt", source_b="new.txt")
+# Returns: {"success": True, "diff": "...", "similarity": 0.85}
 ```
 
 **WRITE Tools (Caution)**
@@ -498,8 +495,9 @@ class MyCommand(Command):
 
 # In your app
 class MyApp(BaseCLIApp):
-    def get_custom_commands(self) -> list[Command]:
-        return [MyCommand()]
+    def register_commands(self) -> None:
+        super().register_commands()
+        self.command_registry.register(MyCommand())
 ```
 
 ## Events
@@ -562,7 +560,7 @@ See the `examples/` directory for complete working examples:
 **Feature Demos**
 - **arxiv_demo.py** - ArXiv paper search and analysis
 - **fileops_demo.py** - File operation tools (read, write, grep, glob)
-- **memory_demo.py** - Working and long-term memory management
+- **memory_demo.py** - Memory persistence system
 - **planning_demo.py** - Task graph and planning tools
 - **shell_demo.py** - Shell security pattern detection
 - **webfetch_demo.py** - Web fetching and summarization
@@ -612,27 +610,43 @@ agentic-cli/
 │   ├── __init__.py           # Package exports
 │   ├── config.py             # BaseSettings, SettingsContext
 │   ├── constants.py          # Shared constants (truncation, limits)
+│   ├── resolvers.py          # Model/path constants (GOOGLE_MODELS, etc.)
+│   ├── settings_persistence.py
 │   ├── logging.py            # Structlog configuration
 │   ├── cli/
 │   │   ├── app.py            # BaseCLIApp
 │   │   ├── commands.py       # Command, CommandRegistry
 │   │   ├── builtin_commands.py
 │   │   ├── workflow_controller.py  # Workflow orchestration
-│   │   └── message_processor.py    # Event stream processing
+│   │   ├── message_processor.py    # Event stream processing
+│   │   └── settings*.py      # Settings UI (introspection, dialog)
 │   ├── workflow/
 │   │   ├── base_manager.py   # BaseWorkflowManager (abstract)
 │   │   ├── events.py         # WorkflowEvent, EventType
 │   │   ├── config.py         # AgentConfig
 │   │   ├── context.py        # Context variables for tools
 │   │   ├── thinking.py       # ThinkingDetector
-│   │   ├── adk_manager.py    # GoogleADKWorkflowManager
+│   │   ├── task_progress.py  # Task progress events
+│   │   ├── tool_summaries.py # Tool result summaries
+│   │   ├── settings.py       # WorkflowSettingsMixin
+│   │   ├── adk/              # ADK orchestrator
+│   │   │   ├── manager.py    # GoogleADKWorkflowManager
+│   │   │   ├── event_processor.py  # ADK event processing
+│   │   │   └── llm_event_logger.py # LLM traffic logging
 │   │   └── langgraph/        # LangGraph submodule
 │   │       ├── manager.py    # LangGraphWorkflowManager
-│   │       ├── state.py      # AgentState, CheckpointData
-│   │       └── persistence/  # Checkpointers and stores
+│   │       ├── graph_builder.py # Graph + LLM factory
+│   │       ├── state.py      # AgentState
+│   │       ├── persistence/  # Checkpointers and stores
+│   │       └── tools/        # LangChain-compatible wrappers
 │   ├── tools/
 │   │   ├── registry.py       # ToolRegistry, ToolCategory, PermissionLevel
 │   │   ├── executor.py       # SafePythonExecutor
+│   │   ├── arxiv_tools.py    # search_arxiv, fetch_arxiv_paper
+│   │   ├── arxiv_source.py   # ArxivSearchSource
+│   │   ├── execution_tools.py # execute_python
+│   │   ├── interaction_tools.py # ask_clarification
+│   │   ├── knowledge_tools.py # search/ingest_to_knowledge_base
 │   │   ├── file_read.py      # read_file, diff_compare
 │   │   ├── file_write.py     # write_file, edit_file
 │   │   ├── grep_tool.py      # grep (pattern search)
@@ -642,8 +656,7 @@ agentic-cli/
 │   │   ├── memory_tools.py   # MemoryStore, save/search_memory
 │   │   ├── planning_tools.py # PlanStore, save/get_plan
 │   │   ├── task_tools.py     # TaskStore, save/get_tasks
-│   │   ├── hitl_tools.py     # request_approval, create_checkpoint
-│   │   ├── standard.py       # ArXiv tools, ask_clarification
+│   │   ├── hitl_tools.py     # request_approval, ApprovalManager
 │   │   └── shell/            # Shell executor with security
 │   │       ├── executor.py   # Main entry point (disabled by default)
 │   │       ├── tokenizer.py  # Command parsing
@@ -655,15 +668,12 @@ agentic-cli/
 │   │   ├── models.py         # Document, SearchResult
 │   │   ├── embeddings.py     # EmbeddingService
 │   │   ├── vector_store.py   # VectorStore
-│   │   └── sources.py        # ArxivSearchSource, SearchSourceRegistry
-│   ├── persistence/
-│   │   ├── session.py        # SessionPersistence
-│   │   ├── artifacts.py      # ArtifactManager
-│   │   └── _utils.py         # Atomic write utilities
-│   └── hitl/
-│       ├── approval.py       # ApprovalManager
-│       ├── checkpoints.py    # CheckpointManager
-│       └── config.py         # HITLConfig
+│   │   ├── _mocks.py         # MockEmbeddingService, MockVectorStore
+│   │   └── sources.py        # ArxivSearchSource
+│   └── persistence/
+│       ├── session.py        # SessionPersistence
+│       ├── artifacts.py      # ArtifactManager
+│       └── _utils.py         # Atomic write utilities
 ├── examples/
 │   ├── hello_agent.py        # Basic ADK example
 │   ├── hello_langgraph.py    # Basic LangGraph example

{agentic_cli-0.4.2 → agentic_cli-0.4.4}/examples/arxiv_demo.py

@@ -22,11 +22,8 @@ Usage:
 import sys
 import time
 
-from agentic_cli.knowledge_base.sources import (
-    ArxivSearchSource,
-    SearchSourceResult,
-    get_search_registry,
-)
+from agentic_cli.knowledge_base.sources import SearchSourceResult
+from agentic_cli.tools.arxiv_source import ArxivSearchSource
 
 
 # =============================================================================
@@ -234,34 +231,6 @@ def demo_rate_limiting():
         print()
 
 
-def demo_registry_integration():
-    """Demo integration with search source registry."""
-    print("\n" + "=" * 60)
-    print("Registry Integration Demo")
-    print("=" * 60)
-
-    registry = get_search_registry()
-
-    # List registered sources
-    sources = registry.list_sources()
-    print(f"  Registered sources: {[s.name for s in sources]}")
-
-    # Get arXiv source from registry
-    arxiv = registry.get("arxiv")
-    if arxiv:
-        print(f"  arXiv source available: {arxiv.is_available()}")
-
-    # Search through registry
-    print("\n  Searching via registry...")
-    results = registry.search("reinforcement learning", sources=["arxiv"], max_results=2)
-
-    arxiv_results = results.get("arxiv", [])
-    print(f"  Results from arXiv: {len(arxiv_results)}")
-    for r in arxiv_results:
-        print(f"    - {r.title[:50]}...")
-    print()
-
-
 def main():
     """Run all demos."""
     print("\n" + "#" * 60)
@@ -292,7 +261,6 @@ def main():
     demo_sort_options()
     demo_caching()
     demo_rate_limiting()
-    demo_registry_integration()
 
     print("\n" + "#" * 60)
     print("#  Demo Complete!")