address comments

Author: miguelg719

.github/workflows/publish.yml

@@ -35,7 +35,7 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install --upgrade pip
-          pip install build twine wheel setuptools ruff black tomllib
+          pip install build twine wheel setuptools ruff black
           pip install -r requirements.txt
           if [ -f requirements-dev.txt ]; then pip install -r requirements-dev.txt; fi
           
@@ -52,12 +52,60 @@ jobs:
       #   run: |
       #     pytest
 
-      - name: Get project version
-        id: get_version
+      - name: Calculate new version
+        id: version
         run: |
-          VERSION=$(python -c "import tomli; print(tomli.load(open('pyproject.toml', 'rb'))['project']['version'])")
-          echo "version=$VERSION" >> $GITHUB_OUTPUT
-
+          # Get current version from pyproject.toml
+          CURRENT_VERSION=$(python -c "import tomllib; print(tomllib.load(open('pyproject.toml', 'rb'))['project']['version'])")
+          echo "current_version=$CURRENT_VERSION" >> $GITHUB_OUTPUT
+          
+          # Parse version components
+          IFS='.' read -r MAJOR MINOR PATCH <<< "$CURRENT_VERSION"
+          
+          # Calculate new version based on release type
+          case "${{ github.event.inputs.release_type }}" in
+            "major")
+              NEW_MAJOR=$((MAJOR + 1))
+              NEW_MINOR=0
+              NEW_PATCH=0
+              ;;
+            "minor")
+              NEW_MAJOR=$MAJOR
+              NEW_MINOR=$((MINOR + 1))
+              NEW_PATCH=0
+              ;;
+            "patch")
+              NEW_MAJOR=$MAJOR
+              NEW_MINOR=$MINOR
+              NEW_PATCH=$((PATCH + 1))
+              ;;
+          esac
+          
+          NEW_VERSION="${NEW_MAJOR}.${NEW_MINOR}.${NEW_PATCH}"
+          echo "new_version=$NEW_VERSION" >> $GITHUB_OUTPUT
+          echo "Bumping version from $CURRENT_VERSION to $NEW_VERSION"
+          
+      - name: Update version files
+        run: |
+          CURRENT_VERSION="${{ steps.version.outputs.current_version }}"
+          NEW_VERSION="${{ steps.version.outputs.new_version }}"
+          
+          # Update pyproject.toml
+          sed -i "s/version = \"$CURRENT_VERSION\"/version = \"$NEW_VERSION\"/" pyproject.toml
+          
+          # Update __init__.py
+          sed -i "s/__version__ = \"$CURRENT_VERSION\"/__version__ = \"$NEW_VERSION\"/" stagehand/__init__.py
+          
+          echo "Updated version to $NEW_VERSION in pyproject.toml and __init__.py"
+          
+      - name: Commit version bump
+        run: |
+          git config --local user.email "action@github.com"
+          git config --local user.name "GitHub Action"
+          git add pyproject.toml stagehand/__init__.py
+          git commit -m "Bump version to ${{ steps.version.outputs.new_version }}"
+          git tag "v${{ steps.version.outputs.new_version }}"
+          
       - name: Build package
         run: |
           python -m build
@@ -69,10 +117,15 @@ jobs:
         run: |
           twine upload dist/*
           
+      - name: Push version bump
+        run: |
+          git push
+          git push --tags
+          
       - name: Create GitHub Release
         if: ${{ github.event.inputs.create_release == 'true' }}
         uses: softprops/action-gh-release@v1
         with:
-          tag_name: v${{ steps.get_version.outputs.version }}
-          name: Release v${{ steps.get_version.outputs.version }}
+          tag_name: v${{ steps.version.outputs.new_version }}
+          name: Release v${{ steps.version.outputs.new_version }}
           generate_release_notes: true 

stagehand/browser.py

@@ -64,8 +64,8 @@ async def connect_browserbase_browser(
             session = bb.sessions.create(**browserbase_session_create_params)
             if not session.id:
                 raise Exception("Could not create Browserbase session")
+            stagehand_instance.session_id = session.id
         connect_url = session.connectUrl
-        stagehand_instance.session_id = session.id
     except Exception as e:
         logger.error(f"Error retrieving or validating Browserbase session: {str(e)}")
         raise

stagehand/config.py

@@ -98,12 +98,12 @@ class StagehandConfig(BaseModel):
     )
     use_api: Optional[bool] = Field(
         True,
-        alias="useAPI",
+        alias=None,
         description="Whether to use the Stagehand API",
     )
     experimental: Optional[bool] = Field(
         False,
-        alias="experimental",
+        alias=None,
         description="Whether to use experimental features",
     )