Spaces:
Running
on
Zero
Running
on
Zero
File size: 387 Bytes
0bdbec3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 |
from __future__ import annotations
from typing import Dict
from ..llm.gemini import call_gemini
from ..state import AppState
def run(state: AppState) -> AppState:
if state.get("prompts"):
return state
res: Dict = call_gemini("prompt_generate", spec_text=state.get("spec_text", ""), K=state.get("K", 3))
state["prompts"] = res.get("prompts", [])
return state
|