add fix for self-hosted HF models

Author: jardinetsouffleton

src/agentlab/llm/huggingface_utils.py

@@ -7,6 +7,7 @@
 
 from agentlab.llm.base_api import AbstractChatModel
 from agentlab.llm.prompt_templates import PromptTemplate, get_prompt_template
+from agentlab.llm.llm_utils import Discussion
 
 
 class HFBaseChatModel(AbstractChatModel):
@@ -59,6 +60,7 @@ def __call__(
         if self.tokenizer:
             # messages_formated = _convert_messages_to_dict(messages) ## ?
             try:
+                messages.merge()
                 prompt = self.tokenizer.apply_chat_template(messages, tokenize=False)
             except Exception as e:
                 if "Conversation roles must alternate" in str(e):

src/agentlab/llm/llm_utils.py

@@ -386,6 +386,8 @@ def merge(self):
             else:
                 new_content.append(elem)
         self["content"] = new_content
+        if len(self["content"]) == 1:
+            self["content"] = self["content"][0]["text"]
 
 
 class SystemMessage(BaseMessage):