From da84406676eb4da431f30092ec124f637672f0a5 Mon Sep 17 00:00:00 2001 From: Diwank Tomer Date: Tue, 30 Jul 2024 11:42:51 -0400 Subject: [PATCH] fix(agents-api): Fix the settings class in session data Signed-off-by: Diwank Tomer --- agents-api/agents_api/common/protocol/sessions.py | 4 +++- agents-api/agents_api/routers/sessions/session.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/agents-api/agents_api/common/protocol/sessions.py b/agents-api/agents_api/common/protocol/sessions.py index 2859a4630..93c5f1ac7 100644 --- a/agents-api/agents_api/common/protocol/sessions.py +++ b/agents-api/agents_api/common/protocol/sessions.py @@ -10,11 +10,13 @@ from ...autogen.openapi_model import ( Agent, Entry, + GenerationPresetSettings, + OpenAISettings, Session, Settings, - GenerationPresetSettings, OpenAISettings, VLLMSettings, Tool, User, + VLLMSettings, ) from .agents import AgentDefaultSettings diff --git a/agents-api/agents_api/routers/sessions/session.py b/agents-api/agents_api/routers/sessions/session.py index ddd78b315..a9276a79e 100644 --- a/agents-api/agents_api/routers/sessions/session.py +++ b/agents-api/agents_api/routers/sessions/session.py @@ -37,8 +37,8 @@ validate_and_extract_tool_calls, ) from ...models.entry.create_entries import create_entries -from ...models.session.prepare_chat_context import prepare_chat_context from ...models.session.get_cached_response import get_cached_response +from ...models.session.prepare_chat_context import prepare_chat_context from ...models.session.prepare_session_data import prepare_session_data from ...models.session.set_cached_response import set_cached_response from .exceptions import InputTooBigError