send input mode to saas backend

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

View File

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

View File

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