minor test fix

Author: DinisCruz

osbot_llms/llms/prompt_to_json/Prompt_To_Json__Open_AI.py

@@ -1,11 +1,10 @@
-from typing import List, Dict
-from pydantic._internal._model_construction import ModelMetaclass
-
-from osbot_llms.llms.API_Open_AI import API_Open_AI
-from osbot_utils.base_classes.Type_Safe import Type_Safe
-from osbot_utils.context_managers.capture_duration import capture_duration
-from osbot_utils.helpers.Timestamp_Now import Timestamp_Now
-from osbot_utils.utils.Json import str_to_json
+from typing                                         import List, Dict
+from pydantic._internal._model_construction         import ModelMetaclass
+from osbot_llms.llms.API_Open_AI                    import API_Open_AI
+from osbot_utils.base_classes.Type_Safe             import Type_Safe
+from osbot_utils.context_managers.capture_duration  import capture_duration
+from osbot_utils.helpers.Timestamp_Now              import Timestamp_Now
+from osbot_utils.utils.Json                         import str_to_json
 
 
 class Prompt_To_Json__Open_AI(Type_Safe):

tests/integration/llms/providers/open_router/test_LLM_Open_Router.py

@@ -28,7 +28,7 @@ def test__init__(self):
 
     def test_send_user_prompt(self):
         user_prompt = "2+2"
-        self.llm_open_router.add_message__system('reply in one word only')
+        self.llm_open_router.add_message__system('reply in one number only')
         response = self.llm_open_router.send_user_prompt(user_prompt=user_prompt)
         if response.get('status') == 'error':
             raise Exception(f"Error: {pformat(response)}")