Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

send input mode to saas backend

Browse Source
This commit is contained in:
Abi Raja 2024-07-09 16:45:07 -04:00
parent 8c11366bd0
commit 7f7d0c3504
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
backend/routes/generate_code.py
View File

@ -393,6 +393,7 @@ async def stream_code(websocket: WebSocket):
stack=valid_stack, stack=valid_stack,
is_imported_from_code=bool(params.get("isImportedFromCode", False)), is_imported_from_code=bool(params.get("isImportedFromCode", False)),
includes_result_image=bool(params.get("resultImage", False)), includes_result_image=bool(params.get("resultImage", False)),
input_mode=validated_input_mode,
auth_token=params["authToken"], auth_token=params["authToken"],
) )
except Exception as e: except Exception as e:

3
backend/routes/logging_utils.py
View File

@ -5,6 +5,7 @@ from typing import List
import json import json
from config import BACKEND_SAAS_URL, IS_PROD from config import BACKEND_SAAS_URL, IS_PROD
from custom_types import InputMode
from llm import Llm from llm import Llm
from prompts.types import Stack from prompts.types import Stack
@ -25,6 +26,7 @@ async def send_to_saas_backend(
stack: Stack, stack: Stack,
is_imported_from_code: bool, is_imported_from_code: bool,
includes_result_image: bool, includes_result_image: bool,
input_mode: InputMode,
auth_token: str | None = None, auth_token: str | None = None,
): ):
if IS_PROD: if IS_PROD:
@ -40,6 +42,7 @@ async def send_to_saas_backend(
"stack": stack, "stack": stack,
"is_imported_from_code": is_imported_from_code, "is_imported_from_code": is_imported_from_code,
"includes_result_image": includes_result_image, "includes_result_image": includes_result_image,
"input_mode": input_mode,
} }
) )