33 lines
824 B
Python
33 lines
824 B
Python
from fastapi import APIRouter, HTTPException
|
|
|
|
generated_code = [] # In-memory storage
|
|
|
|
router = APIRouter()
|
|
|
|
@router.post("/code-generation")
|
|
async def generate_code_demo(
|
|
prompt: str = "example prompt",
|
|
language: str = "python",
|
|
model: str = "gpt-3.5"
|
|
):
|
|
"""Demo code generation endpoint"""
|
|
if not prompt or not language:
|
|
raise HTTPException(status_code=400, detail="Missing required parameters")
|
|
|
|
code_id = len(generated_code) + 1
|
|
generated_code.append({
|
|
"id": code_id,
|
|
"prompt": prompt,
|
|
"language": language,
|
|
"model": model
|
|
})
|
|
|
|
return {
|
|
"message": "Code generated successfully",
|
|
"code_id": code_id,
|
|
"language": language,
|
|
"features": {
|
|
"tokens": 150,
|
|
"completion_time": 2.5
|
|
}
|
|
} |