Skip to content

Commit

Permalink
Ruff linting
Browse files Browse the repository at this point in the history
  • Loading branch information
CollectiveUnicorn committed Jul 3, 2024
1 parent c9a2e81 commit 6b1ff89
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 22 deletions.
32 changes: 16 additions & 16 deletions src/leapfrogai_api/backend/types.py
Original file line number Diff line number Diff line change
Expand Up @@ -395,13 +395,13 @@ class CreateTranscriptionRequest(BaseModel):

@classmethod
def as_form(
cls,
file: UploadFile = File(...),
model: str = Form(...),
language: str | None = Form(""),
prompt: str | None = Form(""),
response_format: str | None = Form(""),
temperature: float | None = Form(1.0),
cls,
file: UploadFile = File(...),
model: str = Form(...),
language: str | None = Form(""),
prompt: str | None = Form(""),
response_format: str | None = Form(""),
temperature: float | None = Form(1.0),
) -> CreateTranscriptionRequest:
return cls(
file=file,
Expand Down Expand Up @@ -448,12 +448,12 @@ class CreateTranslationRequest(BaseModel):

@classmethod
def as_form(
cls,
file: UploadFile = File(...),
model: str = Form(...),
prompt: str | None = Form(""),
response_format: str | None = Form(""),
temperature: float | None = Form(1.0),
cls,
file: UploadFile = File(...),
model: str = Form(...),
prompt: str | None = Form(""),
response_format: str | None = Form(""),
temperature: float | None = Form(1.0),
) -> CreateTranslationRequest:
return cls(
file=file,
Expand Down Expand Up @@ -493,9 +493,9 @@ class UploadFileRequest(BaseModel):

@classmethod
def as_form(
cls,
file: UploadFile = File(...),
purpose: str | None = Form("assistants"),
cls,
file: UploadFile = File(...),
purpose: str | None = Form("assistants"),
) -> UploadFileRequest:
"""Create an instance of the class from form data."""
return cls(file=file, purpose=purpose)
Expand Down
12 changes: 6 additions & 6 deletions src/leapfrogai_api/routers/openai/audio.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,9 +21,9 @@

@router.post("/transcriptions")
async def transcribe(
session: Session,
model_config: Annotated[Config, Depends(get_model_config)],
req: CreateTranscriptionRequest = Depends(CreateTranscriptionRequest.as_form),
session: Session,
model_config: Annotated[Config, Depends(get_model_config)],
req: CreateTranscriptionRequest = Depends(CreateTranscriptionRequest.as_form),
) -> CreateTranscriptionResponse:
"""Create a transcription from the given audio file."""
model = model_config.get_model_backend(req.model)
Expand All @@ -50,9 +50,9 @@ async def transcribe(

@router.post("/translations")
async def translate(
session: Session,
model_config: Annotated[Config, Depends(get_model_config)],
req: CreateTranscriptionRequest = Depends(CreateTranscriptionRequest.as_form),
session: Session,
model_config: Annotated[Config, Depends(get_model_config)],
req: CreateTranscriptionRequest = Depends(CreateTranscriptionRequest.as_form),
) -> CreateTranscriptionResponse:
"""Create a transcription from the given audio file."""
model = model_config.get_model_backend(req.model)
Expand Down

0 comments on commit 6b1ff89

Please sign in to comment.