From b6111ad8d514b3539ca4328cfcfa02bd2acfcce1 Mon Sep 17 00:00:00 2001
From: Ardian <ardianabazi09@gmail.com>
Date: Tue, 12 Sep 2023 14:32:04 +0200
Subject: [PATCH] feat: update tools

---
 tools/prediction_request_sme.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tools/prediction_request_sme.py b/tools/prediction_request_sme.py
index 4e698da1..4efdddc9 100644
--- 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 = """