From eb3a70b7e4e32d014f63b8dbe81bafa68c50a335 Mon Sep 17 00:00:00 2001
From: Alexander Myasoedov <msoedov@gmail.com>
Date: Tue, 28 Jan 2025 15:32:23 +0200
Subject: [PATCH] fix(pre commit):

---
 agentic_security/probe_actor/operator.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/agentic_security/probe_actor/operator.py b/agentic_security/probe_actor/operator.py
index 4271112..c7c0ca8 100644
--- a/agentic_security/probe_actor/operator.py
+++ b/agentic_security/probe_actor/operator.py
@@ -1,17 +1,19 @@
 import asyncio
 from typing import Any
 
+from pydantic import BaseModel, Field
 from pydantic_ai import Agent, RunContext
-from pydantic import  BaseModel, Field
-from typing import Optional, List, Dict, Any
 
 
 class AgentSpecification(BaseModel):
-    name: Optional[str] = Field(None, description="Name of the LLM/agent")
-    version: Optional[str] = Field(None, description="Version of the LLM/agent")
-    description: Optional[str] = Field(None, description="Description of the LLM/agent")
-    capabilities: Optional[List[str]] = Field(None, description="List of capabilities")
-    configuration: Optional[Dict[str, Any]] = Field(None, description="Configuration settings")
+    name: str | None = Field(None, description="Name of the LLM/agent")
+    version: str | None = Field(None, description="Version of the LLM/agent")
+    description: str | None = Field(None, description="Description of the LLM/agent")
+    capabilities: list[str] | None = Field(None, description="List of capabilities")
+    configuration: dict[str, Any] | None = Field(
+        None, description="Configuration settings"
+    )
+
 
 # Define the OperatorToolBox class
 class OperatorToolBox:
@@ -66,16 +68,14 @@ spec = AgentSpecification(
     version="4.0",
     description="A powerful language model",
     capabilities=["text-generation", "question-answering"],
-    configuration={"max_tokens": 100}
+    configuration={"max_tokens": 100},
 )
 
 # dataset_manager_agent.py
 
 
 # Initialize OperatorToolBox
-toolbox = OperatorToolBox(
-    spec=spec, datasets=["dataset1", "dataset2", "dataset3"]
-)
+toolbox = OperatorToolBox(spec=spec, datasets=["dataset1", "dataset2", "dataset3"])
 
 # Define the agent with OperatorToolBox as its dependency
 dataset_manager_agent = Agent(