anderson-ufrj commited on
Commit
7878f71
·
1 Parent(s): 0bfbae0

fix: add missing ConversationContext class and fix import path

Browse files

Fixed 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 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.conversational_memory import ConversationalMemory, ConversationContext
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