Skip to content

Commit

Permalink
wip add cli log10 feedback or log10 feedback-task create
Browse files Browse the repository at this point in the history
  • Loading branch information
wenzhe-log10 committed Feb 9, 2024
1 parent 6cb9a5c commit df7a078
Show file tree
Hide file tree
Showing 4 changed files with 53 additions and 12 deletions.
32 changes: 32 additions & 0 deletions log10/__main__.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,32 @@
import click

from log10.feedback.feedback import create_feedback
from log10.feedback.feedback_task import create_feedback_task


@click.group()
def cli():
pass


@click.group()
def feedback():
pass


@click.group()
def feedback_task():
pass


cli.add_command(feedback)
feedback.add_command(create_feedback, "create")
cli.add_command(feedback_task)
feedback_task.add_command(create_feedback_task, "create")

if __name__ == "__main__":
cli()


# cli.add_command(feedback.create_feedback)
# cli.add_command(feedback_task.create_feedback_task)
15 changes: 9 additions & 6 deletions log10/feedback/feedback.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import click
import logging

import click
import httpx
from dotenv import load_dotenv

Expand Down Expand Up @@ -57,9 +57,12 @@ def create(self, task_id: str, rate: dict) -> httpx.Response:
return res

@click.command()
@click.option("--task_id", help="Task ID")
@click.option("--rate", help="Rate in JSON format")
@click.option("--task_id", prompt="Enter task id", help="Task ID")
@click.option("--rate", prompt="Enter task rate", help="Rate in JSON format")
def create_feedback(task_id, rate):
fb = Feedback()
feedback = fb.create(task_id=task_id, rate=rate)
print(feedback)
click.echo("Creating feedback")
click.echo(f"Task ID: {task_id}")
click.echo(f"Rate: {rate}")
# fb = Feedback()
# feedback = fb.create(task_id=task_id, rate=rate)
# click.echo(feedback)
15 changes: 9 additions & 6 deletions log10/feedback/feedback_task.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
import click
import logging

import click
import httpx
from dotenv import load_dotenv

Expand Down Expand Up @@ -47,9 +47,12 @@ def create(self, name: str, task_schema: dict) -> httpx.Response:

# create a cli interface for FeebackTask.create function
@click.command()
@click.option("--name", help="Name of the task")
@click.option("--task_schema", help="Task schema")
@click.option("--name", prompt="Enter feedback task name", help="Name of the task")
@click.option("--task_schema", prompt="Enter feedback task schema", help="Task schema")
def create_feedback_task(name, task_schema):
fb_task = FeedbackTask()
task = fb_task.create(name=name, task_schema=task_schema)
print(task)
click.echo("Creating feedback task")
click.echo(f"Name: {name}")
click.echo(f"Task Schema: {task_schema}")
# fb_task = FeedbackTask()
# task = fb_task.create(name=name, task_schema=task_schema)
# click.echo(task)
3 changes: 3 additions & 0 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,9 @@ packages = [
{ include = "log10" },
]

[tool.poetry.scripts]
log10 = "log10.__main__:cli"

[tool.poetry.group.dev.dependencies]
build = "^0.10.0"
pytest = "^8.0.0"
Expand Down

0 comments on commit df7a078

Please sign in to comment.