Skyvern/streamlit_app/visualizer/sample_data.py

190 lines
8.9 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

from skyvern.forge.sdk.schemas.tasks import TaskRequest
class SampleTaskRequest(TaskRequest):
name: str
finditparts_sample_data = SampleTaskRequest(
name="Finditparts",
url="https://www.finditparts.com",
navigation_goal="Search for the specified product id, add it to cart and then navigate to the cart page",
data_extraction_goal="Extract all product quantity information from the cart page",
navigation_payload={
"product_id": "W01-377-8537",
},
)
geico_sample_data = SampleTaskRequest(
name="Geico",
url="https://www.geico.com",
navigation_goal="Navigate through the website until you generate an auto insurance quote. Do not generate a home insurance quote. If this page contains an auto insurance quote, consider the goal achieved",
data_extraction_goal="Extract all quote information in JSON format including the premium amount, the timeframe for the quote.",
navigation_payload={
"licensed_at_age": 19,
"education_level": "HIGH_SCHOOL",
"phone_number": "8042221111",
"full_name": "Chris P. Bacon",
"past_claim": [],
"has_claims": False,
"spouse_occupation": "Florist",
"auto_current_carrier": "None",
"home_commercial_uses": None,
"spouse_full_name": "Amy Stake",
"auto_commercial_uses": None,
"requires_sr22": False,
"previous_address_move_date": None,
"line_of_work": None,
"spouse_age": "1987-12-12",
"auto_insurance_deadline": None,
"email": "chris.p.bacon@abc.com",
"net_worth_numeric": 1000000,
"spouse_gender": "F",
"marital_status": "married",
"spouse_licensed_at_age": 20,
"license_number": "AAAAAAA090AA",
"spouse_license_number": "AAAAAAA080AA",
"how_much_can_you_lose": 25000,
"vehicles": [
{
"annual_mileage": 10000,
"commute_mileage": 4000,
"existing_coverages": None,
"ideal_coverages": {
"bodily_injury_per_incident_limit": 50000,
"bodily_injury_per_person_limit": 25000,
"collision_deductible": 1000,
"comprehensive_deductible": 1000,
"personal_injury_protection": None,
"property_damage_per_incident_limit": None,
"property_damage_per_person_limit": 25000,
"rental_reimbursement_per_incident_limit": None,
"rental_reimbursement_per_person_limit": None,
"roadside_assistance_limit": None,
"underinsured_motorist_bodily_injury_per_incident_limit": 50000,
"underinsured_motorist_bodily_injury_per_person_limit": 25000,
"underinsured_motorist_property_limit": None,
},
"ownership": "Owned",
"parked": "Garage",
"purpose": "commute",
"vehicle": {
"style": "AWD 3.0 quattro TDI 4dr Sedan",
"model": "A8 L",
"price_estimate": 29084,
"year": 2015,
"make": "Audi",
},
"vehicle_id": None,
"vin": None,
}
],
"additional_drivers": [],
"home": [
{
"home_ownership": "owned",
}
],
"spouse_line_of_work": "Agriculture, Forestry and Fishing",
"occupation": "Customer Service Representative",
"id": None,
"gender": "M",
"credit_check_authorized": False,
"age": "1987-11-11",
"license_state": "Washington",
"cash_on_hand": "$1000014999",
"address": {
"city": "HOUSTON",
"country": "US",
"state": "TX",
"street": "9625 GARFIELD AVE.",
"zip": "77082",
},
"spouse_education_level": "MASTERS",
"spouse_email": "amy.stake@abc.com",
"spouse_added_to_auto_policy": True,
},
extracted_information_schema={
"additionalProperties": False,
"properties": {
"quotes": {
"items": {
"additionalProperties": False,
"properties": {
"coverages": {
"items": {
"additionalProperties": False,
"properties": {
"amount": {
"description": "The coverage amount in USD, which can be a single value or a range (e.g., '$300,000' or '$300,000/$300,000').",
"type": "string",
},
"included": {
"description": "Indicates whether the coverage is included in the policy (true or False).",
"type": "boolean",
},
"type": {
"description": "The limit of the coverage (e.g., 'bodily_injury_limit', 'property_damage_limit', 'underinsured_motorist_bodily_injury_limit').\nTranslate the english name of the coverage to snake case values in the following list:\n * bodily_injury_limit\n * property_damage_limit\n * underinsured_motorist_bodily_injury_limit\n * personal_injury_protection\n * accidental_death\n * work_loss_exclusion\n",
"type": "string",
},
},
"type": "object",
},
"type": "array",
},
"premium_amount": {
"description": "The total premium amount for the whole quote timeframe in USD, formatted as a string (e.g., '$321.57').",
"type": "string",
},
"quote_number": {
"description": "The quote number generated by the carrier that identifies this quote",
"type": "string",
},
"timeframe": {
"description": "The duration of the coverage, typically expressed in months or years.",
"type": "string",
},
"vehicle_coverages": {
"items": {
"additionalProperties": False,
"properties": {
"collision_deductible": {
"description": "The collision deductible amount in USD, which is a single value (e.g., '$500') or null if it is not included",
"type": "string",
},
"comprehensive_deductible": {
"description": "The collision deductible amount in USD, which is a single value (e.g., '$500') or null if it is not included",
"type": "string",
},
"for_vehicle": {
"additionalProperties": False,
"description": "The vehicle that the collision and comprehensive coverage is for",
"properties": {
"make": {"description": "The make of the vehicle", "type": "string"},
"model": {"description": "The model of the vehicle", "type": "string"},
"year": {"description": "The year of the vehicle", "type": "string"},
},
"type": "object",
},
"underinsured_property_damage": {
"description": "The underinsured property damage limit for this vehicle, which is a limit and a deductible (e.g., '$25,000/$250 deductible') or null if it is not included",
"type": "string",
},
},
"type": "object",
},
"type": "array",
},
},
"type": "object",
},
"type": "array",
}
},
"type": "object",
},
)
supported_examples = [geico_sample_data, finditparts_sample_data]