Auto-deploy from GitHub: 141b37810ab063d3984b529a3bda3b8633f874c3

Dockerfile

@@ -14,11 +14,16 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     zstd \
     procps \
+    nodejs \
+    npm \
     && rm -rf /var/lib/apt/lists/*
 
 # Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
+# Install opencode CLI
+RUN npm install -g opencode-ai
+
 # Copy project files
 COPY pyproject.toml .
 COPY . .

app/api/routes.py

@@ -22,8 +22,9 @@ async def submit_task(request: Request):
     input_text = data['text'].strip()
     system_prompt = data.get('system_prompt', '').strip() or None
     hide_from_ui = 1 if data.get('hide_from_ui') else 0
+    model = (data.get('model') or 'qwen').strip().lower()
 
-    await crud.insert_task(task_id, input_text, system_prompt, 'not_started', hide_from_ui)
+    await crud.insert_task(task_id, input_text, system_prompt, 'not_started', hide_from_ui, model)
     
     await start_worker()
 
@@ -31,6 +32,7 @@ async def submit_task(request: Request):
         'id': task_id,
         'filename': input_text[:50] + ("..." if len(input_text) > 50 else ""),
         'status': 'not_started',
+        'model': model,
         'message': 'Task submitted successfully'
     })
 

app/db/crud.py

@@ -3,14 +3,14 @@ from datetime import datetime, timedelta
 from app.core.config import settings
 from custom_logger import logger_config as logger
 
-async def insert_task(task_id: str, input_text: str, system_prompt: str, status: str, hide_from_ui: int):
+async def insert_task(task_id: str, input_text: str, system_prompt: str, status: str, hide_from_ui: int, model: str = 'qwen'):
     async with aiosqlite.connect(settings.DATABASE_FILE) as db:
         await db.execute('''INSERT INTO text_tasks 
-                     (id, input_text, system_prompt, status, created_at, hide_from_ui)
-                     VALUES (?, ?, ?, ?, ?, ?)''',
-                  (task_id, input_text, system_prompt, status, datetime.now().isoformat(), hide_from_ui))
+                     (id, input_text, system_prompt, model, status, created_at, hide_from_ui)
+                     VALUES (?, ?, ?, ?, ?, ?, ?)''',
+                  (task_id, input_text, system_prompt, model, status, datetime.now().isoformat(), hide_from_ui))
         await db.commit()
-    logger.debug(f"Inserted task (ID: {task_id}) into database.")
+    logger.debug(f"Inserted task (ID: {task_id}, model: {model}) into database.")
 
 async def update_status(task_id: str, status: str, result: str = None, error: str = None):
     async with aiosqlite.connect(settings.DATABASE_FILE) as db:

app/db/database.py

@@ -9,6 +9,7 @@ async def init_db():
                      (id TEXT PRIMARY KEY,
                       input_text TEXT NOT NULL,
                       system_prompt TEXT,
+                      model TEXT DEFAULT 'qwen',
                       status TEXT NOT NULL,
                       result TEXT,
                       created_at TEXT NOT NULL,

app/services/worker.py

@@ -1,5 +1,7 @@
 import asyncio
 import json
+import subprocess
+import shutil
 from app.core.config import settings
 from custom_logger import logger_config as logger
 from app.db import crud
@@ -33,9 +35,7 @@ async def worker_loop():
         await loop.run_in_executor(None, lambda: initiate({'text': 'Hi', 'model': 'qwen', 'max_new_tokens': 1}))
         logger.info("✅ Qwen model ready. Monitoring for new tasks...")
     except Exception as e:
-        logger.error(f"❌ Failed to load model: {e}")
-        worker_running = False
-        return
+        logger.warning(f"⚠️ Qwen model not available (opencode-only tasks will still work): {e}")
 
     while worker_running:
         logger.debug("Worker loop iteration, checking for files...")
@@ -48,8 +48,9 @@ async def worker_loop():
                 task_id = row['id']
                 input_text = row['input_text']
                 system_prompt = row['system_prompt'] or "You are a helpful assistant."
+                model = row.get('model', 'qwen')
                 
-                logger.info(f"\n{'='*60}\nProcessing task: {task_id}\n📌 Input: {input_text[:100]}...\n{'='*60}")
+                logger.info(f"\n{'='*60}\nProcessing task: {task_id} (model: {model})\n📌 Input: {input_text[:100]}...\n{'='*60}")
                 
                 await crud.update_status(task_id, 'processing')
                 
@@ -63,21 +64,31 @@ async def worker_loop():
 
                 try:
                     await crud.update_progress(task_id, 5, "Starting...")
-                    
-                    result = await loop.run_in_executor(None, lambda: initiate(
-                        {
-                            'text': input_text,
-                            'system_prompt': system_prompt,
-                            'model': 'qwen',
-                        },
-                        progress_callback=progress_cb
-                    ))
 
-                    if result:
-                        logger.success(f"Successfully processed: {task_id}")
-                        await crud.update_status(task_id, 'completed', result=json.dumps(result))
+                    if model == 'opencode':
+                        await crud.update_progress(task_id, 10, "Running opencode...")
+                        result = await loop.run_in_executor(None, lambda: _run_opencode(input_text))
+                        if result:
+                            logger.success(f"Successfully processed (opencode): {task_id}")
+                            await crud.update_progress(task_id, 100, "Completed")
+                            await crud.update_status(task_id, 'completed', result=json.dumps({"response": result}))
+                        else:
+                            raise Exception("opencode returned empty result")
                     else:
-                        raise Exception("initiate() returned empty result")
+                        result = await loop.run_in_executor(None, lambda: initiate(
+                            {
+                                'text': input_text,
+                                'system_prompt': system_prompt,
+                                'model': 'qwen',
+                            },
+                            progress_callback=progress_cb
+                        ))
+
+                        if result:
+                            logger.success(f"Successfully processed: {task_id}")
+                            await crud.update_status(task_id, 'completed', result=json.dumps(result))
+                        else:
+                            raise Exception("initiate() returned empty result")
 
                 except Exception as e:
                     logger.error(f"Failed to process {task_id}: {str(e)}")
@@ -89,3 +100,19 @@ async def worker_loop():
         except Exception as e:
             logger.error(f"Worker error: {str(e)}")
             await asyncio.sleep(settings.POLL_INTERVAL)
+
+
+def _run_opencode(text: str) -> str:
+    if not shutil.which('opencode'):
+        raise FileNotFoundError(
+            "opencode CLI not found. Install it from https://opencode.ai"
+        )
+    result = subprocess.run(
+        ['opencode', 'run', text],
+        capture_output=True,
+        text=True,
+        timeout=300
+    )
+    if result.returncode != 0:
+        raise RuntimeError(f"opencode failed: {result.stderr.strip()}")
+    return result.stdout.strip()

index.html

@@ -783,12 +783,12 @@
             const doc = {
                 base_url: window.location.origin,
                 endpoints: [
-                    { method: "POST", path: "/api/tasks/upload", desc: "Submit text generation task", body: "{ text: string, system_prompt?: string }" },
+                    { method: "POST", path: "/api/tasks/upload", desc: "Submit text generation task", body: "{ text: string, system_prompt?: string, model?: 'qwen' | 'opencode' }" },
                     { method: "GET", path: "/api/tasks", desc: "List all tasks" },
                     { method: "GET", path: "/api/tasks/{task_id}", desc: "Get task details & result" },
                     { method: "GET", path: "/health", desc: "Service health" }
                 ],
-                example_usage: `curl -X POST -H 'Content-Type: application/json' -d '{"text": "Hello"}' ${window.location.origin}/api/tasks/upload`
+                example_usage: `curl -X POST -H 'Content-Type: application/json' -d '{"text": "Hello"}' ${window.location.origin}/api/tasks/upload` + '\n' + `curl -X POST -H 'Content-Type: application/json' -d '{"text": "list files", "model": "opencode"}' ${window.location.origin}/api/tasks/upload`
             };
             UI.resultText.innerText = JSON.stringify(doc, null, 2);
             UI.modalTitle.innerText = "API Documentation";