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 +0 -1
- test_maritaca_endpoints.sh +59 -0
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 ==="
|