refactor: install cognee defined baml version for CI/CD

.github/actions/cognee_setup/action.yml

@@ -24,7 +24,7 @@ runs:
       uses: astral-sh/setup-uv@v4
       with:
         enable-cache: true
-    
+
     - name: Rebuild uv lockfile
       shell: bash
       run: |
@@ -41,4 +41,4 @@ runs:
             EXTRA_ARGS="$EXTRA_ARGS --extra $extra"
           done
         fi
-        uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra neo4j $EXTRA_ARGS
+        uv sync --extra api --extra docs --extra evals --extra gemini --extra codegraph --extra ollama --extra dev --extra baml --extra neo4j $EXTRA_ARGS

.github/workflows/basic_tests.yml

@@ -189,9 +189,6 @@ jobs:
         with:
           python-version: ${{ inputs.python-version }}
 
-      - name: Install BAML Dependencies
-        run: uv add "baml-py>=0.201.0,<0.202.0"
-
       - name: Run Simple Examples
         run: uv run python ./examples/python/simple_example.py