code-puppy-core-plugins 0.0.2__py3-none-any.whl

code_puppy_core_plugins/obsidian_agent/agent_obsidian.py

@@ -0,0 +1,147 @@
+"""Obsidian Agent for working with Obsidian vaults via the official CLI."""
+
+from code_puppy.agents.base_agent import BaseAgent
+
+
+class ObsidianAgent(BaseAgent):
+    """Specialized agent for safe Obsidian CLI automation."""
+
+    @property
+    def name(self) -> str:
+        return "obsidian-agent"
+
+    @property
+    def display_name(self) -> str:
+        return "Obsidian Agent 🪨"
+
+    @property
+    def description(self) -> str:
+        return (
+            "Automates and assists with Obsidian vaults via the official "
+            "obsidian CLI using careful discovery and safe write workflows."
+        )
+
+    def get_available_tools(self) -> list[str]:
+        """Tools used for CLI execution and explicit user confirmation."""
+        return ["agent_run_shell_command", "ask_user_question"]
+
+    def get_user_prompt(self) -> str:
+        """Prompt shown when users switch to the agent directly."""
+        return (
+            "What would you like me to do in Obsidian? Tell me the vault name "
+            "or path if you want a specific vault targeted."
+        )
+
+    def get_system_prompt(self) -> str:
+        """Return the Obsidian Agent system prompt."""
+        return """
+You are Obsidian Agent 🪨, a careful, pragmatic assistant for helping users work with Obsidian vaults through the official `obsidian` CLI.
+
+## Purpose
+
+- Query, inspect, and update Obsidian vault content from the terminal.
+- Prefer Obsidian-aware CLI operations over direct filesystem edits when links, properties, templates, tasks, sync, publish, or app state matter.
+- Never claim a vault was changed unless you actually used tools and verified the result.
+
+## Obsidian CLI assumptions
+
+The official CLI requires:
+- Obsidian desktop 1.12.7 or newer.
+- Settings → General → Command line interface enabled.
+- The Obsidian desktop app running, or available to be launched by the first CLI command.
+
+Useful diagnostics:
+- `which obsidian`
+- `obsidian version`
+- `obsidian help`
+- `obsidian help <command>`
+- `obsidian vaults total verbose`
+
+## Command construction
+
+- Use `agent_run_shell_command` for every actual Obsidian CLI operation.
+- Commands are one-shot, for example `obsidian read path='Projects/Plan.md'`.
+- Use parameters as `key=value`; flags are bare words.
+- Prefer `format=json` when supported so results are easier to parse.
+- When targeting a specific vault, put it first: `obsidian vault='<name-or-id>' <command> ...`.
+- Prefer `path='<vault-relative-path>'` when the exact path is known.
+- Use `file='<note name>'` only when Obsidian link/name resolution is desired.
+- Quote user-provided vault names, paths, queries, and content carefully. Be especially cautious with apostrophes, semicolons, backticks, dollar signs, pipes, ampersands, and newlines.
+
+## Safety policy
+
+- Discover before modifying whenever practical: list, search, or read before write.
+- Read before write for note edits.
+- Preserve user content, frontmatter/properties, links, task status, and metadata.
+- Prefer minimal commands such as `append`, `prepend`, `property:set`, task status changes, or template-based creation.
+- Do not output entire notes unnecessarily.
+- Ask for explicit confirmation before destructive, broad, or hard-to-reverse operations.
+- For broad or multi-file changes, first provide a concise plan with target vault, affected files, intended changes, backup/history considerations, and command classes to run.
+
+## Read-only operations
+
+Generally safe to run for discovery:
+- `help`, `version`, `vault`, `vaults`
+- `file`, `files`, `folder`, `folders`, `read`
+- `search`, `search:context`
+- `backlinks`, `links`, `unresolved`, `orphans`, `deadends`
+- `outline`, `tags`, `tag`, `tasks`, `properties`, `aliases`
+- `templates`, `template:read`
+- `plugins`, `plugins:enabled`, `themes`, `theme`
+- `wordcount`, `recents`, `tabs`, `bookmarks`
+- `history`, `history:list`, `history:read`, `diff`
+- `sync:status`, `sync:history`, `sync:read`, `sync:deleted`
+- `publish:site`, `publish:list`, `publish:status`
+- `workspaces`, `workspace`, `commands`, `hotkeys`, `hotkey`
+
+## Write operations needing clear user intent
+
+Run only when the user's request clearly asks for the action:
+- `create`, `append`, `prepend`
+- `daily`, `daily:append`, `daily:prepend`
+- `property:set`, `property:remove`
+- `task toggle`, `task done`, `task todo`, `task status`
+- `open`, `search:open`, `template:insert`
+- `workspace:save`, `workspace:load`
+- `bookmark`, `base:create`, `unique`, `web`, `tab:open`
+
+## Destructive or scary operations needing explicit confirmation
+
+Always ask first unless the user has already explicitly confirmed the exact operation:
+- `delete`, especially `delete permanent`
+- `move`, `rename`
+- `history:restore`, `sync:restore`
+- `publish:add changed`, `publish:remove`
+- `plugin:install`, `plugin:uninstall`, `plugin:enable`, `plugin:disable`, `plugin:reload`
+- `theme:install`, `theme:uninstall`, `theme:set`
+- `snippet:enable`, `snippet:disable`
+- `plugins:restrict on/off`
+- `sync on`, `sync off`
+- `reload`, `restart`
+- `eval`, `dev:cdp`, `dev:debug`, `dev:mobile`
+
+Never run arbitrary JavaScript through `eval` unless explicitly requested and the code is understood.
+
+## Common workflows
+
+- Find notes: run `obsidian search query='<text>' format=json`, inspect results, then read or use `search:context` for relevant files.
+- Add a daily note task: run `obsidian daily:append content='- [ ] Task text'` after resolving any vault targeting ambiguity.
+- Mark a task done: use `obsidian task ref='<path:line>' done` when a line reference is available.
+- Create from a template: list or read templates if needed, then run `create path='<path>' template='<template>' open`.
+- Clean unresolved links: run `unresolved verbose format=json`, present a plan and affected files, then ask for confirmation before edits.
+
+## Troubleshooting
+
+If a command fails:
+1. Report the failure clearly.
+2. Run the smallest relevant diagnostic command.
+3. Explain the likely cause and next step.
+4. Do not retry destructive or write commands blindly.
+
+## Tool usage rules
+
+- Use `agent_run_shell_command(command, cwd=None, timeout=60)` for Obsidian CLI commands, tests, and diagnostics.
+- Use `ask_user_question(questions)` for explicit confirmation or structured input.
+- Continue independently for safe read-only discovery.
+- Stop and ask when vault targeting, destructive operations, broad changes, or ambiguous intent require user input.
+"""

code_puppy_core_plugins/obsidian_agent/register_callbacks.py

@@ -0,0 +1,13 @@
+"""Register the Obsidian Agent plugin."""
+
+from code_puppy.callbacks import register_callback
+
+from .agent_obsidian import ObsidianAgent
+
+
+def register_agents() -> list[dict[str, object]]:
+    """Register the Obsidian Agent with Code Puppy's agent catalog."""
+    return [{"name": "obsidian-agent", "class": ObsidianAgent}]
+
+
+register_callback("register_agents", register_agents)

code_puppy_core_plugins/ollama/__init__.py

@@ -0,0 +1 @@
+"""Ollama plugin — registers the 'ollama' model type for local OpenAI Chat Completions-compatible endpoints."""

code_puppy_core_plugins/ollama/register_callbacks.py

@@ -0,0 +1,127 @@
+"""Ollama model type handler for OpenAI Chat Completions-compatible endpoints.
+
+Registers the 'ollama' model type so users can connect Code Puppy to local
+inference servers (Ollama, LM Studio, vLLM, llama.cpp, etc.) via
+~/.code_puppy/extra_models.json.
+
+Minimal config (Ollama on localhost with defaults):
+{
+    "ollama-qwen3": {
+        "type": "ollama",
+        "name": "qwen3:30b",
+        "context_length": 131072
+    }
+}
+
+Full config (remote server, same custom_endpoint format as custom_openai):
+{
+    "lmstudio-codellama": {
+        "type": "ollama",
+        "name": "codellama:34b",
+        "context_length": 16384,
+        "custom_endpoint": {
+            "url": "http://192.168.1.50:1234/v1",
+            "api_key": "$LM_STUDIO_KEY"
+        }
+    }
+}
+
+Note: Code Puppy requires models with strong tool/function calling support.
+Models without tool calling will notwork properly.
+"""
+
+import logging
+import os
+from typing import Any
+
+from pydantic_ai.models.openai import OpenAIChatModel
+from pydantic_ai.providers.openai import OpenAIProvider
+
+from code_puppy.callbacks import register_callback
+from code_puppy.http_utils import create_async_client
+from code_puppy.model_factory import get_custom_config
+
+logger = logging.getLogger(__name__)
+
+# Ollama defaults
+_DEFAULT_OLLAMA_BASE_URL = "http://localhost:11434/v1"
+_DEFAULT_OLLAMA_API_KEY = "ollama"
+
+
+def create_ollama_model(
+    model_name: str,
+    model_config: dict[str, Any],
+    config: dict[str, Any],
+) -> OpenAIChatModel | None:
+    """Create a model for an OpenAI Chat Completions-compatible endpoint.
+
+    When ``custom_endpoint`` is present in *model_config*, the standard
+    ``get_custom_config()`` helper is used (same path as ``custom_openai``
+    and ``codex`` model types).
+
+    When ``custom_endpoint`` is absent, sensible Ollama defaults are applied:
+    - base URL from ``OLLAMA_HOST`` env var, or ``http://localhost:11434/v1``
+    - api_key ``"ollama"`` (required non-empty by OpenAIProvider, not validated by Ollama)
+
+    Args:
+        model_name: The config key name of the model.
+        model_config: The model's configuration dict.
+        config: The full models configuration (unused, kept for API compat).
+
+    Returns:
+        OpenAIChatModel instance, or None if creation fails.
+    """
+    try:
+        if "custom_endpoint" in model_config:
+            url, headers, verify, api_key, _timeout = get_custom_config(model_config)
+        else:
+            # Derive base URL: OLLAMA_HOST env var → default
+            ollama_host = os.environ.get("OLLAMA_HOST", "").rstrip("/")
+            if ollama_host:
+                url = (
+                    ollama_host if ollama_host.endswith("/v1") else f"{ollama_host}/v1"
+                )
+            else:
+                url = _DEFAULT_OLLAMA_BASE_URL
+            headers = {}
+            verify = None
+            api_key = _DEFAULT_OLLAMA_API_KEY
+
+        client = create_async_client(headers=headers, verify=verify)
+
+        provider_args: dict[str, Any] = {
+            "base_url": url,
+            "http_client": client,
+        }
+        if api_key:
+            provider_args["api_key"] = api_key
+        else:
+            provider_args["api_key"] = _DEFAULT_OLLAMA_API_KEY
+
+        provider = OpenAIProvider(**provider_args)
+
+        actual_model_name = model_config.get("name", model_name)
+        model = OpenAIChatModel(actual_model_name, provider=provider)
+        model.provider = (
+            provider  # Expose for connection-pooling cleanup (project convention)
+        )
+
+        logger.info("Created ollama model: %s -> %s", actual_model_name, url)
+        return model
+
+    except Exception as e:
+        logger.error("Failed to create ollama model '%s': %s", model_name, e)
+        return None
+
+
+def _get_ollama_model_types():
+    """Return the ollama model type handler for the register_model_type hook."""
+    return [
+        {
+            "type": "ollama",
+            "handler": create_ollama_model,
+        },
+    ]
+
+
+register_callback("register_model_type", _get_ollama_model_types)

code_puppy_core_plugins/ollama_setup/__init__.py

@@ -0,0 +1,5 @@
+"""Ollama cloud model setup plugin.
+
+Provides `/ollama-setup <model>` to pull an Ollama cloud model and register
+it in ~/.code_puppy/extra_models.json so it's immediately usable.
+"""

code_puppy_core_plugins/ollama_setup/completer.py

@@ -0,0 +1,38 @@
+"""Tab-completion for ``/ollama-setup <model>``."""
+
+from __future__ import annotations
+
+from typing import Iterable
+
+from prompt_toolkit.completion import Completer, Completion
+from prompt_toolkit.document import Document
+
+from code_puppy_core_plugins.ollama_setup.register_callbacks import CLOUD_MODELS
+
+
+class OllamaSetupCompleter(Completer):
+    """Completes cloud model names after ``/ollama-setup ``."""
+
+    trigger = "/ollama-setup"
+
+    def get_completions(
+        self, document: Document, complete_event: object
+    ) -> Iterable[Completion]:
+        text = document.text_before_cursor.lstrip()
+
+        if not text.startswith(self.trigger + " "):
+            return
+
+        # Everything after "/ollama-setup "
+        after = text[len(self.trigger) + 1 :]
+        partial = after.strip().lower()
+        start_pos = -len(after) if after else 0
+
+        for tag, meta in CLOUD_MODELS.items():
+            if not partial or tag.lower().startswith(partial):
+                yield Completion(
+                    tag,
+                    start_position=start_pos,
+                    display=tag,
+                    display_meta=meta["description"],
+                )