Revert "feat: add local langfuse tracing option (#106)"

This reverts commit 56d88a8.

.github/workflows/ci.yaml

@@ -28,7 +28,7 @@ jobs:
           uv run pytest tests -m 'not integration'
 
   goose:
-    runs-on: ubuntu-latest 
+    runs-on: ubuntu-latest
 
     steps:
       - uses: actions/checkout@v4
@@ -48,27 +48,6 @@ jobs:
         run: |
           uv run pytest tests -m 'not integration'
 
-  langfuse-wrapper:
-    runs-on: ubuntu-latest
-
-    steps:
-      - uses: actions/checkout@v4
-
-      - name: Install UV
-        run: curl -LsSf https://astral.sh/uv/install.sh | sh
-
-      - name: Source Cargo Environment
-        run: source $HOME/.cargo/env
-
-      - name: Ruff
-        run: |
-          uvx ruff check packages/langfuse-wrapper
-          uvx ruff format packages/langfuse-wrapper --check
-
-      - name: Run tests
-        working-directory: ./packages/langfuse-wrapper
-        run: |
-          uv run pytest tests -m 'not integration'
 
   # This runs integration tests of the OpenAI API, using Ollama to host models.
   # This lets us test PRs from forks which can't access secrets like API keys.

.gitignore

@@ -125,6 +125,3 @@ docs/docs/reference
 
 # uv lock file
 uv.lock
-
-# langfuse docker file
-**/packages/langfuse-wrapper/scripts/docker-compose.yaml

packages/exchange/pyproject.toml

@@ -13,8 +13,6 @@ dependencies = [
     "tiktoken>=0.7.0",
     "httpx>=0.27.0",
     "tenacity>=9.0.0",
-    "python-dotenv>=1.0.1",
-    "langfuse-wrapper"   
 ]
 
 [tool.hatch.build.targets.wheel]
@@ -49,10 +47,3 @@ ai-exchange = "exchange:module_name"
 markers = [
     "integration: marks tests that need to authenticate (deselect with '-m \"not integration\"')",
 ]
-
-[tool.uv.sources]
-langfuse-wrapper = { workspace = true}
-
-[tool.uv.workspace]
-members = ["../langfuse-wrapper"]
-

packages/exchange/src/exchange/exchange.py

@@ -14,7 +14,6 @@
 from exchange.providers import Provider, Usage
 from exchange.tool import Tool
 from exchange.token_usage_collector import _token_usage_collector
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
 
 
 def validate_tool_output(output: str) -> None:
@@ -128,7 +127,6 @@ def reply(self, max_tool_use: int = 128) -> Message:
 
         return response
 
-    @observe_wrapper()
     def call_function(self, tool_use: ToolUse) -> ToolResult:
         """Call the function indicated by the tool use"""
         tool = self._toolmap.get(tool_use.name)

packages/exchange/src/exchange/providers/anthropic.py

@@ -8,7 +8,6 @@
 from exchange.providers.base import Provider, Usage
 from tenacity import retry, wait_fixed, stop_after_attempt
 from exchange.providers.utils import retry_if_status, raise_for_status
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
 
 ANTHROPIC_HOST = "https://api.anthropic.com/v1/messages"
 
@@ -124,7 +123,6 @@ def messages_to_anthropic_spec(messages: List[Message]) -> List[Dict[str, Any]]:
             messages_spec.append(converted)
         return messages_spec
 
-    @observe_wrapper(as_type="generation")
     def complete(
         self,
         model: str,

packages/exchange/src/exchange/providers/bedrock.py

@@ -15,7 +15,6 @@
 from tenacity import retry, wait_fixed, stop_after_attempt
 from exchange.providers.utils import raise_for_status, retry_if_status
 from exchange.tool import Tool
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
 
 SERVICE = "bedrock-runtime"
 UTC = timezone.utc
@@ -176,7 +175,6 @@ def from_env(cls: Type["BedrockProvider"]) -> "BedrockProvider":
         )
         return cls(client=client)
 
-    @observe_wrapper(as_type="generation")
     def complete(
         self,
         model: str,

packages/exchange/src/exchange/providers/databricks.py

@@ -13,7 +13,7 @@
     tools_to_openai_spec,
 )
 from exchange.tool import Tool
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
+
 
 retry_procedure = retry(
     wait=wait_fixed(2),
@@ -69,7 +69,6 @@ def get_usage(data: dict) -> Usage:
             total_tokens=total_tokens,
         )
 
-    @observe_wrapper(as_type="generation")
     def complete(
         self,
         model: str,

packages/exchange/src/exchange/providers/google.py

@@ -8,7 +8,6 @@
 from exchange.providers.base import Provider, Usage
 from tenacity import retry, wait_fixed, stop_after_attempt
 from exchange.providers.utils import raise_for_status, retry_if_status
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
 
 GOOGLE_HOST = "https://generativelanguage.googleapis.com/v1beta"
 
@@ -122,7 +121,6 @@ def messages_to_google_spec(messages: List[Message]) -> List[Dict[str, Any]]:
 
         return messages_spec
 
-    @observe_wrapper(as_type="generation")
     def complete(
         self,
         model: str,

packages/exchange/src/exchange/providers/openai.py

@@ -15,7 +15,6 @@
 from exchange.tool import Tool
 from tenacity import retry, wait_fixed, stop_after_attempt
 from exchange.providers.utils import retry_if_status
-from langfuse_wrapper.langfuse_wrapper import observe_wrapper
 
 OPENAI_HOST = "https://api.openai.com/"
 
@@ -66,7 +65,6 @@ def get_usage(data: dict) -> Usage:
             total_tokens=total_tokens,
         )
 
-    @observe_wrapper(as_type="generation")
     def complete(
         self,
         model: str,