Added memory api

1 files changed, 2 insertions, 3 deletions

diff --git a/llm.py b/llm.py
index 4dde6bd..afd890c 100644
--- a/llm.py
+++ b/llm.py
@@ -1,6 +1,5 @@
-from dataclasses import dataclass
 import json
-from collections.abc import Iterator
+from dataclasses import dataclass
 
 import requests
 
@@ -68,7 +67,7 @@ class LLM:
 
 
 if __name__ == "__main__":
-    from tools import ToolRegistry, HelloTool
+    from tools import HelloTool, ToolRegistry
 
     registry = ToolRegistry()
     registry.register(HelloTool())