Tuning warning down to info in BaseMessage (#136)

* switching warning to info

* making warning optionnal

Author: TLSDC

src/agentlab/agents/dynamic_prompting.py

@@ -260,7 +260,7 @@ def fit_tokens(
             )
             prompt_str = "\n".join([p["text"] for p in prompt if p["type"] == "text"])
         elif isinstance(prompt, BaseMessage):
-            prompt_str = str(prompt)
+            prompt_str = prompt.__str__(warn_if_image=False)
         else:
             raise ValueError(f"Unrecognized type for prompt: {type(prompt)}")
         n_token = count_tokens(prompt_str, model=model_name)

src/agentlab/llm/llm_utils.py

@@ -328,13 +328,16 @@ def __init__(self, role: str, content: Union[str, list[dict]]):
         self["role"] = role
         self["content"] = deepcopy(content)
 
-    def __str__(self) -> str:
+    def __str__(self, warn_if_image=False) -> str:
         if isinstance(self["content"], str):
             return self["content"]
         if not all(elem["type"] == "text" for elem in self["content"]):
-            logging.warning(
-                "The content of the message has images, which are not displayed in the string representation."
-            )
+            msg = "The content of the message has images, which are not displayed in the string representation."
+            if warn_if_image:
+                logging.warning(msg)
+            else:
+                logging.info(msg)
+
         return "\n".join([elem["text"] for elem in self["content"] if elem["type"] == "text"])
 
     def add_content(self, type: str, content: Any):