wassname/Open-Assistant
1
0
Fork 0
mirror of https://github.com/wassname/Open-Assistant.git synced 2026-06-27 16:10:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

improvements to the text frontend

Browse Source
This commit is contained in:
Yannic Kilcher
2022-12-16 22:00:44 +01:00
parent 6f37a3120a
commit 2c3e854bf7
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
text-frontend/__main__.py
+3 -3
View File
@@ -25,7 +25,7 @@ def _render_message(message: dict) -> str:
@app.command()
def main(backend_url: str, api_key: str):
def main(backend_url: str = "http://127.0.0.1:8000", api_key: str = "DUMMY_KEY"):
"""Simple REPL frontend."""
def _post(path: str, json: dict) -> dict:
@@ -164,7 +164,7 @@ def main(backend_url: str, api_key: str):
_post(f"/api/v1/tasks/{task['id']}/ack", {"post_id": post_id})
ranking_str = typer.prompt("Enter the prompt numbers in order of preference, separated by commas")
ranking = [int(x) for x in ranking_str.split(",")]
ranking = [int(x) - 1 for x in ranking_str.split(",")]
# send ranking
new_task = _post(
@@ -190,7 +190,7 @@ def main(backend_url: str, api_key: str):
_post(f"/api/v1/tasks/{task['id']}/ack", {"post_id": post_id})
ranking_str = typer.prompt("Enter the reply numbers in order of preference, separated by commas")
ranking = [int(x) for x in ranking_str.split(",")]
ranking = [int(x) - 1 for x in ranking_str.split(",")]
# send ranking
new_task = _post(