Skip to content

Commit

Permalink
Add more tracing
Browse files Browse the repository at this point in the history
  • Loading branch information
bonk1t committed Dec 10, 2024
1 parent bb27f8c commit d5ca39f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
2 changes: 2 additions & 0 deletions agency_swarm/threads/thread.py
Original file line number Diff line number Diff line change
Expand Up @@ -78,6 +78,7 @@ def init_thread(self):
**example,
)

@get_tracker().get_observe_decorator()
def get_completion_stream(
self,
message: Union[str, List[dict], None],
Expand All @@ -101,6 +102,7 @@ def get_completion_stream(
response_format=response_format,
)

@get_tracker().get_observe_decorator()
def get_completion(
self,
message: Union[str, List[dict], None],
Expand Down
4 changes: 2 additions & 2 deletions tests/demos/demo_observability.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@
from agency_swarm import Agency, Agent
from agency_swarm.util.oai import _get_openai_module, set_tracker

TRACKER = "sqlite"
# TRACKER = "langfuse"
# TRACKER = "sqlite"
TRACKER = "langfuse"

if TRACKER == "sqlite":
# Test SQLite configuration
Expand Down

0 comments on commit d5ca39f

Please sign in to comment.