refactor: Lint agents-api (CI)

julep-ai · May 1, 2024 · 0b77d71 · 0b77d71
1 parent 39b118b
commit 0b77d71
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git a/agents-api/agents_api/model_registry.py b/agents-api/agents_api/model_registry.py
@@ -8,10 +8,8 @@
     AgentModelNotValid,
     MissingAgentModelAPIKeyError,
 )
-from openai import AsyncOpenAI
 import litellm
 from litellm.utils import get_valid_models
-from agents_api.env import model_inference_url, model_api_key
 
 GPT4_MODELS: Dict[str, int] = {
     # stable model names:
@@ -117,7 +115,6 @@ def validate_configuration(model: str):
         raise MissingAgentModelAPIKeyError(model)
 
 
-
 def load_context(init_context: list[ChatML], model: str):
     """
     Converts the message history into a format supported by the model.

diff --git a/agents-api/agents_api/routers/sessions/session.py b/agents-api/agents_api/routers/sessions/session.py
@@ -351,7 +351,7 @@ async def generate(
             response_format=settings.response_format,
             api_base=api_base,
             api_key=api_key,
-            **extra_body
+            **extra_body,
         )
         return res