add prompt in chat function

Author: renansantosmendes

app/agent/agent.py

@@ -29,6 +29,10 @@ def get_llm() -> ChatOpenAI:
     )
 
 
+def _get_system_message() -> SystemMessage:
+    return SystemMessage(content=langfuse.get_prompt("neuralearn_ai_system_prompt").prompt)
+
+
 def _parse_messages(messages: list[dict]) -> list[BaseMessage]:
     parsed = []
     for message in messages:
@@ -84,7 +88,7 @@ def ask_agent(question: str) -> str:
     logger.info("[%s] Pergunta recebida: %s", trace_id, question)
 
     messages = [
-        SystemMessage(content=langfuse.get_prompt("neuralearn_ai_system_prompt").prompt),
+        _get_system_message(),
         HumanMessage(content=question),
     ]
 
@@ -96,6 +100,6 @@ def ask_agent_with_messages(messages: list[dict]) -> str:
     trace_id = str(uuid.uuid4())
     logger.info("[%s] Lista de mensagens recebida (total=%d)", trace_id, len(messages))
 
-    parsed_messages = _parse_messages(messages)
+    parsed_messages = [_get_system_message(), *_parse_messages(messages)]
     response = _invoke_llm(parsed_messages, trace_id)
     return response.content

tests/test_agent.py

@@ -63,10 +63,12 @@ def test_ask_agent_with_messages_returns_llm_content(mocker):
     fake_llm.invoke.return_value = fake_response
 
     mocker.patch.object(agent, "get_llm", return_value=fake_llm)
+    mocker.patch.object(
+        agent.langfuse, "get_prompt", return_value=MagicMock(prompt="prompt do sistema")
+    )
     mocker.patch.object(agent, "get_langfuse_handler", return_value=None)
 
     messages = [
-        {"role": "system", "content": "voce e um assistente"},
         {"role": "user", "content": "oi"},
         {"role": "assistant", "content": "ola, como posso ajudar?"},
         {"role": "user", "content": "qual a capital do Brasil?"},
@@ -76,21 +78,30 @@ def test_ask_agent_with_messages_returns_llm_content(mocker):
 
     assert result == "resposta do llm"
     sent_messages = fake_llm.invoke.call_args.args[0]
-    assert [m.content for m in sent_messages] == [m["content"] for m in messages]
     assert sent_messages[0].type == "system"
+    assert sent_messages[0].content == "prompt do sistema"
+    assert [m.content for m in sent_messages[1:]] == [m["content"] for m in messages]
     assert sent_messages[1].type == "human"
     assert sent_messages[2].type == "ai"
     assert sent_messages[3].type == "human"
 
 
-def test_ask_agent_with_messages_rejects_invalid_role():
+def test_ask_agent_with_messages_rejects_invalid_role(mocker):
+    mocker.patch.object(
+        agent.langfuse, "get_prompt", return_value=MagicMock(prompt="prompt do sistema")
+    )
+
     messages = [{"role": "foo", "content": "oi"}]
 
     with pytest.raises(ValueError):
         agent.ask_agent_with_messages(messages)
 
 
-def test_ask_agent_with_messages_requires_content():
+def test_ask_agent_with_messages_requires_content(mocker):
+    mocker.patch.object(
+        agent.langfuse, "get_prompt", return_value=MagicMock(prompt="prompt do sistema")
+    )
+
     messages = [{"role": "user"}]
 
     with pytest.raises(ValueError):