anderson-ufrj commited on
Commit
d3a8f3c
·
1 Parent(s): f317ea3

fix: remove session_id parameter from ConversationalMemory.add_message

Browse files

- Fixed TypeError in Drummond agent when processing chat messages
- ConversationalMemory doesn't accept session_id parameter
- Allows chat to work properly on HuggingFace Spaces

src/agents/drummond.py CHANGED
@@ -489,7 +489,6 @@ LEMBRE: "No meio do caminho tinha uma pedra" - vá direto ao essencial."""
489
 
490
  # Atualizar contexto conversacional
491
  await self.conversational_memory.add_message(
492
- session_id=context.session_id,
493
  role="user",
494
  content=message
495
  )
 
489
 
490
  # Atualizar contexto conversacional
491
  await self.conversational_memory.add_message(
 
492
  role="user",
493
  content=message
494
  )
test_maritaca_endpoints.sh ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/bin/bash
2
+ # Script para testar os endpoints de chat com Maritaca AI
3
+
4
+ BASE_URL="https://neural-thinker-cidadao-ai-backend.hf.space"
5
+
6
+ echo "=== Testando Endpoints de Chat com Maritaca AI ==="
7
+ echo ""
8
+
9
+ # 1. Teste do endpoint /api/v1/chat/stable
10
+ echo "1. Testando /api/v1/chat/stable (recomendado):"
11
+ curl -X POST "$BASE_URL/api/v1/chat/stable" \
12
+ -H "Content-Type: application/json" \
13
+ -d '{
14
+ "message": "Olá, como você pode me ajudar com transparência pública?",
15
+ "session_id": "test-stable-123"
16
+ }' | jq .
17
+
18
+ echo ""
19
+ echo "----------------------------------------"
20
+ echo ""
21
+
22
+ # 2. Teste do endpoint /api/v1/chat/simple/maritaca
23
+ echo "2. Testando /api/v1/chat/simple/maritaca:"
24
+ curl -X POST "$BASE_URL/api/v1/chat/simple/maritaca" \
25
+ -H "Content-Type: application/json" \
26
+ -d '{
27
+ "message": "Quero investigar contratos da saúde",
28
+ "session_id": "test-simple-456"
29
+ }' | jq .
30
+
31
+ echo ""
32
+ echo "----------------------------------------"
33
+ echo ""
34
+
35
+ # 3. Teste do endpoint /api/v1/chat/optimized
36
+ echo "3. Testando /api/v1/chat/optimized (com Drummond):"
37
+ curl -X POST "$BASE_URL/api/v1/chat/optimized" \
38
+ -H "Content-Type: application/json" \
39
+ -d '{
40
+ "message": "Me explique o que é o Cidadão.AI",
41
+ "session_id": "test-optimized-789",
42
+ "use_drummond": true
43
+ }' | jq .
44
+
45
+ echo ""
46
+ echo "----------------------------------------"
47
+ echo ""
48
+
49
+ # 4. Teste do endpoint /api/v1/chat/emergency
50
+ echo "4. Testando /api/v1/chat/emergency (ultra resiliente):"
51
+ curl -X POST "$BASE_URL/api/v1/chat/emergency" \
52
+ -H "Content-Type: application/json" \
53
+ -d '{
54
+ "message": "Como posso analisar gastos públicos?",
55
+ "session_id": "test-emergency-999"
56
+ }' | jq .
57
+
58
+ echo ""
59
+ echo "=== Fim dos Testes ==="