229 - Traces (#230)

thenewboston-developers · Aug 17, 2024 · 6611cb6 · 6611cb6
1 parent fe970c1
commit 6611cb6
Show file tree

Hide file tree

Showing 3 changed files with 90 additions and 17 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,8 +35,8 @@ jinja2 = "^3.1.3"
 sentry-sdk = {extras = ["django"], version = "^1.45.0"}
 django-restql = "^0.15.4"
 discord-py = "^2.4.0"
-promptlayer = "^1.0.9"
 anthropic = "^0.32.0"
+promptlayer = "^1.0.14"
 
 [tool.poetry.group.dev.dependencies]
 colorlog = "^6.7.0"

diff --git a/thenewboston/general/clients/llm.py b/thenewboston/general/clients/llm.py
@@ -53,7 +53,7 @@ def promptlayer_client(self):
         if anthropic_api_key := self.anthropic_api_key:
             os.environ['ANTHROPIC_API_KEY'] = anthropic_api_key
 
-        return PromptLayer(api_key=self.promptlayer_api_key)
+        return PromptLayer(api_key=self.promptlayer_api_key, enable_tracing=True, workspace_id=8353)
 
     @property
     def openai_client(self):