import cleanup

libs/ollama_functions.py

@@ -22,7 +22,7 @@ from langchain_core.outputs import ChatGeneration, ChatResult
 from langchain_core.prompts import SystemMessagePromptTemplate
 from langchain_core.pydantic_v1 import BaseModel 
 from langchain_core.runnables import Runnable
-from langchain_core.tools import BaseTool, Tool
+from langchain_core.tools import BaseTool 
 from langchain_core.utils.pydantic import is_basemodel_instance, is_basemodel_subclass
 
 from libs.functions import nxhash

libs/run_tests.py

@@ -1,4 +1,3 @@
-from os import name
 from libs.classes import Technique, Test, Model 
 from libs.functions import nxhash
 from typing import Union

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(

libs/tools.py

@@ -1,11 +1,9 @@
 from langchain.tools import tool
-from datetime import datetime, timedelta
+from datetime import datetime
 from re import search
 from dataclasses import dataclass
 from typing import Union
-import logging
 
-from langchain_core.tools import Tool
 from langchain_experimental.utilities import PythonREPL
 
 @tool