sky-dev 0.1.8__tar.gz → 0.1.9__tar.gz

{sky_dev-0.1.8 → sky_dev-0.1.9}/CHANGELOG.md

@@ -1,5 +1,21 @@
 # Changelog
 
+## [0.1.9] - 2026-09-28
+
+### Added
+- Dynamic model discovery for guardrails (auto-selects available NIM models)
+- Chat mode now has repository context awareness
+- `_get_repo_context()` for automatic project understanding
+
+### Fixed
+- EOL model issue (`meta/llama-3.1-70b-instruct` was deprecated)
+- Chat mode no longer asks "please share your codebase"
+- Version mismatch between pip and CLI
+
+### Changed
+- Guardrails now auto-discover working models from NVIDIA NIM
+- Chat injects current directory, files, and README snippet
+
 ## [0.1.8] - 2026-09-28
 
 ### Added

{sky_dev-0.1.8 → sky_dev-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sky-dev
-Version: 0.1.8
+Version: 0.1.9
 Summary: Sky - Agentic coding assistant. Build without boundaries.
 Author-email: Aaditya A <aaditya@corover.ai>
 License: MIT

{sky_dev-0.1.8 → sky_dev-0.1.9}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sky-dev"
-version = "0.1.8"
+version = "0.1.9"
 description = "Sky - Agentic coding assistant. Build without boundaries."
 readme = "README.md"
 license = {text = "MIT"}

{sky_dev-0.1.8 → sky_dev-0.1.9}/sky/__init__.py

@@ -1,6 +1,6 @@
 """Sky - Build without boundaries."""
 
-__version__ = "0.1.8"
+__version__ = "0.1.9"
 __author__ = "Aaditya A"
 __author_email__ = "aaditya@corover.ai"
 __description__ = "Local, CLI-based, agentic software development assistant"

{sky_dev-0.1.8 → sky_dev-0.1.9}/sky/core/chat.py

@@ -126,11 +126,57 @@ class ChatEngine:
         asyncio.run(self._get_chat_response(user_input))
 
 
+    def _get_repo_context(self) -> str:
+        """Get brief context about the current repository."""
+        try:
+            from pathlib import Path
+            import logging
+            cwd = Path.cwd()
+            
+            # Get top-level files and directories
+            items = sorted([item.name for item in cwd.iterdir() 
+                           if not item.name.startswith('.')])[:10]
+            
+            # Check for common project files
+            markers = []
+            for marker in ["README.md", "pyproject.toml", "package.json", 
+                           "requirements.txt", "Cargo.toml", "go.mod"]:
+                if (cwd / marker).exists():
+                    markers.append(marker)
+            
+            # Read README first 500 chars if available
+            readme_snippet = ""
+            readme = cwd / "README.md"
+            if readme.exists():
+                readme_snippet = readme.read_text(encoding="utf-8")[:500]
+            
+            context = f"""Current working directory: {cwd.name}
+Top-level items: {', '.join(items)}
+Project markers: {', '.join(markers) if markers else 'none'}
+
+README excerpt:
+{readme_snippet}
+"""
+            return context
+        except Exception as e:
+            import logging
+            logging.getLogger(__name__).error(f"Failed to get repo context: {e}")
+            return ""
+
     async def _get_chat_response(self, user_input: str):
         """Get response from chat model."""
         if not any(m.get("role") == "system" for m in self.messages):
             from sky.core.mode_prompts import get_mode_prompt
-            self.messages.insert(0, {"role": "system", "content": get_mode_prompt("chat")})
+            repo_context = self._get_repo_context()
+            chat_system_prompt = get_mode_prompt("chat")
+            system_prompt = f"""{chat_system_prompt}
+
+## Current Repository Context
+{repo_context}
+
+Use this context to answer questions about the repository.
+"""
+            self.messages.insert(0, {"role": "system", "content": system_prompt})
             
         self.messages.append({"role": "user", "content": user_input})
         

{sky_dev-0.1.8 → sky_dev-0.1.9}/sky/security/guardrails_nvidia.py

@@ -6,12 +6,49 @@ import asyncio
 from typing import Optional, Dict, Any
 from openai import AsyncOpenAI
 
+import httpx
+
 logger = logging.getLogger(__name__)
 
-# Updated models (v0.1.7+)
-# Note: Verify availability at https://build.nvidia.com
-NVIDIA_JAILBREAK_MODEL = "meta/llama-3.1-70b-instruct"
-NVIDIA_TOPIC_MODEL = "meta/llama-3.1-70b-instruct"
+# Preferred models (in order of preference)
+PREFERRED_MODELS = [
+    "nvidia/llama-3.1-nemotron-70b-instruct",  # Reliable
+    "nvidia/nemotron-3-super-120b-a12b",        # Newest
+    "meta/llama-3.3-70b-instruct",              # If available
+    "mistralai/mistral-large-2-instruct",       # Alternative
+]
+
+def discover_guardrail_model() -> Optional[str]:
+    """Discover an available model from NVIDIA NIM."""
+    api_key = os.getenv("NVIDIA_NIM_API_KEY")
+    if not api_key:
+        return None
+    
+    try:
+        res = httpx.get(
+            "https://integrate.api.nvidia.com/v1/models",
+            headers={"Authorization": f"Bearer {api_key}"},
+            timeout=10
+        )
+        available = {m["id"] for m in res.json().get("data", [])}
+        
+        # Pick the first preferred model that's available
+        for model in PREFERRED_MODELS:
+            if model in available:
+                logger.info(f"Using guardrail model: {model}")
+                return model
+        
+        # Fallback: any Llama or Nemotron model
+        for model_id in available:
+            if "llama" in model_id.lower() or "nemotron" in model_id.lower():
+                logger.info(f"Fallback guardrail model: {model_id}")
+                return model_id
+        
+        logger.warning("No suitable guardrail model found")
+        return None
+    except Exception as e:
+        logger.error(f"Model discovery failed: {e}")
+        return None
 
 
 class NVIDIAGuardrails:
@@ -26,18 +63,19 @@ class NVIDIAGuardrails:
         return cls._instance
     
     def __init__(self):
-        if self._initialized:
+        if getattr(self, "_initialized", False):
             return
         
         self.client = AsyncOpenAI(
             base_url="https://integrate.api.nvidia.com/v1",
             api_key=os.getenv("NVIDIA_NIM_API_KEY")
         )
+        self.model = discover_guardrail_model()
         self._initialized = True
     
     def is_available(self) -> bool:
         """Check if NVIDIA guardrails are available."""
-        return bool(os.getenv("NVIDIA_NIM_API_KEY"))
+        return bool(self.model)
     
     async def check_input_async(self, user_input: str) -> Dict[str, Any]:
         """Async check for jailbreak attempts."""
@@ -46,7 +84,7 @@ class NVIDIAGuardrails:
         
         try:
             response = await self.client.chat.completions.create(
-                model=NVIDIA_JAILBREAK_MODEL,
+                model=self.model,
                 messages=[
                     {
                         "role": "system",
@@ -83,7 +121,7 @@ class NVIDIAGuardrails:
         
         try:
             response = await self.client.chat.completions.create(
-                model=NVIDIA_TOPIC_MODEL,
+                model=self.model,
                 messages=[
                     {
                         "role": "system",
@@ -119,7 +157,7 @@ class NVIDIAGuardrails:
         
         try:
             result = await self.client.chat.completions.create(
-                model=NVIDIA_JAILBREAK_MODEL,
+                model=self.model,
                 messages=[
                     {
                         "role": "system",

{sky_dev-0.1.8 → sky_dev-0.1.9}/sky_dev.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sky-dev
-Version: 0.1.8
+Version: 0.1.9
 Summary: Sky - Agentic coding assistant. Build without boundaries.
 Author-email: Aaditya A <aaditya@corover.ai>
 License: MIT