You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
33 lines
789 B
33 lines
789 B
12 months ago
|
"""
|
||
|
|
Exposes a SSE streaming server endpoint at /run, which recieves language and code,
|
||
|
|
and streams the output.
|
||
|
|
"""
|
||
12 months ago
|
|
||
|
|
from dotenv import load_dotenv
|
||
|
|
load_dotenv() # take environment variables from .env.
|
||
|
|
|
||
12 months ago
|
import os
|
||
|
12 months ago
|
import json
|
||
|
|
from interpreter import interpreter
|
||
|
|
import uvicorn
|
||
|
|
|
||
|
|
from fastapi import FastAPI
|
||
|
|
from fastapi.responses import StreamingResponse
|
||
|
|
from pydantic import BaseModel
|
||
|
|
|
||
|
|
class Code(BaseModel):
|
||
|
|
language: str
|
||
|
|
code: str
|
||
|
|
|
||
|
|
app = FastAPI()
|
||
|
|
|
||
|
|
@app.post("/run")
|
||
|
|
async def run_code(code: Code):
|
||
|
|
def generator():
|
||
|
12 months ago
|
for chunk in interpreter.computer.run(code.language, code.code):
|
||
|
12 months ago
|
yield json.dumps(chunk)
|
||
|
|
return StreamingResponse(generator())
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
12 months ago
|
uvicorn.run(app, host="0.0.0.0", port=int(os.getenv('COMPUTER_PORT', 9000)))
|