vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-10 15:35:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Refactor streamlit ui to allow more samples (#21)

Browse source
This commit is contained in:
Suchintan 2024-03-04 11:40:19 -05:00 • committed by GitHub
parent 065d2983fa
commit 0495552b11
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 95 additions and 92 deletions
Download patch file Download diff file Expand all files Collapse all files

42
streamlit_app/visualizer/sample_data.py
View file

@ -1,20 +1,21 @@
import json
from pydantic import BaseModel
def get_sample_url() -> str:
return "https://www.geico.com"
class SampleData(BaseModel):
name: str
url: str
navigation_goal: str
data_extraction_goal: str
navigation_payload: dict
extracted_information_schema: dict
def get_sample_navigation_goal() -> str:
return "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"
def get_sample_data_extraction_goal() -> str:
return "Extract all quote information in JSON format including the premium amount, the timeframe for the quote."
def get_sample_navigation_payload() -> str:
navigation_payload = {
geico_sample_data = SampleData(
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",
@ -97,13 +98,8 @@ def get_sample_navigation_payload() -> str:
"spouse_education_level": "MASTERS",
"spouse_email": "amy.stake@abc.com",
"spouse_added_to_auto_policy": True,
}
return json.dumps(navigation_payload)
def get_sample_extracted_information_schema() -> str:
extracted_information_schema = {
},
extracted_information_schema={
"additionalProperties": False,
"properties": {
"quotes": {
@ -181,5 +177,7 @@ def get_sample_extracted_information_schema() -> str:
}
},
"type": "object",
}
return json.dumps(extracted_information_schema)
},
)
supported_examples = [geico_sample_data]