fix: additional information check

valory-xyz · Sep 12, 2023 · a6ae0f5 · a6ae0f5
1 parent b6111ad
commit a6ae0f5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/prediction_request_sme.py b/tools/prediction_request_sme.py
@@ -327,7 +327,7 @@ def run(**kwargs) -> Tuple[str, Optional[Dict[str, Any]]]:
             google_api_key=kwargs["api_keys"]["google_api_key"],
             google_engine=kwargs["api_keys"]["google_engine_id"],
         )
-        if tool == "prediction-online"
+        if tool == "prediction-online-sme"
         else ""
     )
     prediction_prompt = PREDICTION_PROMPT.format(
@@ -351,3 +351,5 @@ def run(**kwargs) -> Tuple[str, Optional[Dict[str, Any]]]:
         stop=None,
     )
     return response.choices[0].message.content, None
+
+