|
- # api/routes/models.py
- from fastapi import APIRouter, Depends, HTTPException
- from typing import Dict
- from openai import OpenAIError
- from ..deps import get_db
- from ...datamodel import Model
- from ...utils import test_model
-
- router = APIRouter()
-
-
- @router.get("/")
- async def list_models(
- user_id: str,
- db=Depends(get_db)
- ) -> Dict:
- """List all models for a user"""
- response = db.get(Model, filters={"user_id": user_id})
- return {
- "status": True,
- "data": response.data
- }
-
-
- @router.get("/{model_id}")
- async def get_model(
- model_id: int,
- user_id: str,
- db=Depends(get_db)
- ) -> Dict:
- """Get a specific model"""
- response = db.get(
- Model,
- filters={"id": model_id, "user_id": user_id}
- )
- if not response.status or not response.data:
- raise HTTPException(status_code=404, detail="Model not found")
- return {
- "status": True,
- "data": response.data[0]
- }
-
-
- @router.post("/")
- async def create_model(
- model: Model,
- db=Depends(get_db)
- ) -> Dict:
- """Create a new model"""
- response = db.upsert(model)
- if not response.status:
- raise HTTPException(status_code=400, detail=response.message)
- return {
- "status": True,
- "data": response.data
- }
-
-
- @router.delete("/{model_id}")
- async def delete_model(
- model_id: int,
- user_id: str,
- db=Depends(get_db)
- ) -> Dict:
- """Delete a model"""
- response = db.delete(
- filters={"id": model_id, "user_id": user_id},
- model_class=Model
- )
- return {
- "status": True,
- "message": "Model deleted successfully"
- }
-
-
- @router.post("/test")
- async def test_model_endpoint(model: Model) -> Dict:
- """Test a model configuration"""
- try:
- response = test_model(model)
- return {
- "status": True,
- "message": "Model tested successfully",
- "data": response
- }
- except OpenAIError as e:
- raise HTTPException(
- status_code=400,
- detail=f"OpenAI API error: {str(e)}"
- )
- except Exception as e:
- raise HTTPException(
- status_code=500,
- detail=f"Error testing model: {str(e)}"
- )
|
