|
4 | 4 |
|
5 | 5 | app = FastAPI() |
6 | 6 | root_folder = os.getcwd() |
7 | | -#root_folder = 'C:/raven_private/REMO/' |
| 7 | +# root_folder = 'C:/raven_private/REMO/' |
8 | 8 | max_cluster_size = 5 |
9 | 9 | # REMO = Rolling Episodic Memory Organizer |
10 | 10 |
|
| 11 | + |
11 | 12 | @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 | +): |
13 | 18 | # Add message to REMO |
14 | 19 | new_message = utils.create_message(message, speaker, timestamp) |
15 | | - print('\n\nADD MESSAGE -', new_message) |
| 20 | + print("\n\nADD MESSAGE -", new_message) |
16 | 21 | utils.save_message(root_folder, new_message) |
17 | 22 |
|
18 | 23 | return {"detail": "Message added"} |
19 | 24 |
|
| 25 | + |
20 | 26 | @app.get("/search") |
21 | 27 | async def search(query: str): |
22 | 28 | # Search the tree for relevant nodes |
23 | | - print('\n\nSEARCH -', query) |
| 29 | + print("\n\nSEARCH -", query) |
24 | 30 | taxonomy = utils.search_tree(root_folder, query) |
25 | 31 |
|
26 | 32 | return {"results": taxonomy} |
27 | 33 |
|
| 34 | + |
28 | 35 | @app.post("/rebuild_tree") |
29 | 36 | async def rebuild_tree(): |
30 | 37 | # Trigger full tree rebuilding event |
31 | | - print('\n\nREBUILD TREE') |
| 38 | + print("\n\nREBUILD TREE") |
32 | 39 | utils.rebuild_tree(root_folder, max_cluster_size) |
33 | 40 |
|
34 | 41 | return {"detail": "Tree rebuilding completed"} |
35 | 42 |
|
| 43 | + |
36 | 44 | @app.post("/maintain_tree") |
37 | 45 | async def maintain_tree(): |
38 | 46 | # Trigger tree maintenance event |
39 | | - print('\n\nMAINTAIN TREE') |
| 47 | + print("\n\nMAINTAIN TREE") |
40 | 48 | utils.maintain_tree(root_folder) |
41 | 49 |
|
42 | 50 | return {"detail": "Tree maintenance completed"} |
0 commit comments