kalibr 1.1.3a0__py3-none-any.whl → 1.4.0__py3-none-any.whl

kalibr_langchain/callback.py

@@ -29,6 +29,8 @@ try:
 except ImportError:
     CostAdapterFactory = None
 
+from kalibr.context import get_goal
+
 # Import tiktoken for token counting
 try:
     import tiktoken
@@ -288,6 +290,25 @@ class KalibrCallbackHandler(BaseCallbackHandler):
         # Compute cost
         cost_usd = self._compute_cost(provider, model, input_tokens, output_tokens)
 
+        # Extract tool_id from operation if this is a tool span
+        tool_id = ""
+        tool_input = ""
+        tool_output = ""
+
+        if span.get("span_type") == "tool":
+            operation = span.get("operation", "")
+            if operation.startswith("tool:"):
+                tool_id = operation[5:]  # Extract "browserless" from "tool:browserless"
+
+            # Get tool input/output from span (truncate to 10KB)
+            if span.get("input"):
+                tool_input = str(span["input"])[:10000]
+            if metadata and metadata.get("output"):
+                tool_output = str(metadata["output"])[:10000]
+
+        # Get goal from context (thread-safe)
+        current_goal = get_goal() or ""
+
         # Build event
         event = {
             "schema_version": "1.0",
@@ -318,6 +339,11 @@ class KalibrCallbackHandler(BaseCallbackHandler):
             "service": self.service,
             "runtime_env": os.getenv("RUNTIME_ENV", "local"),
             "sandbox_id": os.getenv("SANDBOX_ID", "local"),
+            # New fields for tool/goal tracking
+            "tool_id": tool_id,
+            "tool_input": tool_input,
+            "tool_output": tool_output,
+            "goal": current_goal,
             "metadata": {
                 **self.default_metadata,
                 "span_type": span.get("span_type", "llm"),

kalibr_langchain/chat_model.py

@@ -0,0 +1,103 @@
+"""
+Kalibr LangChain Chat Model - Routes requests through Kalibr.
+"""
+
+from typing import Any, List, Optional
+
+from langchain_core.callbacks import CallbackManagerForLLMRun
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_core.messages import AIMessage, BaseMessage
+from langchain_core.outputs import ChatGeneration, ChatResult
+
+
+class KalibrChatModel(BaseChatModel):
+    """
+    LangChain chat model that routes through Kalibr.
+
+    Example:
+        from kalibr import Router
+
+        router = Router(goal="summarize", paths=["gpt-4o", "claude-3"])
+        llm = router.as_langchain()
+
+        chain = prompt | llm | parser
+        result = chain.invoke({"text": "..."})
+    """
+
+    router: Any  # Kalibr Router instance
+
+    model_config = {"arbitrary_types_allowed": True}
+
+    @property
+    def _llm_type(self) -> str:
+        return "kalibr"
+
+    @property
+    def _identifying_params(self) -> dict:
+        return {"goal": self.router.goal}
+
+    def _generate(
+        self,
+        messages: List[BaseMessage],
+        stop: Optional[List[str]] = None,
+        run_manager: Optional[CallbackManagerForLLMRun] = None,
+        **kwargs: Any,
+    ) -> ChatResult:
+        """Generate a response using Kalibr routing."""
+
+        # Convert LangChain messages to OpenAI format
+        openai_messages = []
+        for m in messages:
+            role = self._get_role(m)
+            openai_messages.append({"role": role, "content": m.content})
+
+        # Add stop sequences if provided
+        if stop:
+            kwargs["stop"] = stop
+
+        # Call router
+        response = self.router.completion(messages=openai_messages, **kwargs)
+
+        # Convert response to LangChain format
+        content = response.choices[0].message.content or ""
+
+        return ChatResult(
+            generations=[
+                ChatGeneration(
+                    message=AIMessage(content=content),
+                    generation_info={
+                        "model": response.model,
+                        "finish_reason": response.choices[0].finish_reason,
+                    },
+                )
+            ],
+            llm_output={
+                "model": response.model,
+                "usage": {
+                    "prompt_tokens": response.usage.prompt_tokens,
+                    "completion_tokens": response.usage.completion_tokens,
+                    "total_tokens": response.usage.total_tokens,
+                } if hasattr(response, "usage") else {},
+            },
+        )
+
+    def _get_role(self, message: BaseMessage) -> str:
+        """Convert LangChain message type to OpenAI role."""
+        from langchain_core.messages import (
+            HumanMessage,
+            AIMessage,
+            SystemMessage,
+            FunctionMessage,
+            ToolMessage,
+        )
+
+        if isinstance(message, HumanMessage):
+            return "user"
+        elif isinstance(message, AIMessage):
+            return "assistant"
+        elif isinstance(message, SystemMessage):
+            return "system"
+        elif isinstance(message, (FunctionMessage, ToolMessage)):
+            return "function"
+        else:
+            return "user"

kalibr_openai_agents/__init__.py

@@ -26,7 +26,7 @@ Usage:
 
 Environment Variables:
     KALIBR_API_KEY: API key for authentication
-    KALIBR_ENDPOINT: Backend endpoint URL
+    KALIBR_COLLECTOR_URL: Backend endpoint URL
     KALIBR_TENANT_ID: Tenant identifier
     KALIBR_ENVIRONMENT: Environment (prod/staging/dev)
     KALIBR_SERVICE: Service name

kalibr-1.1.3a0.dist-info/METADATA

@@ -1,236 +0,0 @@
-Metadata-Version: 2.4
-Name: kalibr
-Version: 1.1.3a0
-Summary: Unified LLM Observability & Multi-Model AI Integration Framework - Deploy to GPT, Claude, Gemini, Copilot with full telemetry.
-Author-email: Kalibr Team <team@kalibr.dev>
-License: MIT
-Project-URL: Homepage, https://github.com/kalibr-ai/kalibr-sdk-python
-Project-URL: Documentation, https://docs.kalibr.systems
-Project-URL: Repository, https://github.com/kalibr-ai/kalibr-sdk-python
-Project-URL: Issues, https://github.com/kalibr-ai/kalibr-sdk-python/issues
-Keywords: ai,mcp,gpt,claude,gemini,copilot,openai,anthropic,google,microsoft,observability,telemetry,tracing,llm,schema-generation,api,multi-model,langchain,crewai
-Classifier: Development Status :: 4 - Beta
-Classifier: Intended Audience :: Developers
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
-Requires-Python: >=3.8
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: httpx>=0.27.0
-Requires-Dist: tiktoken>=0.8.0
-Requires-Dist: fastapi>=0.110.1
-Requires-Dist: uvicorn>=0.25.0
-Requires-Dist: pydantic>=2.6.4
-Requires-Dist: typer>=0.9.0
-Requires-Dist: python-multipart>=0.0.9
-Requires-Dist: rich>=10.0.0
-Requires-Dist: requests>=2.31.0
-Requires-Dist: opentelemetry-api>=1.20.0
-Requires-Dist: opentelemetry-sdk>=1.20.0
-Requires-Dist: opentelemetry-exporter-otlp>=1.20.0
-Provides-Extra: langchain
-Requires-Dist: langchain-core>=0.1.0; extra == "langchain"
-Provides-Extra: langchain-openai
-Requires-Dist: langchain-core>=0.1.0; extra == "langchain-openai"
-Requires-Dist: langchain-openai>=0.1.0; extra == "langchain-openai"
-Provides-Extra: langchain-anthropic
-Requires-Dist: langchain-core>=0.1.0; extra == "langchain-anthropic"
-Requires-Dist: langchain-anthropic>=0.1.0; extra == "langchain-anthropic"
-Provides-Extra: langchain-google
-Requires-Dist: langchain-core>=0.1.0; extra == "langchain-google"
-Requires-Dist: langchain-google-genai>=0.0.10; extra == "langchain-google"
-Provides-Extra: langchain-all
-Requires-Dist: langchain-core>=0.1.0; extra == "langchain-all"
-Requires-Dist: langchain-openai>=0.1.0; extra == "langchain-all"
-Requires-Dist: langchain-anthropic>=0.1.0; extra == "langchain-all"
-Requires-Dist: langchain-google-genai>=0.0.10; extra == "langchain-all"
-Provides-Extra: crewai
-Requires-Dist: crewai>=0.28.0; extra == "crewai"
-Provides-Extra: openai-agents
-Requires-Dist: openai-agents>=0.0.3; extra == "openai-agents"
-Provides-Extra: integrations
-Requires-Dist: langchain-core>=0.1.0; extra == "integrations"
-Requires-Dist: crewai>=0.28.0; extra == "integrations"
-Requires-Dist: openai-agents>=0.0.3; extra == "integrations"
-Provides-Extra: dev
-Requires-Dist: pytest>=7.4.0; extra == "dev"
-Requires-Dist: pytest-asyncio>=0.21.0; extra == "dev"
-Requires-Dist: black>=23.0.0; extra == "dev"
-Requires-Dist: ruff>=0.1.0; extra == "dev"
-Dynamic: license-file
-
-# Kalibr Python SDK
-
-Production-grade observability for LLM applications. Automatically instrument OpenAI, Anthropic, and Google AI SDKs with zero code changes.
-
-## Features
-
-- **Zero-code instrumentation** - Automatic tracing for OpenAI, Anthropic, and Google AI
-- **Cost tracking** - Real-time cost calculation for all LLM calls
-- **Token monitoring** - Track input/output tokens across providers
-- **Parent-child traces** - Automatic trace relationship management
-- **Multi-provider support** - Works with GPT-4, Claude, Gemini, and more
-
-## Installation
-
-```bash
-pip install kalibr
-```
-
-## Quick Start
-
-### Auto-instrumentation (Recommended)
-
-Simply import `kalibr` at the start of your application - all LLM calls are automatically traced:
-
-```python
-import kalibr  # Enable auto-instrumentation
-import openai
-
-# Set your Kalibr API key
-import os
-os.environ["KALIBR_API_KEY"] = "your-kalibr-api-key"
-
-# All OpenAI calls are now automatically traced
-client = openai.OpenAI()
-response = client.chat.completions.create(
-    model="gpt-4o",
-    messages=[{"role": "user", "content": "Hello!"}]
-)
-```
-
-### Manual Tracing with Decorator
-
-For more control, use the `@trace` decorator:
-
-```python
-from kalibr import trace
-import openai
-
-@trace(operation="summarize", provider="openai", model="gpt-4o")
-def summarize_text(text: str) -> str:
-    client = openai.OpenAI()
-    response = client.chat.completions.create(
-        model="gpt-4o",
-        messages=[
-            {"role": "system", "content": "Summarize the following text."},
-            {"role": "user", "content": text}
-        ]
-    )
-    return response.choices[0].message.content
-
-result = summarize_text("Your long text here...")
-```
-
-### Multi-Provider Example
-
-```python
-import kalibr
-import openai
-import anthropic
-
-# OpenAI call - automatically traced
-openai_client = openai.OpenAI()
-gpt_response = openai_client.chat.completions.create(
-    model="gpt-4o",
-    messages=[{"role": "user", "content": "Explain quantum computing"}]
-)
-
-# Anthropic call - automatically traced
-anthropic_client = anthropic.Anthropic()
-claude_response = anthropic_client.messages.create(
-    model="claude-sonnet-4-20250514",
-    max_tokens=1024,
-    messages=[{"role": "user", "content": "Explain machine learning"}]
-)
-```
-
-## Configuration
-
-Configure the SDK using environment variables:
-
-| Variable | Description | Default |
-|----------|-------------|---------|
-| `KALIBR_API_KEY` | API key for authentication | *Required* |
-| `KALIBR_COLLECTOR_URL` | Collector endpoint URL | `http://localhost:8001/api/ingest` |
-| `KALIBR_TENANT_ID` | Tenant identifier for multi-tenant setups | `default` |
-| `KALIBR_WORKFLOW_ID` | Workflow identifier for grouping traces | `default` |
-| `KALIBR_SERVICE_NAME` | Service name for OpenTelemetry spans | `kalibr-app` |
-| `KALIBR_ENVIRONMENT` | Environment (prod, staging, dev) | `prod` |
-| `KALIBR_AUTO_INSTRUMENT` | Enable/disable auto-instrumentation | `true` |
-| `KALIBR_CONSOLE_EXPORT` | Enable console span export for debugging | `false` |
-
-## CLI Tools
-
-The SDK includes command-line tools for running and deploying applications:
-
-```bash
-# Run your app locally with tracing
-kalibr serve myapp.py
-
-# Run with managed runtime lifecycle
-kalibr run myapp.py --port 8000
-
-# Deploy to cloud platforms
-kalibr deploy myapp.py --runtime fly.io
-
-# Fetch trace data by ID
-kalibr capsule <trace-id>
-```
-
-## Supported Providers
-
-| Provider | Models | Auto-Instrumentation |
-|----------|--------|---------------------|
-| OpenAI | GPT-4, GPT-4o, GPT-3.5 | Yes |
-| Anthropic | Claude 3 Opus, Sonnet, Haiku | Yes |
-| Google | Gemini Pro, Gemini Flash | Yes |
-
-## Examples
-
-See the [`examples/`](./examples) directory for complete examples:
-
-- `basic_example.py` - Simple tracing example
-- `basic_agent.py` - Agent with auto-instrumentation
-- `advanced_example.py` - Advanced tracing patterns
-- `cross_vendor.py` - Multi-provider workflows
-- `test_mas.py` - Multi-agent system demonstration
-
-## Development
-
-```bash
-# Clone the repository
-git clone https://github.com/kalibr-ai/kalibr-sdk-python.git
-cd kalibr-sdk-python
-
-# Install in development mode
-pip install -e ".[dev]"
-
-# Run tests
-pytest
-
-# Format code
-black kalibr/
-ruff check kalibr/
-```
-
-## Contributing
-
-We welcome contributions! Please see [CONTRIBUTING.md](./CONTRIBUTING.md) for guidelines.
-
-## License
-
-MIT License - see [LICENSE](./LICENSE) for details.
-
-## Links
-
-- [Documentation](https://docs.kalibr.systems)
-- [GitHub Issues](https://github.com/kalibr-ai/kalibr-sdk-python/issues)
-- [PyPI Package](https://pypi.org/project/kalibr/)

kalibr-1.1.3a0.dist-info/RECORD

@@ -1,48 +0,0 @@
-kalibr/__init__.py,sha256=gyljj_U2w1JlQCTZEvdzNA9vKhcr5FIsTiu02voirLY,4284
-kalibr/__main__.py,sha256=jO96I4pqinwHg7ONRvNVKbySBh5pSIhOAiNrgSQrNlY,110
-kalibr/capsule_middleware.py,sha256=pXG_wORgCqo3wHjtkn_zY4doLyiDmTwJtB7XiZNnbPk,3163
-kalibr/client.py,sha256=6D1paakE6zgWJStaow3ak9t0R8afodQhSSpUO3WTs_8,9732
-kalibr/collector.py,sha256=rtTKQLe6NkDSblBIfFooQ-ESFcP0Q1HUp4Bcqqg8JFo,5818
-kalibr/context.py,sha256=hBxWXZx0gcmeGqDMS1rstke_DmrujoRBIsfrG26WKUY,3755
-kalibr/cost_adapter.py,sha256=NerJ7ywaJjBn97gVFr7qKX7318e3Kmy2qqeNlGl9nPE,6439
-kalibr/decorators.py,sha256=m-XBXxWMDVrzaNsljACiGmeGhgiHj_MqSfj6OGK3L5I,4380
-kalibr/kalibr.py,sha256=cNXC3W_TX5SvGsy1lRopkwFqsHOpyd1kkVjEMOz1Yr4,6084
-kalibr/kalibr_app.py,sha256=ItZwEh0FZPx9_BE-zPQajC2yxI2y9IHYwJD0k9tbHvY,2773
-kalibr/models.py,sha256=HwD_-iysZMSnCzMQYO1Qcf0aeXySupY7yJeBwl_dLS0,1024
-kalibr/redaction.py,sha256=XibxX4Lv1Ci0opE6Tb5ZI2GLbO0a8E9U66MAg60llnc,1139
-kalibr/schemas.py,sha256=XLZNLkXca6jbj9AF6gDIyGVnIcr1SVOsNYaKvW-wbgE,3669
-kalibr/simple_tracer.py,sha256=VAhqxGhCMBz9rVFXfpJtRmt6SrM_cpUBKE5ygP9PC9Y,9779
-kalibr/tokens.py,sha256=istjgaxi9S4dMddjuGtoQaTnZYcWLCqdnxRjV86yNXA,1297
-kalibr/trace_capsule.py,sha256=CPMUz5D-fVfao-MozNtSDbgOQKdDAJxTN5KQL6w2Xp8,10154
-kalibr/trace_models.py,sha256=9o7VJQk3gCrvdfXPrNh3Ptkq5sRgA9_qrLLE3jNkSBg,7304
-kalibr/tracer.py,sha256=jwWBpZbGXn6fEv4pw25BLFCH-22QUbyzofPWp1Iwdkk,11911
-kalibr/types.py,sha256=cna4-akpdwfHXfOJCtVIq5lO_jaoG2Am3BRrXi0Vo34,895
-kalibr/utils.py,sha256=DJ9Mp6IPzyQTKboVV0utnfFBMT7LbJ9vAs0gOOPCpvw,5052
-kalibr/cli/__init__.py,sha256=FmRGaDMhM9DhrKg1ONkF0emIrJcjFWjlFBl_oenvpsk,77
-kalibr/cli/capsule_cmd.py,sha256=j4XR10mq9Ni6DDHNt3cNYEYuMr7jabExEkEB2vrcgGI,6092
-kalibr/cli/deploy_cmd.py,sha256=kV4uqCN2IdQev1vPBY5qqIHsEhjGBZ7y_rLx8RGAL_4,5178
-kalibr/cli/main.py,sha256=Ob0Vpg8KPnHluP_23pgbEZpDb_iKnjoN7mW52-2Qr44,1939
-kalibr/cli/run.py,sha256=S8l5DCcoMVHl_frb-8DOkpDHS3Q6RuYvns9izNk-lx0,6428
-kalibr/cli/serve.py,sha256=71Xha35qrBNkcQxuUkwC-ixbOriHGUIEgxl7C_qERQo,2085
-kalibr/instrumentation/__init__.py,sha256=YnUJ4gUH8WNxdVv5t1amn0l2WUULJG2MuQIL2ZZhn04,354
-kalibr/instrumentation/anthropic_instr.py,sha256=ozXHr8BPMafIbvgaxunskQi9YX5_Gpoiekye77oRc2E,10058
-kalibr/instrumentation/base.py,sha256=eMFBTIQXtG2bZD5st6vzN72ooeHCANZ3SapYzrdijgk,3109
-kalibr/instrumentation/google_instr.py,sha256=f2um7MB2QCT2u9CFV4-vKke-8M0dSXSpZHTcbdxMZyI,10476
-kalibr/instrumentation/openai_instr.py,sha256=UU0Pi1Gq1FqgetYWDacQhNFdjemuPrc0hRTKd-LIDHI,9250
-kalibr/instrumentation/registry.py,sha256=sfQnXhbPOI5LVon2kFhe8KcXQwWmuKW1XUe50B2AaBc,4749
-kalibr/middleware/__init__.py,sha256=qyDUn_irAX67MS-IkuDVxg4RmFnJHDf_BfIT3qfGoBI,115
-kalibr/middleware/auto_tracer.py,sha256=ZBSBM0O3a6rwVzfik1n5NUmQDah8_iaf86rU64aPYT4,13037
-kalibr-1.1.3a0.dist-info/licenses/LICENSE,sha256=BYlEPoDdYD3iHuAjt2JYGoxDYQaI1gxab2pR4acoz04,1063
-kalibr_crewai/__init__.py,sha256=ClbyRwZ9K3VabYg3RnjnCNQtv1z7UNyCAlMSo0tCGYQ,1791
-kalibr_crewai/callbacks.py,sha256=UBgGw0vdT0Jf9x8fNrHfsUR4unqX4nxNFta07OoSgaI,17162
-kalibr_crewai/instrumentor.py,sha256=AfnK5t7Ynb-7ytZF7XdOSPpr0o8hDf3sFkyzhc1ogY0,19465
-kalibr_langchain/__init__.py,sha256=BPj6JZgWDh3wNQCnIfdC24gupzBjVPxbq3zBsfn7yCY,1368
-kalibr_langchain/async_callback.py,sha256=_Mj_YrKbULNtfxixZ7iwiHyWEV9l178ZA5Oy5A5Pakk,27748
-kalibr_langchain/callback.py,sha256=VVPAvksS8TFMC21QlGj-1NRFsWnkLKPyzqhfA3kmT4c,34265
-kalibr_openai_agents/__init__.py,sha256=ToKJ1Krwc38ADZO0HCWo1_ZMGi4VRIEgym2Avkg0CyQ,1362
-kalibr_openai_agents/processor.py,sha256=F550sdRf3rpguP1yOlgAUQWDLPBy4hSACV3-zOyCpOU,18257
-kalibr-1.1.3a0.dist-info/METADATA,sha256=tkZQEBASAyAucKITC1_yzuASDFq2uSOAfMoQqCKP3TI,7840
-kalibr-1.1.3a0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-kalibr-1.1.3a0.dist-info/entry_points.txt,sha256=Kojlc6WRX8V1qS9lOMdDPZpTUVHCtzGtHqXusErgmLY,47
-kalibr-1.1.3a0.dist-info/top_level.txt,sha256=dIfBOWUnnHGFDwgz5zfIx5_0bU3wOUgAbYr4JcFHZmo,59
-kalibr-1.1.3a0.dist-info/RECORD,,

kalibr-1.1.3a0.dist-info/licenses/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2025 Kalibr
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.