anderson-ufrj
commited on
Commit
·
7878f71
1
Parent(s):
0bfbae0
fix: add missing ConversationContext class and fix import path
Browse filesFixed deployment issues:
1. Corrected import path from conversational_memory to conversational
2. Added missing ConversationContext dataclass to conversational.py
The ConversationContext class stores session information for chat
conversations including session_id, user_id, user_profile and metadata.
This resolves the ModuleNotFoundError preventing HuggingFace deployment.
- src/agents/drummond.py +1 -1
- src/memory/conversational.py +14 -0
src/agents/drummond.py
CHANGED
|
@@ -22,7 +22,7 @@ from src.agents.deodoro import BaseAgent, AgentContext, AgentMessage, AgentRespo
|
|
| 22 |
from src.core import get_logger
|
| 23 |
from src.core.exceptions import AgentExecutionError, DataAnalysisError
|
| 24 |
from src.services.chat_service import IntentType, Intent
|
| 25 |
-
from src.memory.
|
| 26 |
|
| 27 |
|
| 28 |
class CommunicationChannel(Enum):
|
|
|
|
| 22 |
from src.core import get_logger
|
| 23 |
from src.core.exceptions import AgentExecutionError, DataAnalysisError
|
| 24 |
from src.services.chat_service import IntentType, Intent
|
| 25 |
+
from src.memory.conversational import ConversationalMemory, ConversationContext
|
| 26 |
|
| 27 |
|
| 28 |
class CommunicationChannel(Enum):
|
src/memory/conversational.py
CHANGED
|
@@ -2,9 +2,23 @@
|
|
| 2 |
|
| 3 |
from typing import Any, Dict, List, Optional
|
| 4 |
from datetime import datetime
|
|
|
|
| 5 |
from .base import BaseMemory
|
| 6 |
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
class ConversationalMemory(BaseMemory):
|
| 9 |
"""Memory for conversational contexts and chat history."""
|
| 10 |
|
|
|
|
| 2 |
|
| 3 |
from typing import Any, Dict, List, Optional
|
| 4 |
from datetime import datetime
|
| 5 |
+
from dataclasses import dataclass
|
| 6 |
from .base import BaseMemory
|
| 7 |
|
| 8 |
|
| 9 |
+
@dataclass
|
| 10 |
+
class ConversationContext:
|
| 11 |
+
"""Context information for conversations."""
|
| 12 |
+
session_id: str
|
| 13 |
+
user_id: Optional[str] = None
|
| 14 |
+
user_profile: Optional[Dict[str, Any]] = None
|
| 15 |
+
metadata: Dict[str, Any] = None
|
| 16 |
+
|
| 17 |
+
def __post_init__(self):
|
| 18 |
+
if self.metadata is None:
|
| 19 |
+
self.metadata = {}
|
| 20 |
+
|
| 21 |
+
|
| 22 |
class ConversationalMemory(BaseMemory):
|
| 23 |
"""Memory for conversational contexts and chat history."""
|
| 24 |
|