34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
|
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
# Add the src directory to the Python path
|
||
|
|
sys.path.insert(0, str(Path(__file__).parent / "src"))
|
||
|
|
|
||
|
|
from timmy.memory_system import memory_store
|
||
|
|
|
||
|
|
def index_research_documents():
|
||
|
|
research_dir = Path("docs/research")
|
||
|
|
if not research_dir.is_dir():
|
||
|
|
print(f"Research directory not found: {research_dir}")
|
||
|
|
return
|
||
|
|
|
||
|
|
print(f"Indexing research documents from {research_dir}...")
|
||
|
|
indexed_count = 0
|
||
|
|
for file_path in research_dir.glob("*.md"):
|
||
|
|
try:
|
||
|
|
content = file_path.read_text()
|
||
|
|
topic = file_path.stem.replace("-", " ").title() # Derive topic from filename
|
||
|
|
print(f"Storing '{topic}' from {file_path.name}...")
|
||
|
|
# Using type="research" as per issue requirement
|
||
|
|
result = memory_store(topic=topic, report=content, type="research")
|
||
|
|
print(f" Result: {result}")
|
||
|
|
indexed_count += 1
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Error indexing {file_path.name}: {e}")
|
||
|
|
print(f"Finished indexing. Total documents indexed: {indexed_count}")
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
index_research_documents()
|