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

store stack and some other properties for generations on prod

Browse Source
This commit is contained in:
Abi Raja 2024-06-07 15:05:26 -04:00
parent df5f954ee2
commit ae08466405
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/routes/generate_code.py
View File

@ -363,6 +363,9 @@ async def stream_code(websocket: WebSocket):
completion, completion,
payment_method=payment_method, payment_method=payment_method,
llm_version=exact_llm_version, llm_version=exact_llm_version,
stack=valid_stack,
is_imported_from_code=bool(params.get("isImportedFromCode", False)),
includes_result_image=bool(params.get("resultImage", False)),
auth_token=params["authToken"], auth_token=params["authToken"],
) )
except Exception as e: except Exception as e:

7
backend/routes/logging_utils.py
View File

@ -6,6 +6,7 @@ import json
from config import BACKEND_SAAS_URL, IS_PROD from config import BACKEND_SAAS_URL, IS_PROD
from llm import Llm from llm import Llm
from prompts.types import Stack
class PaymentMethod(Enum): class PaymentMethod(Enum):
@ -21,6 +22,9 @@ async def send_to_saas_backend(
completion: str, completion: str,
payment_method: PaymentMethod, payment_method: PaymentMethod,
llm_version: Llm, llm_version: Llm,
stack: Stack,
is_imported_from_code: bool,
includes_result_image: bool,
auth_token: str | None = None, auth_token: str | None = None,
): ):
if IS_PROD: if IS_PROD:
@ -33,6 +37,9 @@ async def send_to_saas_backend(
"completion": completion, "completion": completion,
"payment_method": payment_method.value, "payment_method": payment_method.value,
"llm_version": llm_version.value, "llm_version": llm_version.value,
"stack": stack,
"is_imported_from_code": is_imported_from_code,
"includes_result_image": includes_result_image,
} }
) )