fix(agents-api): Fix the settings class in session data

Signed-off-by: Diwank Tomer <[email protected]>
julep-ai · Jul 30, 2024 · da84406 · da84406
1 parent ae2a544
commit da84406
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -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