import cleanup

libs/runnables.py

@@ -1,17 +1,18 @@
+import json
 from types import NoneType
-from langchain_ollama.chat_models import ChatOllama
-from libs.ollama_functions import OllamaFunctionsLSM, OllamaFunctionsT2S
-from langchain_core.messages import AIMessage, SystemMessage, HumanMessage, ToolMessage
-from libs.classes import Technique, Test, Model
-from langchain.tools import Tool
 from typing import Literal
 
-from langgraph.graph import StateGraph, MessagesState
-import json
+from langchain.tools import Tool
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage, ToolMessage
+from langchain_ollama.chat_models import ChatOllama
+from langgraph.graph import MessagesState, StateGraph
 from pydantic import ValidationError
 
+from libs.classes import Model, Technique, Test
+from libs.ollama_functions import OllamaFunctionsLSM, OllamaFunctionsT2S
 from suite_settings.techniques import techniques
 
+
 def _get_llm(model: Model, base_url: str, seed: int, technique: Technique, tools: list[Tool]|NoneType = None):
     if technique == techniques[1]: # Native
         llm = ChatOllama(