diff options
Diffstat (limited to 'personal_agent/llm.py')
| -rw-r--r-- | personal_agent/llm.py | 6 |
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") |