34 lines
1020 B
Python
34 lines
1020 B
Python
import httpx
|
|
from openai.types.chat import ChatCompletionMessageParam
|
|
from typing import List
|
|
import json
|
|
|
|
from config import IS_PROD
|
|
|
|
|
|
async def send_to_saas_backend(
|
|
prompt_messages: List[ChatCompletionMessageParam],
|
|
completion: str,
|
|
auth_token: str | None = None,
|
|
):
|
|
if IS_PROD:
|
|
async with httpx.AsyncClient() as client:
|
|
url = "https://screenshot-to-code-saas.onrender.com/generations/store"
|
|
# url = "http://localhost:8001/generations/store"
|
|
|
|
data = json.dumps(
|
|
{
|
|
"prompt": json.dumps(prompt_messages),
|
|
"completion": completion,
|
|
}
|
|
)
|
|
|
|
headers = {
|
|
"Content-Type": "application/json",
|
|
"Authorization": f"Bearer {auth_token}", # Add the auth token to the headers
|
|
}
|
|
|
|
response = await client.post(url, content=data, headers=headers)
|
|
response_data = response.json()
|
|
return response_data
|