optexity 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

optexity/examples/add_example.py

@@ -0,0 +1,88 @@
+import argparse
+import logging
+from urllib.parse import urljoin
+
+import httpx
+
+from optexity.examples import (
+    download_pdf_url,
+    file_upload,
+    i94,
+    i94_travel_history,
+    peachstate_medicaid,
+    supabase_login,
+)
+from optexity.utils.settings import Settings
+
+logger = logging.getLogger(__name__)
+settings = Settings()
+
+logger.setLevel(logging.INFO)
+
+
+def main(args):
+    if args.example == "i94":
+        example = i94
+    elif args.example == "i94_travel_history":
+        example = i94_travel_history
+    elif args.example == "peachstate_medicaid":
+        example = peachstate_medicaid
+    elif args.example == "supabase_login":
+        example = supabase_login
+    elif args.example == "download_pdf_url":
+        example = download_pdf_url
+    elif args.example == "file_upload":
+        example = file_upload
+    else:
+        raise ValueError(f"Invalid example: {args.example}")
+    try:
+        logger.info(f"➕ Adding example: {args.example}")
+        headers = {"x-api-key": settings.API_KEY}
+        with httpx.Client() as client:
+            response = client.post(
+                urljoin(
+                    settings.SERVER_URL,
+                    (
+                        settings.ADD_EXAMPLE_ENDPOINT
+                        if not args.update
+                        else settings.UPDATE_EXAMPLE_ENDPOINT
+                    ),
+                ),
+                headers=headers,
+                json={
+                    "automation": example.automation.model_dump(
+                        exclude_none=True, exclude_defaults=True
+                    ),
+                    "description": example.description,
+                    "endpoint_name": example.endpoint_name,
+                },
+            )
+            response.raise_for_status()
+            logger.info(f"✓ Example added successfully: {response.json()}")
+    except Exception as e:
+        logger.error(f"❌ Error adding example: {response.json()}")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument(
+        "--example",
+        type=str,
+        choices=[
+            "i94",
+            "i94_travel_history",
+            "peachstate_medicaid",
+            "supabase_login",
+            "download_pdf_url",
+            "file_upload",
+        ],
+        required=True,
+    )
+    parser.add_argument(
+        "--update",
+        action="store_true",
+        default=False,
+    )
+    args = parser.parse_args()
+
+    main(args)

optexity/examples/download_pdf_url.py

@@ -0,0 +1,29 @@
+from optexity.schema.automation import Automation
+
+description = "Download PDF URL Example"
+endpoint_name = "download_pdf_url"
+automation_json = {
+    "url": "about:blank",
+    "parameters": {
+        "input_parameters": {
+            "pdf_url": ["https://s24.q4cdn.com/216390268/files/doc_downloads/test.pdf"]
+        },
+        "generated_parameters": {},
+    },
+    "nodes": [
+        {
+            "type": "action_node",
+            "interaction_action": {"go_to_url": {"url": "{pdf_url[0]}"}},
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "download_url_as_pdf": {"download_filename": "example.pdf"}
+            },
+            "end_sleep_time": 1.0,
+        },
+    ],
+}
+
+automation = Automation.model_validate(automation_json)

optexity/examples/extract_price_stockanalysis.py

@@ -0,0 +1,44 @@
+from optexity.schema.automation import Automation
+
+description = "Extract stock price from StockAnalysis"
+endpoint_name = "extract_price_stockanalysis"
+automation_json = {
+    "url": "https://stockanalysis.com/",
+    "parameters": {
+        "input_parameters": {"stock_ticker": ["AAPL"]},
+        "generated_parameters": {},
+    },
+    "nodes": [
+        {
+            "interaction_action": {
+                "input_text": {
+                    "command": 'locator("#search-header")',
+                    "prompt_instructions": "Fill the input field with ID 'search-header' with the value of the 'stock_ticker' variable.",
+                    "input_text": "{stock_ticker[0]}",
+                }
+            }
+        },
+        {
+            "interaction_action": {
+                "click_element": {
+                    "prompt_instructions": "Click on the link with the name of the stock equivalent for {stock_ticker[0]}."
+                }
+            }
+        },
+        {
+            "extraction_action": {
+                "llm": {
+                    "source": ["screenshot"],
+                    "extraction_format": {
+                        "stock_name": "str",
+                        "stock_price": "str",
+                        "stock_symbol": "str",
+                    },
+                    "extraction_instructions": "Extract the stock price, stock name, and stock symbol from the webpage.",
+                }
+            }
+        },
+    ],
+}
+
+automation = Automation.model_validate(automation_json)

optexity/examples/file_upload.py

@@ -0,0 +1,59 @@
+from optexity.schema.automation import Automation
+
+description = "File Upload Example"
+endpoint_name = "file_upload"
+automation_json = {
+    "url": "https://www.azurespeed.com/Azure/UploadLargeFile",
+    "parameters": {
+        "input_parameters": {
+            "target_region_option": ["test_region"],
+            "file_path": ["/path/to/test/file.txt"],
+        },
+        "generated_parameters": {},
+    },
+    "nodes": [
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "select_option": {
+                    "command": 'get_by_label("Target region")',
+                    "prompt_instructions": "Select an option from the field labeled 'Target region' with the value from the 'target_region_option' variable.",
+                    "select_values": ["{target_region_option[0]}"],
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "upload_file": {
+                    "command": 'get_by_role("button", name="Test file")',
+                    "prompt_instructions": "Click on the 'Test file' button.",
+                    "file_path": "{file_path[0]}",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="Start test")',
+                    "prompt_instructions": "Click the 'Start test' button",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "assertion_action": {
+                "llm": {
+                    "extraction_instructions": "Check if the file upload was successful"
+                }
+            },
+            "before_sleep_time": 10.0,
+            "end_sleep_time": 0.0,
+        },
+    ],
+}
+automation = Automation.model_validate(automation_json)

optexity/examples/i94.py

@@ -0,0 +1,126 @@
+from optexity.schema.automation import Automation
+
+description = "I94 Example"
+endpoint_name = "i94"
+automation_json = {
+    "url": "https://i94.cbp.dhs.gov/search/recent-search",
+    "nodes": [
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "before_sleep_time": 3,
+            "python_script_action": {
+                "execution_code": 'async def code_fn(page):\n    print("entering code_fn")\n    await page.evaluate(\n        """  const el = document.querySelector(\'mat-dialog-content\');  if (el) el.scrollTop = el.scrollHeight;"""\n    )\n    print("exiting code_fn")\n'
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="I ACKNOWLEDGE AND AGREE")',
+                    "prompt_instructions": "Click the I ACKNOWLEDGE AND AGREE button",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your first name")',
+                    "input_text": "{first_name[0]}",
+                    "prompt_instructions": "Enter the First Name",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your last name")',
+                    "input_text": "{last_name[0]}",
+                    "prompt_instructions": "Enter the Last Name",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Date of Birth")',
+                    "input_text": "{date_of_birth[0]}",
+                    "prompt_instructions": "Enter the Date of Birth",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your document")',
+                    "input_text": "{document_number[0]}",
+                    "prompt_instructions": "Enter the Document Number",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("combobox", name="Please enter your document")',
+                    "input_text": "{nationality[0]}",
+                    "prompt_instructions": "Enter the Nationality",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "prompt_instructions": "Select {nationality[0]} from the options. Be careful to select the correct option. which will be of the format `nationality (code)`"
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="Click to submit the form")',
+                    "prompt_instructions": "Click the Submit button",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 0,
+            "before_sleep_time": 3,
+            "extraction_action": {
+                "network_call": {
+                    "extract_from": "response",
+                    "url_pattern": "https://i94.cbp.dhs.gov/api/services/i94/recent",
+                }
+            },
+        },
+    ],
+    "parameters": {
+        "input_parameters": {
+            "last_name": ["Last Name"],
+            "first_name": ["First Name"],
+            "nationality": ["IND"],
+            "date_of_birth": ["MM/DD/YYYY"],
+            "document_number": ["Document Number"],
+        },
+        "generated_parameters": {},
+    },
+    "browser_channel": "chrome",
+}
+
+
+automation = Automation.model_validate(automation_json)

optexity/examples/i94_travel_history.py

@@ -0,0 +1,126 @@
+from optexity.schema.automation import Automation
+
+description = "I94 Travel History Example"
+endpoint_name = "get_i94_travel_history"
+automation_json = {
+    "url": "https://i94.cbp.dhs.gov/search/history-search",
+    "nodes": [
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "before_sleep_time": 3,
+            "python_script_action": {
+                "execution_code": 'async def code_fn(page):\n    print("entering code_fn")\n    await page.evaluate(\n        """  const el = document.querySelector(\'mat-dialog-content\');  if (el) el.scrollTop = el.scrollHeight;"""\n    )\n    print("exiting code_fn")\n'
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="I ACKNOWLEDGE AND AGREE")',
+                    "prompt_instructions": "Click the I ACKNOWLEDGE AND AGREE button",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your first name")',
+                    "input_text": "{first_name[0]}",
+                    "prompt_instructions": "Enter the First Name",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your last name")',
+                    "input_text": "{last_name[0]}",
+                    "prompt_instructions": "Enter the Last Name",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Date of Birth")',
+                    "input_text": "{date_of_birth[0]}",
+                    "prompt_instructions": "Enter the Date of Birth",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Please enter your document")',
+                    "input_text": "{document_number[0]}",
+                    "prompt_instructions": "Enter the Document Number",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("combobox", name="Please enter your document")',
+                    "input_text": "{nationality[0]}",
+                    "prompt_instructions": "Enter the Nationality",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "prompt_instructions": "Select {nationality[0]} from the options. Be careful to select the correct option. which will be of the format `nationality (code)`"
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 1,
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="Click to submit the form")',
+                    "prompt_instructions": "Click the Submit button",
+                }
+            },
+        },
+        {
+            "type": "action_node",
+            "end_sleep_time": 0,
+            "before_sleep_time": 3,
+            "extraction_action": {
+                "network_call": {
+                    "extract_from": "response",
+                    "url_pattern": "https://i94.cbp.dhs.gov/api/services/travel/history",
+                }
+            },
+        },
+    ],
+    "parameters": {
+        "input_parameters": {
+            "last_name": ["Last Name"],
+            "first_name": ["First Name"],
+            "nationality": ["IND"],
+            "date_of_birth": ["MM/DD/YYYY"],
+            "document_number": ["Document Number"],
+        },
+        "generated_parameters": {},
+    },
+    "browser_channel": "chrome",
+}
+
+
+automation = Automation.model_validate(automation_json)

optexity/examples/peachstate_medicaid.py

@@ -0,0 +1,201 @@
+from optexity.schema.automation import Automation
+
+description = "Peach State Medicaid Insurance Example"
+endpoint_name = "peachstate_medicaid_insurance"
+automation_json = {
+    "url": "https://sso.entrykeyid.com/as/authorization.oauth2?response_type=code&client_id=f6a6219c-be42-421b-b86c-e4fc509e2e87&scope=openid%20profile&state=_igWklSsnrkO5DQfjBMMuN41ksMJePZQ_SM_61wTJlA%3D&redirect_uri=https://provider.pshpgeorgia.com/careconnect/login/oauth2/code/pingcloud&code_challenge_method=S256&nonce=xG41TJjco_x7Vs_MQgcS3bw5njLiJsXCqvO-V8THmY0&code_challenge=ZTaVHaZCNFTejXNJo51RlJ3Kv9dH0tMODPTqO7hiP3A&app_origin=https://provider.pshpgeorgia.com/careconnect/login/oauth2/code/pingcloud&brand=pshpgeorgia",
+    "parameters": {
+        "input_parameters": {
+            "username": [],
+            "password": [],
+            "plan_type": [],
+            "member_id": [],
+            "dob": [],
+        },
+        "generated_parameters": {},
+    },
+    "nodes": [
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_test_id("text-field")',
+                    "prompt_instructions": "Enter the email in the text field",
+                    "input_text": "{username[0]}",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="Continue")',
+                    "prompt_instructions": "Click the Continue button",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_role("textbox", name="Password")',
+                    "prompt_instructions": "Enter the password",
+                    "input_text": "{password[0]}",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="Login")',
+                    "prompt_instructions": "Click the Login button",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "select_option": {
+                    "command": 'get_by_label("Plan Type")',
+                    "prompt_instructions": "Select the Plan Type 8774789",
+                    "select_values": ["{plan_type[0]}"],
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("button", name="GO")',
+                    "prompt_instructions": "Click the GO button",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "input_text": {
+                    "command": 'get_by_test_id("MemberIDOrLastName")',
+                    "prompt_instructions": "Enter the Member ID or Last Name",
+                    "input_text": "{member_id[0]}",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "input_text": {
+                    "command": 'locator("#tDatePicker")',
+                    "prompt_instructions": "Enter the Date of Birth",
+                    "input_text": "{dob[0]}",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_role("combobox", name="Select Action Type Select")',
+                    "prompt_instructions": "Click the Select Action Type Select combobox",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_test_id("ActionType-option-0")',
+                    "prompt_instructions": "Click the View eligibility & patient info option",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_test_id("submitBtn")',
+                    "prompt_instructions": "Click the Submit button",
+                }
+            },
+            "end_sleep_time": 1.0,
+            "expect_new_tab": True,
+            "max_new_tab_wait_time": 10.0,
+        },
+        {
+            "type": "action_node",
+            "interaction_action": {
+                "click_element": {
+                    "command": 'get_by_label("Eligibility", exact=True).get_by_role("link", name="Authorizations")',
+                    "prompt_instructions": "Click the Authorizations link",
+                }
+            },
+            "end_sleep_time": 1.0,
+        },
+        {
+            "type": "action_node",
+            "extraction_action": {
+                "llm": {
+                    "extraction_format": {"authorization_numbers": "List[str]"},
+                    "extraction_instructions": "I am giving you an axtree of a webpage that shows the information about authorizations in a tabular format. Status, Auth Nbr, From Date, To Date, Diagnosis, Auth Type, Service. You need to output me a list of all Auth Nbr. Do not output any other information.",
+                    "output_variable_names": ["authorization_numbers"],
+                }
+            },
+            "before_sleep_time": 3.0,
+            "end_sleep_time": 0.0,
+        },
+        {
+            "type": "for_loop_node",
+            "variable_name": "authorization_numbers",
+            "nodes": [
+                {
+                    "type": "action_node",
+                    "interaction_action": {
+                        "click_element": {
+                            "command": 'get_by_role("link", name="{authorization_numbers[index]}")',
+                            "prompt_instructions": "Click the Authorizations link for the authorization number {authorization_numbers[index]}",
+                        }
+                    },
+                    "end_sleep_time": 1.0,
+                },
+                {
+                    "type": "action_node",
+                    "extraction_action": {
+                        "llm": {
+                            "extraction_format": {
+                                "Auth Nbr": "str",
+                                "End Date": "str",
+                                "Auth Type": "str",
+                                "Start Date": "str",
+                                "Auth Status": "str",
+                                "Service Type": "str",
+                                "Units Approved": "str",
+                                "Units Required": "str",
+                            },
+                            "extraction_instructions": "I am giving you an axtree of a webpage that shows information about authorizations, and I want the 8 following fields. 'Auth Status', 'Auth Nbr', 'Auth Type', 'Service Type', 'Start Date', 'End Date', 'Units Required', 'Units Approved'. Fields 'Auth Status', 'Auth Nbr', 'Auth Type' can be found in the top and rest of the information can be found in the tabular format. You need to output me key-value pairs for all 8 fields.",
+                        }
+                    },
+                    "before_sleep_time": 3.0,
+                    "end_sleep_time": 0.0,
+                },
+                {
+                    "type": "action_node",
+                    "interaction_action": {"go_back": {}},
+                    "end_sleep_time": 1.0,
+                },
+            ],
+        },
+    ],
+}
+
+automation = Automation.model_validate(automation_json)