harry-lu-0708 commited on
Commit
c76d17b
·
1 Parent(s): cade43f

fixed memo path deployment issue

Browse files
scievo/rbank/subgraph/mem_consolidation.py CHANGED
@@ -73,9 +73,13 @@ def compute_embeddings_node(state: MemConsolidationState) -> MemConsolidationSta
73
  # Convert mem_dir to Path object
74
  mem_dir_path = Path(state.mem_dir)
75
 
76
- if not mem_dir_path.exists() or not mem_dir_path.is_dir():
 
 
 
 
77
  raise AgentError(
78
- f"mem_dir does not exist or is not a directory: {mem_dir_path}", agent_name=AGENT_NAME
79
  )
80
 
81
  # Find all markdown files in the directory
 
73
  # Convert mem_dir to Path object
74
  mem_dir_path = Path(state.mem_dir)
75
 
76
+ # Create directory if it doesn't exist (it may be empty, which is fine)
77
+ if not mem_dir_path.exists():
78
+ logger.debug("Creating mem_dir: {}", mem_dir_path)
79
+ mem_dir_path.mkdir(parents=True, exist_ok=True)
80
+ elif not mem_dir_path.is_dir():
81
  raise AgentError(
82
+ f"mem_dir exists but is not a directory: {mem_dir_path}", agent_name=AGENT_NAME
83
  )
84
 
85
  # Find all markdown files in the directory
scievo/workflows/data_workflow.py CHANGED
@@ -95,6 +95,11 @@ class DataWorkflow(BaseModel):
95
  else:
96
  logger.debug(f"Using provided sess_dir: {self.sess_dir}")
97
 
 
 
 
 
 
98
  logger.info(f"Session directory: {self.sess_dir}")
99
  logger.debug(f"Long-term memory: {self.long_term_mem_dir}")
100
  logger.debug(f"Project memory: {self.project_mem_dir}")
 
95
  else:
96
  logger.debug(f"Using provided sess_dir: {self.sess_dir}")
97
 
98
+ # Ensure short_term directory exists in session directory
99
+ short_term_dir = Path(self.sess_dir) / "short_term"
100
+ short_term_dir.mkdir(parents=True, exist_ok=True)
101
+ logger.debug(f"Short-term memory directory: {short_term_dir}")
102
+
103
  logger.info(f"Session directory: {self.sess_dir}")
104
  logger.debug(f"Long-term memory: {self.long_term_mem_dir}")
105
  logger.debug(f"Project memory: {self.project_mem_dir}")