RafalB commited on
Commit
379dfef
·
1 Parent(s): bdf26c9

Pass connected_services to /generate so ALEMBIC_API helper gets injected

Browse files
Files changed (1) hide show
  1. app.py +11 -5
app.py CHANGED
@@ -236,16 +236,19 @@ async def call_backend_agent(user_message: str, user_id: str, state: dict) -> di
236
  return result
237
 
238
 
239
- async def call_backend_generate(requirements: str, design_prefs: dict, user_id: str) -> dict:
240
  if MOCK_MODE:
241
  return await mock_build(requirements)
242
 
243
  from model_runtime import _post
244
- result = await _post("/generate", {
245
  "requirements": requirements,
246
  "design_prefs": design_prefs,
247
  "user_id": user_id,
248
- })
 
 
 
249
  return result
250
 
251
 
@@ -763,7 +766,10 @@ async def handle_generate(
763
  yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
764
 
765
  try:
766
- result = await call_backend_generate(requirements, design_prefs, session_state["user_id"])
 
 
 
767
 
768
  new_preview = BUILDING_HTML_TEMPLATE.format(
769
  status="Deploying...", message="Uploading to CloudFront...", progress=85
@@ -875,7 +881,7 @@ async def handle_recheck_services(chat_html, preview_html, gallery_html, session
875
  yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
876
 
877
  try:
878
- result = await call_backend_generate(requirements, {}, session_state["user_id"])
879
 
880
  new_preview = BUILDING_HTML_TEMPLATE.format(
881
  status="Deploying...", message="Uploading to CloudFront...", progress=85
 
236
  return result
237
 
238
 
239
+ async def call_backend_generate(requirements: str, design_prefs: dict, user_id: str, connected_services: list = None) -> dict:
240
  if MOCK_MODE:
241
  return await mock_build(requirements)
242
 
243
  from model_runtime import _post
244
+ payload = {
245
  "requirements": requirements,
246
  "design_prefs": design_prefs,
247
  "user_id": user_id,
248
+ }
249
+ if connected_services:
250
+ payload["connected_services"] = connected_services
251
+ result = await _post("/generate", payload)
252
  return result
253
 
254
 
 
766
  yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
767
 
768
  try:
769
+ # Pass connected services so codegen injects the API helper
770
+ services = (session_state.get("pending_build", {}).get("required_services")
771
+ or all_required_services or [])
772
+ result = await call_backend_generate(requirements, design_prefs, session_state["user_id"], services)
773
 
774
  new_preview = BUILDING_HTML_TEMPLATE.format(
775
  status="Deploying...", message="Uploading to CloudFront...", progress=85
 
881
  yield render_chat_html(session_state["chat_history"]), new_preview, gallery_html, session_state, "", HIDE_SAVE
882
 
883
  try:
884
+ result = await call_backend_generate(requirements, {}, session_state["user_id"], required_services)
885
 
886
  new_preview = BUILDING_HTML_TEMPLATE.format(
887
  status="Deploying...", message="Uploading to CloudFront...", progress=85