Upload 4 files

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+WORKDIR /app
+
+RUN apt-get update && apt-get install -y --no-install-recommends build-essential \ 
+ && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY api.py .
+COPY skill_adapter.py .
+
+ENV PORT=7860
+EXPOSE 7860
+CMD ["uvicorn", "api:app", "--host", "0.0.0.0", "--port", "7860"]

api.py

@@ -0,0 +1,85 @@
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import HTMLResponse
+from pydantic import BaseModel
+from typing import Optional, Literal
+import asyncio, os, uuid, requests
+from datetime import datetime
+from skill_adapter import build_full_flow_plan, build_open_url_plan
+
+app = FastAPI()
+
+agents, queues, jobs, results, waiters = {}, {}, {}, {}, {}
+ADMIN_SECRET = os.environ.get("ADMIN_SECRET")
+
+def now(): return datetime.utcnow().isoformat()+"Z"
+
+def cond_for(agent_id): 
+    if agent_id not in waiters: waiters[agent_id]=asyncio.Condition()
+    return waiters[agent_id]
+
+def push_job(agent_id, payload):
+    job_id = str(uuid.uuid4())
+    job = {"id":job_id,"agent_id":agent_id,"payload":payload,"status":"queued","created_at":now(),"updated_at":now()}
+    jobs[job_id]=job
+    queues.setdefault(agent_id,[]).append(job)
+    async def notify(): 
+        async with cond_for(agent_id): cond_for(agent_id).notify_all()
+    asyncio.create_task(notify())
+    return job
+
+class RegisterBody(BaseModel): display_name:str
+class RegisterResp(BaseModel): agent_id:str; token:str
+class EnqueuePlanBody(BaseModel): agent_id:str; plan:list; admin_secret:Optional[str]=None
+class SubmitResultBody(BaseModel):
+    agent_id:str; token:str; job_id:str
+    status:Literal["completed","failed"]; message:Optional[str]=None; artifacts:Optional[dict]=None
+class FullFlowReq(BaseModel): agent_id:str; email:str; password:str; project_id:str; admin_secret:Optional[str]=None
+class OpenURLReq(BaseModel): agent_id:str; url:str; admin_secret:Optional[str]=None
+
+@app.post("/register", response_model=RegisterResp)
+async def register(body:RegisterBody):
+    agent_id, token = str(uuid.uuid4()), str(uuid.uuid4())
+    agents[agent_id]={"agent_id":agent_id,"token":token,"display_name":body.display_name,"registered_at":now(),"last_seen":now()}
+    queues[agent_id]=[]
+    return RegisterResp(agent_id=agent_id, token=token)
+
+@app.get("/agents") 
+async def list_agents(): return {"agents":list(agents.values())}
+
+@app.get("/next-job")
+async def next_job(agent_id:str, token:str, wait:int=25):
+    a=agents.get(agent_id)
+    if not a or a["token"]!=token: raise HTTPException(status_code=401, detail="bad agent/token")
+    a["last_seen"]=now(); q=queues.get(agent_id,[])
+    if q: job=q.pop(0); job["status"]="taken"; job["updated_at"]=now(); return {"job":job}
+    c=cond_for(agent_id)
+    try:
+        async with asyncio.timeout(wait):
+            async with c: await c.wait()
+            q=queues.get(agent_id,[])
+            if q: job=q.pop(0); job["status"]="taken"; job["updated_at"]=now(); return {"job":job}
+    except: pass
+    return {}
+
+@app.post("/submit-result")
+async def submit_result(body:SubmitResultBody):
+    j=jobs.get(body.job_id); 
+    if not j: raise HTTPException(status_code=404, detail="no job")
+    j["status"]=body.status; j["updated_at"]=now()
+    results[body.job_id]={"status":body.status,"message":body.message,"artifacts":body.artifacts,"submitted_at":now()}
+    return {"ok":True}
+
+@app.post("/skill/full-flow")
+async def skill_full_flow(body:FullFlowReq):
+    if ADMIN_SECRET and body.admin_secret!=ADMIN_SECRET: raise HTTPException(status_code=401, detail="bad secret")
+    plan=build_full_flow_plan(body.email, body.password, body.project_id)
+    job=push_job(body.agent_id, {"type":"plan","steps":plan}); return {"job":job}
+
+@app.post("/skill/open-url")
+async def skill_open_url(body:OpenURLReq):
+    if ADMIN_SECRET and body.admin_secret!=ADMIN_SECRET: raise HTTPException(status_code=401, detail="bad secret")
+    plan=build_open_url_plan(body.url)
+    job=push_job(body.agent_id, {"type":"plan","steps":plan}); return {"job":job}
+
+@app.get("/")
+async def home(): return HTMLResponse("<h3>SkillFlow Server Running</h3>")

requirements.txt

@@ -0,0 +1,5 @@
+fastapi==0.111.0
+uvicorn==0.30.1
+pydantic==2.7.4
+jinja2==3.1.4
+requests==2.32.3

skill_adapter.py

@@ -0,0 +1,45 @@
+from typing import List, Dict
+
+LAB_URL = "https://www.cloudskillsboost.google/course_templates/976/labs/550878"
+SIGN_IN_URL = "https://accounts.google.com/signin/v2/identifier"
+MARKETPLACE_URL = "https://console.cloud.google.com/marketplace/product/google/generativelanguage.googleapis.com"
+
+def step(act, **kw) -> Dict:
+    d = {"act": act}; d.update(kw); return d
+
+def wait_css(sel: str, timeout: int = 25) -> Dict:
+    return {"act": "wait", "by": "css", "sel": sel, "timeout": timeout}
+
+def click_css(sel: str) -> Dict:
+    return {"act": "click", "by": "css", "sel": sel}
+
+def type_css(sel: str, text: str) -> Dict:
+    return {"act": "type", "by": "css", "sel": sel, "text": text}
+
+def screenshot(label: str) -> Dict:
+    return {"act": "screenshot", "label": label}
+
+def click_xpath(xpath: str) -> Dict:
+    return {"act": "click", "by": "xpath", "sel": xpath}
+
+def wait_xpath(xpath: str, timeout: int = 25) -> Dict:
+    return {"act": "wait", "by": "xpath", "sel": xpath, "timeout": timeout}
+
+def build_open_url_plan(url: str) -> List[Dict]:
+    return [ step("get", sel=url), wait_css("body", 20), screenshot("opened") ]
+
+def build_full_flow_plan(email: str, password: str, project_id: str) -> List[Dict]:
+    steps: List[Dict] = []
+    # NOTE: shortened for clarity; fill in with full plan steps from previous response
+    steps += [
+        step("get", sel=SIGN_IN_URL),
+        wait_css("input[type=email]"),
+        type_css("input[type=email]", email),
+        click_css("#identifierNext"),
+        wait_css("input[type=password]"),
+        type_css("input[type=password]", password),
+        click_css("#passwordNext"),
+        wait_css("body", 40),
+        screenshot("after_login")
+    ]
+    return steps