From 7f7d0c3504c89676f9dd22dab9fa3150434f9163 Mon Sep 17 00:00:00 2001 From: Abi Raja Date: Tue, 9 Jul 2024 16:45:07 -0400 Subject: [PATCH] send input mode to saas backend --- backend/routes/generate_code.py | 1 + backend/routes/logging_utils.py | 3 +++ 2 files changed, 4 insertions(+) diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py index 4d17696..d37bc26 100644 --- a/backend/routes/generate_code.py +++ b/backend/routes/generate_code.py @@ -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: diff --git a/backend/routes/logging_utils.py b/backend/routes/logging_utils.py index ae27c87..684f2ff 100644 --- a/backend/routes/logging_utils.py +++ b/backend/routes/logging_utils.py @@ -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, } )