Loads a prompt text from a file. Args: base_dir: The directory containing the 'prompts' subdirectory. Typically pa

Loads requirements from pyproject.toml.

Load test cases from a JSON file.

Forecasts regional labor market disruption and AI diffusion shifts (automation risk, productivity growth, and occupational transition forecas

Extracts the first color found if exists

Computes the expected number of rate moves between two prices. Args: front_ff_future_px: Front fed funds future price. back_ff_future

Parse action string to action name and its arguments.

Process PDF pages in parallel with Gemini. Args: pages: List of single-page PDF bytes. genai_client: Gemini client instance.

Pop the oldest long-running call information. Returns: A tuple containing the call ID and call name, or None if no information

Pop long-running call information by call ID. Args: call_id: The ID of the long-running call to pop. Returns: A

Mock posting to a platform.

Return probabilities for a single image.

Return probabilities for a batch.

Preprocess glasses image: remove background to isolate the product. Returns: bytes: The preprocessed glasses image with background remove

Preprocess full body model image: remove background and upscale. Args: client: Gemini client instance img_bytes (bytes): The inpu

Preprocess person image: remove background and upscale. Args: client: Gemini client instance for background removal upscale_clien

Main pipeline orchestrator

Replace GCS URLs in a string with Proxy or Signed URLs.

Generate random index by sampling uniformly from sum of all weights, then selecting the `min` between the position to keep the list sorted (via b

Collect and log feedback.

Collect and log feedback.

Collect and log feedback.

Fast background removal for white/gray backgrounds using HSV color detection. This is a fast alternative to ML-based background removal (like rem

Create a new session and reset environment variables

Reconstruct a GCS URI from a filename based on naming convention. Filenames follow the pattern: {timestamp}_{type}_{i}.{ext} - _u_ → user upl

Load conversation history from GCS for the given session. Returns an empty list if the session file doesn't exist yet.

Run the full glasses VTO pipeline. Yields results as each variation completes. Yield order: 1. A single VTOResult with status="reference_f

Run the full video VTO pipeline. Yields dicts suitable for SSE. Runs image VTO first to generate the best static VTO image, then delegates to

Saves the deck_spec to persistent session STATE. Returns an invisible confirmation instead of a visible filename.

Inspects the LLM request for user-uploaded files and saves them as artifacts.

Save all progress data for a run. Args: predictions_data: A dictionary of prediction results. trajectories_data: A dictionary of traj

Search the chunks collection (hybrid semantic + text), then resolve each unique file_id hit to its full document text via the documents KV col

Search for messages using LangChain Gmail toolkit. Args: query: Gmail search query max_results: Maximum numb

Initialize session and return the search results page

Searches for a product in BigQuery by its name within the search_tags array. Args: product_name (str): The name of the product t

Send an email using LangChain Gmail toolkit. Args: to: Recipient email address subject: Email subject

Enable or disable debug image saving.

Sets a key-value pair in the mutable mapping. Args: key: The key to set. value: The value to set. Returns:

Sets an API key in the current session's environment variables. Use this when the user provides a missing API key in the chat. Args:

Configure OpenTelemetry and GenAI telemetry with GCS upload.

Configure OpenTelemetry and GenAI telemetry with GCS upload.

Map Invoice Status to decision class using master data or fallback.

Takes an action, updates WebShop environment, and returns (observation, reward, done, info) Arguments: action (`str`): An action sho

* Converts the Customer object to a JSON string. * * @returns A JSON string representing the Customer object.

Convert to a dict suitable for SSE JSON serialisation.

Upload an asset to GCS. Args: data: The file bytes. blob_path: Full blob path within the bucket. content_type: The conten

Get the GCS blob path for a user upload.

Validate that all required configuration is present.

Live web search using Serper.dev API. Ensure SERPER_API_KEY is defined in your environment or .env file.

Write deployment metadata to file.

AddItemPhoto ensures the checklist item exists (via UPSERT) and then links the artifact to it.

AddTraceToContext adds the trace to the context

Artifact operations

CreateSession - POST /api/agent/sessions

DeleteTrip removes a trip and all its associated data (checklists, artifacts).

FindCrewMember attempts to find a crew member by name using fuzzy matching.

FindUserByName retrieves a user by their name (case-insensitive).

GetArtifact handles the retrieval of an artifact's data.

GetArtifactByID retrieves artifact metadata by ID.

Trip operations

GetPublicURL returns a public URL for the file if supported.

GetPublicURL for local disk returns an empty string as it is not supported.

GetSession - GET /api/agent/sessions/{sessionID}?userId=...

GetTraceFromContext returns the trace from the context

GetTrip handles the retrieval of a trip's details.

GetUser retrieves a user by ID.

HandleConnect authenticates the user based on the context populated by middleware.

InitLogging sets up the global logger based on the environment.

ListenToDB starts a loop to listen for database notifications and publish them to Centrifuge.

Load populates the Config struct using the provided environment lookup function.

Load retrieves the data from the local disk.

LoggingMiddleware logs the details of each request.