Updated huggingface-hub version + Added simple test examples

TEST-EXAMPLES.md

@@ -0,0 +1,257 @@
+# 🧪 Simple Test Examples for AJ STUDIOZ API
+
+## ✅ Super Simple Tests (No Auth Required)
+
+### 1. Basic Chat Test
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/chat \
+  -H "Content-Type: application/json" \
+  -d "{\"message\": \"Hello AJ!\"}"
+```
+
+### 2. Code Generation Test
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/chat \
+  -H "Content-Type: application/json" \
+  -d "{\"message\": \"Write a Python function to reverse a string\"}"
+```
+
+### 3. Direct Generation
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/api/generate \
+  -H "Content-Type: application/json" \
+  -d "{\"prompt\": \"Explain what is an API\", \"max_tokens\": 500}"
+```
+
+### 4. Health Check
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/health
+```
+
+### 5. Service Info
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/
+```
+
+## 🔐 With Authentication
+
+### OpenAI Format
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/v1/chat/completions \
+  -H "Content-Type: application/json" \
+  -H "Authorization: Bearer aj_test123456" \
+  -d "{
+    \"model\": \"aj-mini\",
+    \"messages\": [{\"role\": \"user\", \"content\": \"Hello\"}]
+  }"
+```
+
+### Anthropic Claude Format
+```bash
+curl https://kamesh14151-aj-studioz-api.hf.space/v1/messages \
+  -H "x-api-key: sk-ant-test123456" \
+  -H "anthropic-version: 2023-06-01" \
+  -H "content-type: application/json" \
+  -d "{
+    \"model\": \"claude-sonnet-4-20250514\",
+    \"max_tokens\": 1024,
+    \"messages\": [{\"role\": \"user\", \"content\": \"Hello world\"}]
+  }"
+```
+
+## 💻 PowerShell Examples (Windows)
+
+### Simple Chat
+```powershell
+curl.exe https://kamesh14151-aj-studioz-api.hf.space/chat `
+  -H "Content-Type: application/json" `
+  -d '{\"message\": \"Hello AJ!\"}'
+```
+
+### Code Generation
+```powershell
+$body = @{
+    message = "Write a JavaScript function to check if a number is prime"
+} | ConvertTo-Json
+
+Invoke-RestMethod -Uri "https://kamesh14151-aj-studioz-api.hf.space/chat" `
+  -Method Post `
+  -ContentType "application/json" `
+  -Body $body
+```
+
+## 🐍 Python Examples
+
+### Simple Chat
+```python
+import requests
+
+response = requests.post(
+    "https://kamesh14151-aj-studioz-api.hf.space/chat",
+    json={"message": "Write a function to calculate factorial"}
+)
+
+print(response.json()["reply"])
+```
+
+### With OpenAI SDK
+```python
+from openai import OpenAI
+
+client = OpenAI(
+    base_url="https://kamesh14151-aj-studioz-api.hf.space/v1",
+    api_key="aj_test123456"
+)
+
+response = client.chat.completions.create(
+    model="aj-mini",
+    messages=[
+        {"role": "user", "content": "Explain Python decorators"}
+    ]
+)
+
+print(response.choices[0].message.content)
+```
+
+### With Anthropic SDK
+```python
+import anthropic
+
+client = anthropic.Anthropic(
+    api_key="sk-ant-test123",
+    base_url="https://kamesh14151-aj-studioz-api.hf.space"
+)
+
+message = client.messages.create(
+    model="claude-sonnet-4-20250514",
+    max_tokens=1024,
+    messages=[
+        {"role": "user", "content": "Explain async/await"}
+    ]
+)
+
+print(message.content[0].text)
+```
+
+## 🌐 JavaScript/Node.js Examples
+
+### Fetch API
+```javascript
+const response = await fetch('https://kamesh14151-aj-studioz-api.hf.space/chat', {
+  method: 'POST',
+  headers: { 'Content-Type': 'application/json' },
+  body: JSON.stringify({ 
+    message: 'Write a sorting algorithm in JavaScript' 
+  })
+});
+
+const data = await response.json();
+console.log(data.reply);
+```
+
+### Axios
+```javascript
+const axios = require('axios');
+
+const response = await axios.post(
+  'https://kamesh14151-aj-studioz-api.hf.space/chat',
+  { message: 'Explain promises in JavaScript' }
+);
+
+console.log(response.data.reply);
+```
+
+## 📊 Expected Response Formats
+
+### Chat Endpoint Response
+```json
+{
+  "reply": "Hello! I'm AJ, your AI assistant...",
+  "model": "AJ-Mini v1.0",
+  "provider": "AJ STUDIOZ"
+}
+```
+
+### Health Check Response
+```json
+{
+  "status": "healthy",
+  "model": "aj-mini",
+  "provider": "huggingface"
+}
+```
+
+### OpenAI Format Response
+```json
+{
+  "id": "chatcmpl-abc123...",
+  "object": "chat.completion",
+  "created": 1704067200,
+  "model": "aj-mini",
+  "choices": [
+    {
+      "index": 0,
+      "message": {
+        "role": "assistant",
+        "content": "Response text here..."
+      },
+      "finish_reason": "stop"
+    }
+  ],
+  "usage": {
+    "prompt_tokens": 10,
+    "completion_tokens": 50,
+    "total_tokens": 60
+  }
+}
+```
+
+### Anthropic Format Response
+```json
+{
+  "id": "msg_abc123...",
+  "type": "message",
+  "role": "assistant",
+  "content": [
+    {
+      "type": "text",
+      "text": "Response text here..."
+    }
+  ],
+  "model": "claude-sonnet-4-20250514",
+  "stop_reason": "end_turn",
+  "usage": {
+    "input_tokens": 10,
+    "output_tokens": 50
+  }
+}
+```
+
+---
+
+## 🎯 Quick Start
+
+1. **Test if it's working:**
+   ```bash
+   curl https://kamesh14151-aj-studioz-api.hf.space/health
+   ```
+
+2. **Send your first message:**
+   ```bash
+   curl https://kamesh14151-aj-studioz-api.hf.space/chat \
+     -H "Content-Type: application/json" \
+     -d '{"message": "Hello!"}'
+   ```
+
+3. **Ask for code:**
+   ```bash
+   curl https://kamesh14151-aj-studioz-api.hf.space/chat \
+     -H "Content-Type: application/json" \
+     -d '{"message": "Write a Python web scraper"}'
+   ```
+
+---
+
+**Your API is ready!** 🚀  
+**URL**: https://kamesh14151-aj-studioz-api.hf.space  
+**Status**: FREE FOREVER • UNLIMITED USAGE • NO RATE LIMITS

app.py

@@ -151,8 +151,9 @@ async def anthropic_messages(
                 }
             )
         
-        # Convert to HF format and call API
-        hf_messages = []
+        # Convert to prompt format for text_generation
+        prompt_parts = ["You are AJ, a powerful AI assistant created by AJ STUDIOZ with advanced coding and problem-solving abilities.\n"]
+        
         for msg in messages:
             role = msg.get("role")
             content = msg.get("content")
@@ -160,24 +161,24 @@ async def anthropic_messages(
                 # Handle complex content (text, images, etc.)
                 text_parts = [c.get("text", "") for c in content if c.get("type") == "text"]
                 content = " ".join(text_parts)
-            hf_messages.append({"role": role, "content": content})
+            
+            if role == "user":
+                prompt_parts.append(f"User: {content}")
+            elif role == "assistant":
+                prompt_parts.append(f"Assistant: {content}")
+            elif role == "system":
+                prompt_parts.insert(0, content)
         
-        # Add system message for AJ branding
-        if not any(msg["role"] == "system" for msg in hf_messages):
-            hf_messages.insert(0, {
-                "role": "system",
-                "content": "You are AJ, a powerful AI assistant created by AJ STUDIOZ with advanced coding and problem-solving abilities."
-            })
+        prompt_parts.append("Assistant:")
+        full_prompt = "\n\n".join(prompt_parts)
         
-        response = client.chat_completion(
-            messages=hf_messages,
+        assistant_message = client.text_generation(
+            full_prompt,
             model=MODEL_NAME,
-            max_tokens=max_tokens,
+            max_new_tokens=max_tokens,
             temperature=temperature
         )
         
-        assistant_message = response.choices[0].message.content
-        
         # Return Anthropic-compatible response
         return {
             "id": f"msg_{secrets.token_hex(12)}",
@@ -246,23 +247,20 @@ async def list_models(authorization: Optional[str] = Header(None)):
 async def stream_chat_response(prompt: str, model: str, temperature: float, max_tokens: int, completion_id: str):
     """Generator for streaming responses using Hugging Face Inference API"""
     try:
-        # Format as chat messages
-        messages = [
-            {"role": "system", "content": "You are AJ, a professional AI assistant created by AJ STUDIOZ. You provide thoughtful, accurate, and helpful responses with enterprise-grade reliability."},
-            {"role": "user", "content": prompt}
-        ]
+        # Format as a single prompt for text generation
+        full_prompt = f"""You are AJ, a professional AI assistant created by AJ STUDIOZ with advanced coding and problem-solving abilities.
+
+User: {prompt}
+Assistant:"""
         
-        stream = client.chat_completion(
-            messages=messages,
+        for token in client.text_generation(
+            full_prompt,
             model=MODEL_NAME,
-            max_tokens=max_tokens,
+            max_new_tokens=max_tokens,
             temperature=temperature,
             stream=True
-        )
-        
-        for chunk in stream:
-            if chunk.choices and chunk.choices[0].delta.content:
-                content = chunk.choices[0].delta.content
+        ):
+            if token:
                 # OpenAI-compatible streaming format
                 stream_chunk = {
                     "id": completion_id,
@@ -271,7 +269,7 @@ async def stream_chat_response(prompt: str, model: str, temperature: float, max_
                     "model": model,
                     "choices": [{
                         "index": 0,
-                        "delta": {"content": content},
+                        "delta": {"content": token},
                         "finish_reason": None
                     }]
                 }
@@ -350,20 +348,17 @@ async def chat_completions(request: Request, authorization: Optional[str] = Head
             )
         
         # Non-streaming response using Hugging Face Inference API
-        messages = [
-            {"role": "system", "content": "You are AJ, a professional AI assistant created by AJ STUDIOZ. You provide thoughtful, accurate, and helpful responses with enterprise-grade reliability."},
-            {"role": "user", "content": prompt}
-        ]
+        full_prompt = f"""You are AJ, a professional AI assistant created by AJ STUDIOZ with advanced coding and problem-solving abilities.
+
+{prompt}"""
         
-        response = client.chat_completion(
-            messages=messages,
+        assistant_message = client.text_generation(
+            full_prompt,
             model=MODEL_NAME,
-            max_tokens=max_tokens,
+            max_new_tokens=max_tokens,
             temperature=temperature
         )
         
-        assistant_message = response.choices[0].message.content
-        
         # OpenAI-compatible response
         return {
             "id": completion_id,
@@ -411,20 +406,18 @@ async def completions(request: Request, authorization: Optional[str] = Header(No
             raise HTTPException(status_code=400, detail="Prompt is required")
         
         # Call Hugging Face Inference API
-        messages = [
-            {"role": "system", "content": "You are AJ, a professional AI assistant created by AJ STUDIOZ. You provide thoughtful, accurate, and helpful responses."},
-            {"role": "user", "content": prompt}
-        ]
+        full_prompt = f"""You are AJ, a professional AI assistant created by AJ STUDIOZ with advanced coding abilities.
+
+User: {prompt}
+Assistant:"""
         
-        response = client.chat_completion(
-            messages=messages,
+        completion_text = client.text_generation(
+            full_prompt,
             model=MODEL_NAME,
-            max_tokens=max_tokens,
+            max_new_tokens=max_tokens,
             temperature=temperature
         )
         
-        completion_text = response.choices[0].message.content
-        
         return {
             "id": f"cmpl-{secrets.token_hex(12)}",
             "object": "text_completion",
@@ -460,20 +453,18 @@ async def chat(request: Request):
             return JSONResponse({"error": "Message is required"}, status_code=400)
         
         # Call Hugging Face Inference API
-        messages = [
-            {"role": "system", "content": "You are AJ, a professional AI assistant created by AJ STUDIOZ."},
-            {"role": "user", "content": message}
-        ]
+        full_message = f"""You are AJ, a professional AI assistant created by AJ STUDIOZ with advanced coding abilities.
+
+User: {message}
+Assistant:"""
         
-        response = client.chat_completion(
-            messages=messages,
+        reply = client.text_generation(
+            full_message,
             model=MODEL_NAME,
-            max_tokens=1000,
-            temperature=0.3
+            max_new_tokens=1000,
+            temperature=0.7
         )
         
-        reply = response.choices[0].message.content
-        
         return JSONResponse({
             "reply": reply,
             "model": "AJ-Mini v1.0",
@@ -493,24 +484,18 @@ async def generate(request: Request):
         data = await request.json()
         prompt = data.get("prompt", "")
         max_tokens = data.get("max_tokens", 1000)
-        temperature = data.get("temperature", 0.3)
+        temperature = data.get("temperature", 0.7)
         
         if not prompt:
             return JSONResponse({"error": "Prompt is required"}, status_code=400)
         
-        messages = [
-            {"role": "user", "content": prompt}
-        ]
-        
-        response = client.chat_completion(
-            messages=messages,
+        response_text = client.text_generation(
+            prompt,
             model=MODEL_NAME,
-            max_tokens=max_tokens,
+            max_new_tokens=max_tokens,
             temperature=temperature
         )
         
-        response_text = response.choices[0].message.content
-        
         return JSONResponse({
             "response": response_text,
             "model": "AJ-Mini v1.0",
@@ -545,12 +530,11 @@ async def tags():
 async def health():
     """Health check endpoint"""
     try:
-        # Test HF API with a simple message
-        test_messages = [{"role": "user", "content": "Hi"}]
-        test = client.chat_completion(
-            messages=test_messages,
+        # Test HF API with a simple text generation
+        test = client.text_generation(
+            "Hello",
             model=MODEL_NAME,
-            max_tokens=5
+            max_new_tokens=5
         )
         return {"status": "healthy", "model": "aj-mini", "provider": "huggingface"}
     except Exception as e:

requirements.txt

@@ -1,5 +1,5 @@
 fastapi==0.104.1
 uvicorn[standard]==0.24.0
-huggingface-hub==0.19.4
+huggingface-hub>=0.20.0
 requests==2.31.0
 python-multipart==0.0.6