From 0b77d71107369e26091071d81d22dbd993112639 Mon Sep 17 00:00:00 2001 From: alt-glitch Date: Wed, 1 May 2024 10:04:44 +0000 Subject: [PATCH] refactor: Lint agents-api (CI) --- agents-api/agents_api/model_registry.py | 3 --- agents-api/agents_api/routers/sessions/session.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/agents-api/agents_api/model_registry.py b/agents-api/agents_api/model_registry.py index 796ae45eb..e5a72c7fc 100644 --- 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 index 0c32ba607..e3381a5af 100644 --- 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