updates

Author: filip-michalsky

.github/workflows/test.yml

@@ -115,10 +115,6 @@ jobs:
     name: API Integration Tests
     runs-on: ubuntu-latest
     needs: test-unit
-    if: |
-      github.event_name == 'schedule' || 
-      contains(github.event.pull_request.labels.*.name, 'test-api') ||
-      contains(github.event.pull_request.labels.*.name, 'api')
 
     steps:
     - uses: actions/checkout@v4

tests/end_to_end/test_act_integration.py renamed to tests/e2e/test_act_integration.py

@@ -2,8 +2,10 @@
 
 import pytest
 import pytest_asyncio
+from pydantic import BaseModel, Field
 
 from stagehand import Stagehand, StagehandConfig
+from stagehand.schemas import ExtractOptions
 
 
 skip_if_no_creds = pytest.mark.skipif(
@@ -12,6 +14,12 @@
 )
 
 
+class Article(BaseModel):
+    """Schema for article extraction tests"""
+    title: str = Field(..., description="The title of the article")
+    summary: str = Field(None, description="A brief summary or description of the article")
+
+
 @pytest_asyncio.fixture(scope="module")
 @skip_if_no_creds
 async def stagehand_api():
@@ -35,4 +43,48 @@ async def stagehand_api():
 @pytest.mark.asyncio
 async def test_stagehand_api_initialization(stagehand_api):
     """Ensure that Stagehand initializes correctly against the Browserbase API."""
-    assert stagehand_api.session_id is not None 
+    assert stagehand_api.session_id is not None
+
+
+@skip_if_no_creds
+@pytest.mark.integration
+@pytest.mark.api
+@pytest.mark.asyncio
+async def test_api_extract_functionality(stagehand_api):
+    """Test core extract functionality in API mode - extracted from e2e tests."""
+    stagehand = stagehand_api
+    
+    # Navigate to a content-rich page
+    await stagehand.page.goto("https://news.ycombinator.com")
+    
+    # Test simple text-based extraction
+    titles_text = await stagehand.page.extract(
+        "Extract the titles of the first 3 articles on the page as a JSON array"
+    )
+    
+    # Verify extraction worked
+    assert titles_text is not None
+    
+    # Test schema-based extraction
+    extract_options = ExtractOptions(
+        instruction="Extract the first article's title and any available summary",
+        schema_definition=Article
+    )
+    
+    article_data = await stagehand.page.extract(extract_options)
+    assert article_data is not None
+    
+    # Validate the extracted data structure (Browserbase format)
+    if hasattr(article_data, 'data') and article_data.data:
+        # BROWSERBASE mode format
+        article = Article.model_validate(article_data.data)
+        assert article.title
+        assert len(article.title) > 0
+    elif hasattr(article_data, 'title'):
+        # Fallback format
+        article = Article.model_validate(article_data.model_dump())
+        assert article.title
+        assert len(article.title) > 0
+    
+    # Verify API session is active
+    assert stagehand.session_id is not None 

tests/end_to_end/test_extract_integration.py renamed to tests/e2e/test_extract_integration.py

@@ -19,4 +19,42 @@ async def stagehand_local():
 @pytest.mark.asyncio
 async def test_stagehand_local_initialization(stagehand_local):
     """Ensure that Stagehand initializes correctly in LOCAL mode."""
-    assert stagehand_local._initialized is True 
+    assert stagehand_local._initialized is True
+
+
+@pytest.mark.integration
+@pytest.mark.local
+@pytest.mark.asyncio
+async def test_local_observe_and_act_workflow(stagehand_local):
+    """Test core observe and act workflow in LOCAL mode - extracted from e2e tests."""
+    stagehand = stagehand_local
+    
+    # Navigate to a form page for testing
+    await stagehand.page.goto("https://httpbin.org/forms/post")
+    
+    # Test OBSERVE primitive: Find form elements
+    form_elements = await stagehand.page.observe("Find all form input elements")
+    
+    # Verify observations
+    assert form_elements is not None
+    assert len(form_elements) > 0
+    
+    # Verify observation structure
+    for obs in form_elements:
+        assert hasattr(obs, "selector")
+        assert obs.selector  # Not empty
+    
+    # Test ACT primitive: Fill form fields
+    await stagehand.page.act("Fill the customer name field with 'Local Integration Test'")
+    await stagehand.page.act("Fill the telephone field with '555-LOCAL'")
+    await stagehand.page.act("Fill the email field with '[email protected]'")
+    
+    # Verify actions worked by observing filled fields
+    filled_fields = await stagehand.page.observe("Find all filled form input fields")
+    assert filled_fields is not None
+    assert len(filled_fields) > 0
+    
+    # Test interaction with specific elements
+    customer_field = await stagehand.page.observe("Find the customer name input field")
+    assert customer_field is not None
+    assert len(customer_field) > 0