Skip to content
This repository was archived by the owner on Oct 22, 2023. It is now read-only.

Commit 1f8920c

Browse files
committed
Refactor with Black.
1 parent 062688c commit 1f8920c

File tree

2 files changed

+211
-84
lines changed

2 files changed

+211
-84
lines changed

remo.py

Lines changed: 14 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,39 +4,47 @@
44

55
app = FastAPI()
66
root_folder = os.getcwd()
7-
#root_folder = 'C:/raven_private/REMO/'
7+
# root_folder = 'C:/raven_private/REMO/'
88
max_cluster_size = 5
99
# REMO = Rolling Episodic Memory Organizer
1010

11+
1112
@app.post("/add_message")
12-
async def add_message(message: str, speaker: str, timestamp: float):
13+
async def add_message(
14+
message: str,
15+
speaker: str,
16+
timestamp: float,
17+
):
1318
# Add message to REMO
1419
new_message = utils.create_message(message, speaker, timestamp)
15-
print('\n\nADD MESSAGE -', new_message)
20+
print("\n\nADD MESSAGE -", new_message)
1621
utils.save_message(root_folder, new_message)
1722

1823
return {"detail": "Message added"}
1924

25+
2026
@app.get("/search")
2127
async def search(query: str):
2228
# Search the tree for relevant nodes
23-
print('\n\nSEARCH -', query)
29+
print("\n\nSEARCH -", query)
2430
taxonomy = utils.search_tree(root_folder, query)
2531

2632
return {"results": taxonomy}
2733

34+
2835
@app.post("/rebuild_tree")
2936
async def rebuild_tree():
3037
# Trigger full tree rebuilding event
31-
print('\n\nREBUILD TREE')
38+
print("\n\nREBUILD TREE")
3239
utils.rebuild_tree(root_folder, max_cluster_size)
3340

3441
return {"detail": "Tree rebuilding completed"}
3542

43+
3644
@app.post("/maintain_tree")
3745
async def maintain_tree():
3846
# Trigger tree maintenance event
39-
print('\n\nMAINTAIN TREE')
47+
print("\n\nMAINTAIN TREE")
4048
utils.maintain_tree(root_folder)
4149

4250
return {"detail": "Tree maintenance completed"}

0 commit comments

Comments
 (0)