diff --git a/agentic_security/probe_actor/operator.py b/agentic_security/probe_actor/operator.py new file mode 100644 index 0000000..5b8d31a --- /dev/null +++ b/agentic_security/probe_actor/operator.py @@ -0,0 +1,30 @@ +class OperatorToolBox: + def __init__(self, llm_spec, datasets): + self.llm_spec = llm_spec + self.datasets = datasets + + def get_spec(self): + return self.llm_spec + + def get_datasets(self): + return self.datasets + + def validate(self): + # Validate the tool box + pass + + def stop(self): + # Stop the tool box + pass + + def run(self): + # Run the tool box + pass + + def get_results(self): + # Get the results + pass + + def get_failures(self): + # Handle failure + pass diff --git a/agentic_security/refusal_classifier/oneclass_svm_model.joblib b/agentic_security/refusal_classifier/oneclass_svm_model.joblib index 1ef700f..bc4b97e 100644 Binary files a/agentic_security/refusal_classifier/oneclass_svm_model.joblib and b/agentic_security/refusal_classifier/oneclass_svm_model.joblib differ diff --git a/agentic_security/refusal_classifier/scaler.joblib b/agentic_security/refusal_classifier/scaler.joblib index b906c2b..05313e2 100644 Binary files a/agentic_security/refusal_classifier/scaler.joblib and b/agentic_security/refusal_classifier/scaler.joblib differ diff --git a/agentic_security/refusal_classifier/tfidf_vectorizer.joblib b/agentic_security/refusal_classifier/tfidf_vectorizer.joblib index 7c654ca..920a178 100644 Binary files a/agentic_security/refusal_classifier/tfidf_vectorizer.joblib and b/agentic_security/refusal_classifier/tfidf_vectorizer.joblib differ