added models endpoint

README.md

@@ -55,4 +55,8 @@ The Space will build the Docker image and start the service. The API endpoint wi
 *   **Method:** `POST`
 *   **Request Body:** Send a JSON payload conforming to the OpenAI Chat Completions API schema (e.g., specifying `model`, `messages`, `stream`, etc.). The `model` field should correspond to a key in [`models.json`](models.json).
 *   **Authentication:** Requests must include an `Authorization` header with your API key. Use the format `Bearer your_api_key`. For example, if using the default key, the header would be `Authorization: Bearer 123456`.
-*   **Response:** The API will return either a standard JSON response or a server-sent event stream, mimicking the OpenAI API behavior based on the `stream` parameter in the request.
+*   **Response:** The API will return either a standard JSON response or a server-sent event stream, mimicking the OpenAI API behavior based on the `stream` parameter in the request.
+*   **URL:** `/v1/models`
+*   **Method:** `GET`
+*   **Description:** Returns a list of available models supported by the proxy, based on the configuration in [`models.json`](models.json).
+*   **Authentication:** Requires the same `Authorization: Bearer <API_KEY>` header as the chat completions endpoint.

main.py

@@ -62,6 +62,17 @@ class FlowithRequest(BaseModel):
     stream: bool
     nodeId: str # UUID for Flowith
 
+
+# --- OpenAI Models Endpoint Models ---
+class ModelCard(BaseModel):
+    id: str
+    object: str = "model"
+    # owned_by: str = "user" # Optional: Add other fields if needed
+
+class ModelList(BaseModel):
+    object: str = "list"
+    data: List[ModelCard]
+
 # --- FastAPI App ---
 app = FastAPI(
     title="OpenAI to Flowith Proxy",
@@ -245,6 +256,19 @@ async def chat_completions(
              raise HTTPException(status_code=500, detail=f"Internal server error: {exc}")
 
 
+# --- Models Endpoint ---
+@app.get("/v1/models", response_model=ModelList)
+async def list_models(api_key: str = Depends(verify_api_key)): # Protect with existing auth
+    """
+    Lists the available models based on the models.json mapping.
+    Follows the OpenAI API format.
+    """
+    model_cards = [
+        ModelCard(id=model_id) for model_id in model_mappings.keys()
+    ]
+    return ModelList(data=model_cards)
+
+
 # --- Optional: Add a root endpoint for health check ---
 @app.get("/")
 async def root():