Added a new ai model provider for grok

app.py

@@ -127,6 +127,31 @@ def convert_to_groq_schema(messages: List[Dict[str, Any]]) -> List[Dict[str, str
     return converted
 
 
+def conver_to_xai_schema(messages: List[Dict[str, Any]]) -> List[Dict[str, str]]:
+    converted = []
+    for message in messages:
+        role = message.get("role", "user")
+        content = message.get("content", "")
+
+        if isinstance(content, list):
+            # Handle content that's already in parts format
+            parts = content
+            text_content = "\n".join([p.get("text", "") for p in content if p.get("type") == "text"])
+        else:
+            # Create parts format for text content
+            text_content = str(content)
+            parts = [{"type": "text", "text": text_content}]
+            if role == "assistant":
+                parts.insert(0, {"type": "step-start"})
+
+        converted.append({
+            "role": role,
+            "content": text_content,
+            "parts": parts
+        })
+    return converted
+
+
 async def groqgenerate(json_data: Dict[str, Any]):
     try:
         messages = convert_to_groq_schema(json_data["messages"])
@@ -185,6 +210,82 @@ async def groqgenerate(json_data: Dict[str, Any]):
         generate(json_data)
 
 
+async def vercelXaigenerate(json_data: Dict[str, Any]):
+    headers = {
+        'accept': '*/*',
+        'accept-language': 'en-US,en;q=0.9,ja;q=0.8',
+        'content-type': 'application/json',
+        'origin': 'https://ai-sdk-starter-xai.vercel.app',
+        'referer': 'https://ai-sdk-starter-xai.vercel.app/',
+        'sec-ch-ua': '"Google Chrome";v="135", "Not-A.Brand";v="8", "Chromium";v="135"',
+        'sec-ch-ua-mobile': '?0',
+        'sec-ch-ua-platform': '"macOS"',
+        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36'
+    }
+
+    # Convert messages to XAI format
+    messages = conver_to_xai_schema(json_data["messages"])
+    
+    request_data = {
+        "id": "".join(random.choices("0123456789abcdef", k=16)),
+        "messages": messages,
+        "selectedModel": "grok-2-1212"
+    }
+
+    chunk_id = "xai-" + "".join(random.choices("0123456789abcdef", k=32))
+    created = int(asyncio.get_event_loop().time())
+    total_tokens = 0
+
+    try:
+        async with httpx.AsyncClient(timeout=None) as client:
+            async with client.stream(
+                "POST",
+                "https://ai-sdk-starter-xai.vercel.app/api/chat",
+                headers=headers,
+                json=request_data
+            ) as request_ctx:
+                if request_ctx.status_code == 200:
+                    async for line in request_ctx.aiter_lines():
+                        if line:
+                            if line.startswith('0:'):
+                                text = line[2:]
+                                response = {
+                                    "id": chunk_id,
+                                    "object": "chat.completion.chunk",
+                                    "created": created,
+                                    "model": json_data.get("model", "grok-2-1212"),
+                                    "choices": [{
+                                        "index": 0,
+                                        "text": text,
+                                        "logprobs": None,
+                                        "finish_reason": None
+                                    }],
+                                    "usage": None
+                                }
+                                yield f"data: {json.dumps(response)}\n\n"
+                                total_tokens += 1
+                            elif line.startswith('d:'):
+                                final = {
+                                    "id": chunk_id,
+                                    "object": "chat.completion.chunk", 
+                                    "created": created,
+                                    "model": json_data.get("model", "grok-2-1212"),
+                                    "choices": [],
+                                    "usage": {
+                                        "prompt_tokens": len(messages),
+                                        "completion_tokens": total_tokens,
+                                        "total_tokens": len(messages) + total_tokens
+                                    }
+                                }
+                                yield f"data: {json.dumps(final)}\n\n"
+                                yield "data: [DONE]\n\n"
+                    return
+                else:
+                    yield f"data: [Unexpected status code: {request_ctx.status_code}]\n\n"
+    except Exception as e:
+        yield f"data: [Connection error: {str(e)}]\n\n"
+
+
 @app.get("/")
 async def index():
     return {"status": "ok", "message": "Welcome to the Chipling API!", "version": "1.0", "routes": ["/chat", "/generate-modules", "/generate-topics"]}
@@ -218,7 +319,7 @@ async def chat(request: ChatRequest):
         'stream': True,
     }
 
-    selected_generator = random.choice([groqgenerate, generate])
+    selected_generator = random.choice([vercelXaigenerate, generate, groqgenerate])
     log_request("/chat", selected_generator.__name__)
     return StreamingResponse(selected_generator(json_data), media_type='text/event-stream')
 

logs.json

@@ -1 +1,17 @@
-[]
+[
+  {
+    "timestamp": "2025-04-12T11:48:38.183002",
+    "endpoint": "/chat",
+    "query": "vercelXaigenerate"
+  },
+  {
+    "timestamp": "2025-04-12T11:49:55.976382",
+    "endpoint": "/chat",
+    "query": "vercelXaigenerate"
+  },
+  {
+    "timestamp": "2025-04-12T11:51:54.228166",
+    "endpoint": "/chat",
+    "query": "vercelXaigenerate"
+  }
+]