feat: update tools

valory-xyz · Sep 12, 2023 · b6111ad · b6111ad
1 parent f98590f
commit b6111ad
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tools/prediction_request_sme.py b/tools/prediction_request_sme.py
@@ -35,12 +35,12 @@
     "temperature": 0.7,
 }
 ALLOWED_TOOLS = [
-    "prediction-offline",
-    "prediction-online",
+    "prediction-offline-sme",
+    "prediction-online-sme",
 ]
 TOOL_TO_ENGINE = {
-    "prediction-offline": "gpt-3.5-turbo",
-    "prediction-online": "gpt-3.5-turbo",
+    "prediction-offline-sme": "gpt-3.5-turbo",
+    "prediction-online-sme": "gpt-3.5-turbo",
 }
 
 PREDICTION_PROMPT = """