save

DadKnowsSiegeBest-api

@@ -0,0 +1 @@
+Subproject commit f87ff9e0af3e0fbb9a562fce7421d933a164a1af

Dockerfile

@@ -0,0 +1,14 @@
+
+FROM python:3.9
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,10 +1,11 @@
 ---
-title: Dad Knows Best Api
-emoji: 🐠
-colorFrom: blue
-colorTo: blue
+title: DadKnowsSiegeBest Api
+emoji: 📊
+colorFrom: gray
+colorTo: yellow
 sdk: docker
 pinned: false
+short_description: The backend for our submission for the Mistral GameJam
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,55 @@
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+import os
+import httpx
+
+# Load Mistral API token from environment variables
+MISTRAL_API_TOKEN = os.getenv("MISTRAL_API_TOKEN")
+if not MISTRAL_API_TOKEN:
+    raise EnvironmentError("MISTRAL_API_TOKEN environment variable not set.")
+
+# Initialize FastAPI app
+app = FastAPI()
+
+# Define the request model for input validation
+class MistralRequest(BaseModel):
+    system_prompt : str
+    input_text: str
+
+@app.post("/proxy/mistral")
+async def proxy_mistral(request_data: MistralRequest):
+    """Proxy endpoint to send requests to Mistral API."""
+    url = "https://api.mistral.ai/v1/chat/completions"
+    headers = {
+        "Authorization": f"Bearer {MISTRAL_API_TOKEN}",
+        "Content-Type": "application/json"
+    }
+
+    # Construct the data payload
+    payload = {
+        "messages": [
+            {
+              "role" : "system",
+              "content" : request_data.system_prompt
+            },
+            {
+                "role": "user",
+                "content": request_data.input_text
+            }
+        ]
+    }
+
+    try:
+        # Send the request to the Mistral API
+        async with httpx.AsyncClient() as client:
+            response = await client.post(url, headers=headers, json=payload)
+
+        # Check the response status
+        if response.status_code != 200:
+            raise HTTPException(status_code=response.status_code, detail=response.text)
+
+        # Return the Mistral API response
+        return response.json()
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))

requirements.txt

@@ -0,0 +1,2 @@
+fastapi
+uvicorn[standard]