new variables ux and schema

bots/models.py

@@ -17,7 +17,7 @@
 from bots.custom_fields import PostgresJSONEncoder, CustomURLField
 from daras_ai_v2 import icons, urls
 from daras_ai_v2.crypto import get_random_doc_id
-from daras_ai_v2.fastapi_tricks import get_route_path
+from daras_ai_v2.fastapi_tricks import get_route_path, get_api_route_url
 from daras_ai_v2.language_model import format_chat_entry
 from functions.models import CalledFunctionResponse
 from gooeysite.bg_db_conn import get_celery_result_db_safe
@@ -827,6 +827,10 @@ def get_web_widget_config(self, target="#gooey-embed") -> dict:
                 )
             ),
         )
+        if settings.DEBUG:
+            from routers.bots_api import stream_create
+
+            config["apiUrl"] = get_api_route_url(stream_create)
         return config
 
     def translate(self, text: str) -> str:

daras_ai_v2/base.py

@@ -48,13 +48,13 @@
 from daras_ai_v2.html_spinner_widget import html_spinner, scroll_into_view
 from daras_ai_v2.manage_api_keys_widget import manage_api_keys
 from daras_ai_v2.meta_preview_url import meta_preview_url
-from daras_ai_v2.prompt_vars import variables_input
 from daras_ai_v2.query_params_util import extract_query_params
 from daras_ai_v2.ratelimits import ensure_rate_limits, RateLimitExceeded
 from daras_ai_v2.send_email import send_reported_run_email
 from daras_ai_v2.urls import paginate_queryset, paginate_button
 from daras_ai_v2.user_date_widgets import render_local_dt_attrs
-from functions.models import RecipeFunction, FunctionTrigger
+from daras_ai_v2.variables_widget import variables_input
+from functions.models import RecipeFunction, FunctionTrigger, VariableSchema
 from functions.recipe_functions import (
     functions_input,
     call_recipe_functions,
@@ -142,6 +142,10 @@ class RequestModel(BaseModel):
             title="⌥ Variables",
             description="Variables to be used as Jinja prompt templates and in functions as arguments",
         )
+        variables_schema: dict[str, VariableSchema] | None = Field(
+            title="Variables Schema",
+            description="Schema for variables to be used in the variables input",
+        )
 
     ResponseModel: typing.Type[BaseModel]
 

daras_ai_v2/bots.py

@@ -356,16 +356,18 @@ def _process_and_send_msg(
     # get latest messages for context
     saved_msgs = bot.convo.msgs_for_llm_context()
 
-    variables = (bot.saved_run.state.get("variables") or {}) | build_run_vars(
-        bot.convo, bot.user_msg_id
-    )
+    system_vars, system_vars_schema = build_system_vars(bot.convo, bot.user_msg_id)
+    state = bot.saved_run.state
+    variables = (state.get("variables") or {}) | system_vars
+    variables_schema = (state.get("variables_schema") or {}) | system_vars_schema
     body = dict(
         input_prompt=input_text,
         input_audio=input_audio,
         input_images=input_images,
         input_documents=input_documents,
         messages=saved_msgs,
         variables=variables,
+        variables_schema=variables_schema,
     )
     if bot.user_language:
         body["user_language"] = bot.user_language
@@ -484,7 +486,7 @@ def _process_and_send_msg(
     )
 
 
-def build_run_vars(convo: Conversation, user_msg_id: str):
+def build_system_vars(convo: Conversation, user_msg_id: str) -> tuple[dict, dict]:
     from routers.bots_api import MSG_ID_PREFIX
 
     bi = convo.bot_integration
@@ -524,7 +526,8 @@ def build_run_vars(convo: Conversation, user_msg_id: str):
             variables["bot_twilio_phone_number"] = (
                 bi.twilio_phone_number and bi.twilio_phone_number.as_international
             )
-    return variables
+    variables_schema = {var: {"role": "system"} for var in variables}
+    return variables, variables_schema
 
 
 def _save_msgs(

daras_ai_v2/query_generator.py

@@ -5,7 +5,7 @@
 from daras_ai_v2.language_model import (
     run_language_model,
 )
-from daras_ai_v2.prompt_vars import render_prompt_vars
+from daras_ai_v2.variables_widget import render_prompt_vars
 
 Model = typing.TypeVar("Model", bound=BaseModel)