diff --git a/.github/workflows/sync.yml b/.github/workflows/sync.yml index 5c03db36..63386563 100644 --- a/.github/workflows/sync.yml +++ b/.github/workflows/sync.yml @@ -9,30 +9,22 @@ jobs: runs-on: ubuntu-latest steps: - uses: actions/checkout@v3 - - name: Restore cache - id: file-restore - uses: actions/cache/restore@v3 - with: - path: | - sync/endpoint.txt - key: endpoint-${{ github.run_id }} - restore-keys: | - endpoint- - uses: actions/setup-python@v4 with: python-version: '3.10' cache: 'pip' + - name: Download endpoint + uses: actions/download-artifact@v3 + with: + name: endpoint - name: run + if: success() || failure() env: HF_TOKEN: ${{ secrets.HF_TOKEN }} ENDPOINTS: ${{ vars.ENDPOINTS }} - run: cd sync && cat endpoint.txt && pip install huggingface_hub && python sync.py && cat endpoint.txt - - name: Save cache - id: file-save - uses: actions/cache/save@v3 - with: - path: | - sync/endpoint.txt - key: endpoint-${{ github.run_id }} - restore-keys: | - endpoint- + run: cd sync && pip install huggingface_hub && python sync.py && cat endpoint.txt + - name: Upload endpoint + uses: actions/upload-artifact@v3 + with: + name: endpoint + path: sync/endpoint.txt diff --git a/sync/.gitignore b/sync/.gitignore index 406bab72..f10862a6 100644 --- a/sync/.gitignore +++ b/sync/.gitignore @@ -1,2 +1 @@ -/endpoint.txt /.env diff --git a/sync/sync.py b/sync/sync.py index c11ad81b..88a712d6 100644 --- a/sync/sync.py +++ b/sync/sync.py @@ -8,18 +8,19 @@ ENDPOINTS = os.environ.get('ENDPOINTS') file = 'endpoint.txt' endpoints = ENDPOINTS.split('\n') -endpoint = endpoints[0].strip() +endpointIndex = '0' try: with open(file,"r") as f: content=f.read().strip() - endpoint = endpoints[1] if content == endpoints[0] else endpoints[0] + print('content:', content) + endpointIndex = '1' if content == '0' else '0' f.close() except IOError: print("File not accessible") - with open(file, "w") as wf: - wf.write(endpoint) - wf.close() -print('new endpoint', endpoint) -hf_api.add_space_secret('hf4all/bingo2', 'ENDPOINT', endpoint.strip()) +with open(file, "w") as wf: + wf.write(endpointIndex) + wf.close() +print('new endpoint', endpoints[int(endpointIndex)]) +hf_api.add_space_secret('hf4all/bingo2', 'ENDPOINT', endpoints[int(endpointIndex)].strip())