Skip to content

Commit

Permalink
build: sync env
Browse files Browse the repository at this point in the history
  • Loading branch information
weaigc committed Sep 8, 2023
1 parent 2da30dd commit 2478d3c
Show file tree
Hide file tree
Showing 3 changed files with 19 additions and 27 deletions.
30 changes: 11 additions & 19 deletions .github/workflows/sync.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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
1 change: 0 additions & 1 deletion sync/.gitignore
Original file line number Diff line number Diff line change
@@ -1,2 +1 @@
/endpoint.txt
/.env
15 changes: 8 additions & 7 deletions sync/sync.py
Original file line number Diff line number Diff line change
Expand Up @@ -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())

0 comments on commit 2478d3c

Please sign in to comment.