diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py index 539bd56a..507b8504 100644 --- a/agentops/llm_tracker.py +++ b/agentops/llm_tracker.py @@ -7,6 +7,7 @@ from .event import LLMEvent, ErrorEvent from .helpers import get_ISO_time, check_call_stack_for_agent_id import inspect +import pprint class LlmTracker: @@ -26,44 +27,51 @@ class LlmTracker: def __init__(self, client): self.client = client - self.completion = "" - self.llm_event: LLMEvent = None def _handle_response_v0_openai(self, response, kwargs, init_timestamp): """Handle responses for OpenAI versions