From dbec27d3aa5efa74c7471d8d41bafe1bb650f7bc Mon Sep 17 00:00:00 2001 From: Alexander Myasoedov Date: Thu, 23 Jan 2025 00:01:54 +0200 Subject: [PATCH] feat(Update models): --- agentic_security/probe_actor/operator.py | 30 ++++++++++++++++++ .../oneclass_svm_model.joblib | Bin 269967 -> 269967 bytes .../refusal_classifier/scaler.joblib | Bin 24583 -> 24583 bytes .../tfidf_vectorizer.joblib | Bin 37621 -> 37621 bytes 4 files changed, 30 insertions(+) create mode 100644 agentic_security/probe_actor/operator.py 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 1ef700f067563b691ea5883e131b9a8fc150cc21..bc4b97ede85df70265755cdc92f904dc9470ae49 100644 GIT binary patch delta 25 gcmeC*E6~4JprM7ag{g&k3rmL_yP2Nhl+q+U0C+D5UH||9 delta 25 gcmeC*E6~4JprM7ag{g&k3rmL_yQ!Yhl+q+U0C+73UH||9 diff --git a/agentic_security/refusal_classifier/scaler.joblib b/agentic_security/refusal_classifier/scaler.joblib index b906c2b8f2a2818925d0c3c2c067b4f94d6fa4bd..05313e2b6699f3512c9db01ab2b188df2174e67b 100644 GIT binary patch delta 16 XcmZoaz}S9(aX|vRnV#X4(j+|qHLwN7 delta 16 XcmZoaz}S9(aX|vRsh-i4(j+|qHLeB5 diff --git a/agentic_security/refusal_classifier/tfidf_vectorizer.joblib b/agentic_security/refusal_classifier/tfidf_vectorizer.joblib index 7c654cae945f3155c372817968f2559ead526c46..920a178319fe459baf693d4f54be2568ce22597a 100644 GIT binary patch delta 26 icmeyml