Added more memory management and fixed system prompt bug

1 files changed, 5 insertions, 1 deletions

diff --git a/personal_agent/llm.py b/personal_agent/llm.py
index 418c478..cd35676 100644
--- a/personal_agent/llm.py
+++ b/personal_agent/llm.py
@@ -1,8 +1,11 @@
 import json
+import logging
 from dataclasses import dataclass
 
 import requests
 
+logger = logging.getLogger(__name__)
+
 
 @dataclass
 class ToolCall:
@@ -43,7 +46,8 @@ class LLM:
         response.raise_for_status()
 
         data = response.json()
-        print(f"\n[debug: {data}]\n")
+        logger.debug("LLM response: %s", data)
+        # print(f"\n[debug: {data}]\n")
         message = data["choices"][0]["message"]
 
         content = message.get("content")