feat: Added tracing for LLM related calls!

TogetherCrew · Jan 22, 2024 · aad78f7 · aad78f7
1 parent 069cb1f
commit aad78f7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/celery_app/tasks.py b/celery_app/tasks.py
@@ -14,6 +14,7 @@
 )
 from tc_messageBroker.rabbit_mq.payload.payload import Payload
 from tc_messageBroker.rabbit_mq.queue import Queue
+from traceloop.sdk import Traceloop
 
 
 @app.task
@@ -40,6 +41,8 @@ def ask_question_auto_search(
         - `date`
         - `content`: which is the `ChatInputCommandInteraction` as a dictionary
     """
+    Traceloop.init()
+
     prefix = f"COMMUNITY_ID: {community_id} | "
     logging.info(f"{prefix}Processing question!")
 

diff --git a/docker-compose.test.yml b/docker-compose.test.yml
@@ -12,6 +12,7 @@ services:
       - MONGODB_PORT=27017
       - MONGODB_USER=root
       - MONGODB_PASS=pass
+      - TRACELOOP_BASE_URL=some_url
       - NEO4J_PROTOCOL=bolt
       - NEO4J_HOST=neo4j
       - NEO4J_PORT=7687