tzamuncode 0.1.2__tar.gz → 0.1.4__tar.gz

{tzamuncode-0.1.2/tzamuncode.egg-info → tzamuncode-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tzamuncode
-Version: 0.1.2
+Version: 0.1.4
 Summary: TzamunCode - AI Coding Assistant powered by local models
 Author-email: "Tzamun Arabia IT Co." <info@tzamun.com>
 License: MIT

{tzamuncode-0.1.2 → tzamuncode-0.1.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tzamuncode"
-version = "0.1.2"
+version = "0.1.4"
 description = "TzamunCode - AI Coding Assistant powered by local models"
 authors = [
     {name = "Tzamun Arabia IT Co.", email = "info@tzamun.com"}

{tzamuncode-0.1.2 → tzamuncode-0.1.4}/tzamuncode/cli/main.py

@@ -78,6 +78,102 @@ def main(
         
         chat.run()
 
+@app.command()
+def agent(
+    backend: str = typer.Option(
+        "vllm",
+        "--backend",
+        "-b",
+        help="AI backend to use (vllm or ollama)"
+    ),
+    model: str = typer.Option(
+        "qwen2.5-7b-instruct",
+        "--model",
+        "-m",
+        help="Model name to use"
+    ),
+    apply: bool = typer.Option(
+        False,
+        "--apply",
+        help="Auto-apply changes without confirmation"
+    )
+):
+    """
+    Agentic execution mode for Paperclip integration.
+    
+    Reads prompt from stdin, calls AI backend, outputs response to stdout.
+    
+    Examples:
+        echo "Write a Python function" | tzamuncode agent --backend vllm --model qwen2.5-7b-instruct
+        echo "Fix the bug" | tzamuncode agent --apply --backend ollama --model qwen2.5:32b
+    """
+    import requests
+    import os
+    
+    # Read prompt from stdin
+    prompt = sys.stdin.read().strip()
+    
+    if not prompt:
+        sys.stderr.write("Error: No prompt provided via stdin\n")
+        raise typer.Exit(code=1)
+    
+    # Get backend URL from environment or use defaults
+    if backend == "vllm":
+        base_url = os.getenv('VLLM_BASE_URL', 'http://localhost:8000')
+        # Call vLLM (OpenAI-compatible API)
+        try:
+            response = requests.post(
+                f"{base_url}/v1/chat/completions",
+                json={
+                    "model": model,
+                    "messages": [{"role": "user", "content": prompt}],
+                    "temperature": 0.7,
+                    "max_tokens": 2000,
+                    "stream": False
+                },
+                timeout=300
+            )
+            response.raise_for_status()
+            result = response.json()
+            ai_response = result["choices"][0]["message"]["content"]
+        except Exception as e:
+            sys.stderr.write(f"Error calling vLLM: {e}\n")
+            raise typer.Exit(code=1)
+    
+    elif backend == "ollama":
+        base_url = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
+        # Call Ollama API
+        try:
+            response = requests.post(
+                f"{base_url}/api/generate",
+                json={
+                    "model": model,
+                    "prompt": prompt,
+                    "stream": False,
+                    "options": {
+                        "temperature": 0.7,
+                        "num_predict": 2000
+                    }
+                },
+                timeout=300
+            )
+            response.raise_for_status()
+            result = response.json()
+            ai_response = result["response"]
+        except Exception as e:
+            sys.stderr.write(f"Error calling Ollama: {e}\n")
+            raise typer.Exit(code=1)
+    
+    else:
+        sys.stderr.write(f"Error: Unknown backend '{backend}'. Use 'vllm' or 'ollama'\n")
+        raise typer.Exit(code=1)
+    
+    # Output response to stdout (Paperclip captures this)
+    print(ai_response, flush=True)
+    
+    # Exit successfully
+    raise typer.Exit(code=0)
+
 @app.command()
 def chat(
     model: str = typer.Option(

{tzamuncode-0.1.2 → tzamuncode-0.1.4}/tzamuncode/models/ollama.py

@@ -3,6 +3,7 @@ Ollama client for TzamunCode
 """
 
 import requests
+import os
 from typing import List, Dict, Generator, Optional
 import json
 
@@ -11,10 +12,13 @@ class OllamaClient:
     
     def __init__(
         self,
-        base_url: str = "http://localhost:11434",
+        base_url: str = None,
         model: str = "qwen2.5:32b",
         timeout: int = 120
     ):
+        # Use environment variable if available, otherwise default to localhost
+        if base_url is None:
+            base_url = os.getenv('OLLAMA_BASE_URL', 'http://localhost:11434')
         self.base_url = base_url.rstrip('/')
         self.model = model
         self.timeout = timeout

{tzamuncode-0.1.2 → tzamuncode-0.1.4}/tzamuncode/models/vllm_client.py

@@ -4,13 +4,17 @@ Provides faster inference using vLLM OpenAI-compatible API
 """
 
 import requests
+import os
 from typing import Iterator, List, Dict, Optional
 
 
 class VLLMClient:
     """Client for vLLM server using OpenAI-compatible API"""
     
-    def __init__(self, base_url: str = "http://localhost:8000", model: str = "deepseek-coder-7b"):
+    def __init__(self, base_url: str = None, model: str = "deepseek-coder-7b"):
+        # Use environment variable if available, otherwise default to localhost
+        if base_url is None:
+            base_url = os.getenv('VLLM_BASE_URL', 'http://localhost:8000')
         self.base_url = base_url.rstrip('/')
         self.model = model
         self.api_url = f"{self.base_url}/v1"

{tzamuncode-0.1.2 → tzamuncode-0.1.4/tzamuncode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tzamuncode
-Version: 0.1.2
+Version: 0.1.4
 Summary: TzamunCode - AI Coding Assistant powered by local models
 Author-email: "Tzamun Arabia IT Co." <info@tzamun.com>
 License: MIT