Upload 2 files

app.py

@@ -0,0 +1,33 @@
+import subprocess
+import os
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+
+app = FastAPI(title="OpenCode API")
+
+class PromptRequest(BaseModel):
+    prompt: str
+
+@app.post("/ask")
+async def ask_openai(request: PromptRequest):
+    """接收用户提示，调用 opencode CLI 并返回结果"""
+    try:
+        # 调用 opencode -p "prompt" 并捕获输出
+        result = subprocess.run(
+            ["opencode", "-p", request.prompt],
+            capture_output=True,
+            text=True,
+            timeout=60,  # 防止长时间阻塞
+            env={**os.environ, "OPENCODE_DATA_DIR": "/data"}  # 确保数据持久化
+        )
+        if result.returncode != 0:
+            raise HTTPException(status_code=500, detail=result.stderr)
+        return {"response": result.stdout}
+    except subprocess.TimeoutExpired:
+        raise HTTPException(status_code=504, detail="Request timeout")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
+@app.get("/")
+def root():
+    return {"message": "OpenCode API is running. Use POST /ask with JSON {\"prompt\": \"your question\"}"}

requirements.txt

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