fix: Lista de messages não estava retornando corretamente

O metodo .extend é inplace, como estava sendo utilizado em uma lista que não tinha sido criada, estava retornando null related #1
RWallan · May 23, 2024 · e628d1a · e628d1a
1 parent e9e6b40
commit e628d1a
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/openiziai/agents/agent.py b/openiziai/agents/agent.py
@@ -134,24 +134,19 @@ def prompt(
         Returns:
             PromptResponse: Informações do prompt construído.
         """
+        messages = [{'role': 'system', 'content': self._template}]
         if history:
             _history = [m.model_dump() for m in history]
-            messages = [{'role': 'system', 'content': self._template}].extend(
-                _history
-            )
+            messages.extend(_history)
             _prompt = _history[-1].get('content')
         elif prompt:
             _prompt = prompt
-            messages = [
-                {
-                    'role': 'system',
-                    'content': self._template,
-                },
+            messages.extend([
                 {
                     'role': 'user',
                     'content': _prompt,
                 },
-            ]
+            ])
         else:
             raise ValueError('`prompt` ou `history` precisam ser passados.')