clarity-ai 6.5.1 → 6.5.2

package/CHANGELOG.md

@@ -2,6 +2,21 @@
 
 ---
 
+## 6.5.2 (2026-06-06)
+
+### CoT + MoE Space Architecture
+- Dataset repos now hold LoRA weights (flash pushed to `Universal-618/Clarity-flash-weights` dataset)
+- Space app downloads weights from dataset repo on startup, loads base model + LoRA in 4-bit, serves `/v1/chat/completions`
+- CLI routes by model: Flash 14B → Clarity-2 (CoT), Heavy 20B → Clarity-3 (MoE)
+- Multiple fallback Spaces for each model
+- All 6 Spaces connected via shared bucket to Clarity-main
+
+## 6.5.1 (2026-06-06)
+
+### 6-Space Fallback
+- All 6 Spaces tried in sequence for redundancy
+- Lowercase HF Space URL fix
+
 ## 6.5.0 (2026-06-06)
 
 ### Closed Source + Space Proxy

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "clarity-ai",
-  "version": "6.5.1",
+  "version": "6.5.2",
   "description": "CLARITY — terminal AI agent with Clarity Flash 14B",
   "type": "module",
   "bin": {

package/requirements.txt

@@ -1,3 +1,9 @@
 fastapi>=0.100
-httpx>=0.27
 uvicorn>=0.20
+transformers>=4.38
+torch>=2.1
+accelerate>=0.27
+bitsandbytes>=0.42
+peft>=0.8
+huggingface-hub>=0.20
+sentencepiece>=0.1.99

package/space_app.py

@@ -1,31 +1,69 @@
-import os, json, asyncio
-import httpx
+import os, json, time, asyncio
+from threading import Thread
 from fastapi import FastAPI, Request
 from fastapi.responses import StreamingResponse, JSONResponse
 from contextlib import asynccontextmanager
 
 HF_TOKEN = os.environ.get('HF_TOKEN', '')
-MODEL = os.environ.get('HF_MODEL', 'Universal-618/Clarity-flash-weights')
-API = f'https://api-inference.huggingface.co/models/{MODEL}/v1/chat/completions'
-HEADERS = {'Authorization': f'Bearer {HF_TOKEN}', 'Content-Type': 'application/json'}
+MODEL_NAME = os.environ.get('HF_MODEL', 'Clarity-flash-weights')
+BASE_MODEL = os.environ.get('BASE_MODEL_ID', '')
+DS = f'Universal-618/{MODEL_NAME}'
+
+print(f'[startup] model={MODEL_NAME} base={BASE_MODEL}', flush=True)
+
+model = None
+tokenizer = None
+
+def load_model():
+    global model, tokenizer
+    print('[startup] downloading weights...', flush=True)
+    from huggingface_hub import snapshot_download
+    path = snapshot_download(repo_id=DS, repo_type='dataset', token=HF_TOKEN)
+    print(f'[startup] weights at {path}', flush=True)
+    from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig, TextIteratorStreamer
+    from peft import PeftModel
+    bnb = BitsAndBytesConfig(load_in_4bit=True, bnb_4bit_compute_dtype='float16', bnb_4bit_use_double_quant=True)
+    print('[startup] loading base model...', flush=True)
+    base = AutoModelForCausalLM.from_pretrained(
+        BASE_MODEL, quantization_config=bnb, device_map='auto',
+        trust_remote_code=True, low_cpu_mem_usage=True, token=HF_TOKEN)
+    print('[startup] loading LoRA...', flush=True)
+    model = PeftModel.from_pretrained(base, path, token=HF_TOKEN)
+    tokenizer = AutoTokenizer.from_pretrained(path, token=HF_TOKEN)
+    if tokenizer.pad_token is None:
+        tokenizer.pad_token = tokenizer.eos_token
+    print('[startup] ready', flush=True)
+
+load_model()
 
 app = FastAPI()
 
 @app.post('/v1/chat/completions')
-async def proxy(request: Request):
+async def generate(request: Request):
     body = await request.json()
-    body['model'] = MODEL
+    prompt = tokenizer.apply_chat_template(body['messages'], tokenize=False, add_generation_prompt=True)
+    inputs = tokenizer(prompt, return_tensors='pt')
     stream = body.get('stream', True)
-    body['stream'] = stream
-    async with httpx.AsyncClient(timeout=120) as client:
-        resp = await client.post(API, headers=HEADERS, json=body)
-        if resp.status_code != 200:
-            return JSONResponse({'error': await resp.aread()}, status_code=resp.status_code)
-        if stream:
-            return StreamingResponse(resp.aiter_lines(), media_type='text/event-stream', headers={
-                'Cache-Control': 'no-cache', 'Connection': 'keep-alive', 'X-Accel-Buffering': 'no'
-            })
-        return resp.json()
+    max_tokens = body.get('max_tokens', 4096)
+    gen_kwargs = dict(input_ids=inputs.input_ids, attention_mask=inputs.attention_mask,
+                      max_new_tokens=max_tokens, temperature=body.get('temperature', 0.7),
+                      do_sample=True, pad_token_id=tokenizer.pad_token_id)
+    if stream:
+        from transformers import TextIteratorStreamer
+        streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
+        t = Thread(target=model.generate, kwargs={**gen_kwargs, 'streamer': streamer})
+        t.start()
+        async def gen():
+            yield 'data: {"choices":[{"delta":{"role":"assistant"},"index":0}]}\n\n'
+            for text in streamer:
+                if text:
+                    yield f'data: {json.dumps({"choices":[{"delta":{"content":text},"index":0}]})}\n\n'
+            yield 'data: [DONE]\n\n'
+        return StreamingResponse(gen(), media_type='text/event-stream', headers={
+            'Cache-Control': 'no-cache', 'Connection': 'keep-alive', 'X-Accel-Buffering': 'no'})
+    out = model.generate(**gen_kwargs)
+    text = tokenizer.decode(out[0][inputs.input_ids.shape[1]:], skip_special_tokens=True)
+    return {'choices': [{'message': {'role': 'assistant', 'content': text}, 'index': 0}]}
 
 @app.get('/main-data/{path:path}')
 async def main_data(path: str):
@@ -38,7 +76,7 @@ async def main_data(path: str):
 
 @app.get('/')
 async def root():
-    return {'status': 'ok', 'model': MODEL, 'docs': '/v1/chat/completions'}
+    return {'status': 'ok', 'model': MODEL_NAME}
 
 if __name__ == '__main__':
     import uvicorn

package/src/config/models.js

@@ -1,5 +1,6 @@
 export const ALL_MODELS = [
   { id: 'huggingface/Universal-618/Clarity-flash-weights', provider: 'huggingface', label: 'Clarity Flash 14B', badge: '128K' },
+  { id: 'huggingface/Universal-618/Clarity-heavy-weights', provider: 'huggingface', label: 'Clarity Heavy 20B', badge: 'MoE' },
   { id: 'groq/llama-3.3-70b-versatile', provider: 'groq', label: 'Llama 3.3 70B Versatile', badge: null },
   { id: 'groq/llama-3.1-8b-instant', provider: 'groq', label: 'Llama 3.1 8B Instant', badge: 'Fast' },
   { id: 'groq/llama-4-scout-17b-16e-instruct', provider: 'groq', label: 'Llama 4 Scout 17B', badge: null },

package/src/providers/index.js

@@ -1,35 +1,25 @@
 import { getKey } from '../config/keys.js';
 import { streamResponse } from './streaming.js';
-import { parseErrorResponse } from './errors.js';
 
-const SPACES = [
-  'https://universal-618-clarity-main.hf.space',
-  'https://universal-618-clarity-2.hf.space',
-  'https://universal-618-clarity-3.hf.space',
-  'https://universal-618-clarity-4.hf.space',
-  'https://universal-618-clarity-5.hf.space',
-  'https://universal-618-clarity-6.hf.space',
-];
+const MODEL_ROUTES = {
+  'Universal-618/Clarity-flash-weights': [
+    'https://universal-618-clarity-2.hf.space/v1/chat/completions',
+    'https://universal-618-clarity-4.hf.space/v1/chat/completions',
+    'https://universal-618-clarity-5.hf.space/v1/chat/completions',
+  ],
+  'Universal-618/Clarity-heavy-weights': [
+    'https://universal-618-clarity-3.hf.space/v1/chat/completions',
+    'https://universal-618-clarity-5.hf.space/v1/chat/completions',
+    'https://universal-618-clarity-6.hf.space/v1/chat/completions',
+  ],
+};
 
-const PROVIDERS = {
-  huggingface: {
-    endpoints: SPACES.map(s => s + '/v1/chat/completions'),
-    name: 'huggingface',
-  },
-  groq: {
-    endpoints: ['https://api.groq.com/openai/v1/chat/completions'],
-    name: 'groq',
-  },
-  openrouter: {
-    endpoints: ['https://openrouter.ai/api/v1/chat/completions'],
-    name: 'openrouter',
-  },
+const STATIC_ENDPOINTS = {
+  groq: ['https://api.groq.com/openai/v1/chat/completions'],
+  openrouter: ['https://openrouter.ai/api/v1/chat/completions'],
 };
 
 export async function* callAI(providerName, model, messages, options = {}) {
-  const provider = PROVIDERS[providerName];
-  if (!provider) throw { type: 'config_error', message: 'Unknown provider: ' + providerName };
-
   const key = getKey(providerName);
   if (!key && providerName !== 'huggingface') {
     throw { type: 'auth_error', provider: providerName, message: 'No API key set for ' + providerName, hint: '/keys ' + providerName + ' <your-key>' };
@@ -55,8 +45,13 @@ export async function* callAI(providerName, model, messages, options = {}) {
     extraHeaders['X-Title'] = 'CLARITY AI';
   }
 
+  const endpoints = MODEL_ROUTES[modelName] || STATIC_ENDPOINTS[providerName] || [];
+  if (!endpoints.length) {
+    throw { type: 'config_error', message: 'No endpoints for ' + providerName + '/' + modelName };
+  }
+
   let lastErr;
-  for (const endpoint of provider.endpoints) {
+  for (const endpoint of endpoints) {
     try {
       const stream = streamResponse(endpoint, body, key || 'none', extraHeaders, options.signal);
       for await (const event of stream) {