cleanup

src/leapfrogai/cli.py

@@ -2,7 +2,6 @@
 import sys
 import asyncio
 
-from leapfrogai.errors import AppImportError
 from leapfrogai.utils import import_app
 from leapfrogai.serve import serve
 

src/leapfrogai/errors.py

@@ -1,3 +1,4 @@
-# Adapted from Gunicorn's errors module
+# Adapted from Gunicorn's errors module.
+
 class AppImportError(Exception):
     """ Exception raised when loading an application """

src/leapfrogai/llm.py

@@ -1,4 +1,3 @@
-import asyncio
 from typing import Any, Generator, List
 
 from pydantic import BaseModel
@@ -14,7 +13,6 @@
     CompletionRequest,
     CompletionResponse,
     GrpcContext,
-    serve,
 )
 
 

src/leapfrogai/utils.py

@@ -1,11 +1,12 @@
+# Adapted from Gunicorn's utils module.
+
 import sys
 import ast
 import importlib
 import logging
 import traceback
 from leapfrogai.errors import AppImportError
 
-# Adapted from Gunicorn's util module 
 def import_app(module):
     parts = module.split(":", 1)
     if len(parts) == 1: