From 0e19e80e0127231643876a43576d82907dec27aa Mon Sep 17 00:00:00 2001 From: Pratyush Shukla Date: Sat, 10 Aug 2024 23:08:32 +0530 Subject: [PATCH] add Message object to pydantic models --- agentops/llm_tracker.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/agentops/llm_tracker.py b/agentops/llm_tracker.py index c87655ea..0997e16d 100644 --- a/agentops/llm_tracker.py +++ b/agentops/llm_tracker.py @@ -1033,6 +1033,7 @@ def patched_function(*args, **kwargs): def override_anthropic_sync_completion(self): from anthropic.resources import messages from anthropic.types import ( + Message, RawContentBlockDeltaEvent, RawContentBlockStartEvent, RawContentBlockStopEvent, @@ -1057,6 +1058,7 @@ def patched_function(*args, **kwargs): if completion_override: result_model = None pydantic_models = ( + Message, RawContentBlockDeltaEvent, RawContentBlockStartEvent, RawContentBlockStopEvent, @@ -1097,6 +1099,7 @@ def patched_function(*args, **kwargs): def override_anthropic_async_completion(self): from anthropic.resources import messages from anthropic.types import ( + Message, RawContentBlockDeltaEvent, RawContentBlockStartEvent, RawContentBlockStopEvent, @@ -1122,6 +1125,7 @@ async def patched_function(*args, **kwargs): if completion_override: result_model = None pydantic_models = ( + Message, RawContentBlockDeltaEvent, RawContentBlockStartEvent, RawContentBlockStopEvent,