fix: fix DB session management to avoid connection overflow error (#1… #3796
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Run All pytest Tests | |
| env: | |
| MEMGPT_PGURI: ${{ secrets.MEMGPT_PGURI }} | |
| OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }} | |
| on: | |
| push: | |
| branches: [ main ] | |
| pull_request: | |
| branches: [ main ] | |
| jobs: | |
| test: | |
| runs-on: ubuntu-latest | |
| timeout-minutes: 15 | |
| services: | |
| qdrant: | |
| image: qdrant/qdrant | |
| ports: | |
| - 6333:6333 | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - name: Build and run container | |
| run: bash db/run_postgres.sh | |
| - name: "Setup Python, Poetry and Dependencies" | |
| uses: packetcoders/action-setup-cache-python-poetry@main | |
| with: | |
| python-version: "3.12" | |
| poetry-version: "1.8.2" | |
| install-args: "-E dev -E postgres -E milvus -E crewai-tools" | |
| - name: Initialize credentials | |
| run: poetry run memgpt quickstart --backend openai | |
| #- name: Run docker compose server | |
| # env: | |
| # MEMGPT_PG_DB: memgpt | |
| # MEMGPT_PG_USER: memgpt | |
| # MEMGPT_PG_PASSWORD: memgpt | |
| # MEMGPT_SERVER_PASS: test_server_token | |
| # MEMGPT_CONFIG_PATH: configs/server_config.yaml | |
| # run: docker compose up -d | |
| # | |
| - name: Run tools tests | |
| env: | |
| MEMGPT_PG_PORT: 8888 | |
| MEMGPT_PG_USER: memgpt | |
| MEMGPT_PG_PASSWORD: memgpt | |
| MEMGPT_PG_DB: memgpt | |
| MEMGPT_PG_HOST: localhost | |
| MEMGPT_SERVER_PASS: test_server_token | |
| run: | | |
| poetry run pytest -s -vv tests/test_tools.py | |
| - name: Run server tests | |
| env: | |
| MEMGPT_PG_PORT: 8888 | |
| MEMGPT_PG_USER: memgpt | |
| MEMGPT_PG_PASSWORD: memgpt | |
| MEMGPT_PG_DB: memgpt | |
| MEMGPT_PG_HOST: localhost | |
| MEMGPT_SERVER_PASS: test_server_token | |
| run: | | |
| poetry run pytest -s -vv tests/test_server.py | |
| - name: Run tests with pytest | |
| env: | |
| MEMGPT_PG_PORT: 8888 | |
| MEMGPT_PG_USER: memgpt | |
| MEMGPT_PG_PASSWORD: memgpt | |
| MEMGPT_PG_HOST: localhost | |
| MEMGPT_PG_DB: memgpt | |
| MEMGPT_SERVER_PASS: test_server_token | |
| PYTHONPATH: ${{ github.workspace }}:${{ env.PYTHONPATH }} | |
| run: | | |
| poetry run pytest -s -vv -k "not test_tools.py and not test_concurrent_connections.py and not test_quickstart and not test_endpoints and not test_storage and not test_server and not test_openai_client" tests |