docent-python 0.1.8a0__tar.gz → 0.1.9a0__tar.gz

{docent_python-0.1.8a0 → docent_python-0.1.9a0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.8a0
+Version: 0.1.9a0
 Summary: Docent SDK
 Project-URL: Homepage, https://github.com/TransluceAI/docent
 Project-URL: Issues, https://github.com/TransluceAI/docent/issues

{docent_python-0.1.8a0 → docent_python-0.1.9a0}/docent/data_models/chat/__init__.py

@@ -7,7 +7,7 @@ from docent.data_models.chat.message import (
     UserMessage,
     parse_chat_message,
 )
-from docent.data_models.chat.tool import ToolCall, ToolInfo, ToolParams
+from docent.data_models.chat.tool import ToolCall, ToolCallContent, ToolInfo, ToolParams
 
 __all__ = [
     "ChatMessage",
@@ -19,6 +19,7 @@ __all__ = [
     "ContentReasoning",
     "ContentText",
     "ToolCall",
+    "ToolCallContent",
     "ToolInfo",
     "ToolParams",
     "parse_chat_message",

{docent_python-0.1.8a0 → docent_python-0.1.9a0}/docent/trace.py

@@ -200,7 +200,6 @@ class DocentTracer:
     def _signal_handler(self, signum: int, frame: Optional[object]):
         """Handle shutdown signals."""
         self.cleanup()
-        sys.exit(0)
 
     def _init_spans_exporter(self, endpoint: str) -> Optional[Union[HTTPExporter, GRPCExporter]]:
         """Initialize the appropriate span exporter based on endpoint."""
@@ -463,6 +462,10 @@ class DocentTracer:
         except Exception as e:
             logger.error(f"Error during flush: {e}")
 
+    def is_disabled(self) -> bool:
+        """Check if tracing is disabled."""
+        return self._disabled
+
     def set_disabled(self, disabled: bool) -> None:
         """Enable or disable tracing."""
         self._disabled = disabled
@@ -616,9 +619,10 @@ class DocentTracer:
         Returns:
             Dictionary of headers including Authorization
         """
+
         return {
             "Content-Type": "application/json",
-            "Authorization": f"Bearer {self.headers.get('Authorization', '').replace('Bearer ', '')}",
+            "Authorization": self.headers.get("Authorization", ""),
         }
 
     def _post_json(self, path: str, data: Dict[str, Any]) -> None:
@@ -1163,6 +1167,13 @@ def verify_initialized() -> bool:
     return _global_tracer.verify_initialized()
 
 
+def is_disabled() -> bool:
+    """Check if global tracing is disabled."""
+    if _global_tracer:
+        return _global_tracer.is_disabled()
+    return True
+
+
 def set_disabled(disabled: bool) -> None:
     """Enable or disable global tracing."""
     if _global_tracer:
@@ -1180,6 +1191,8 @@ def agent_run_score(name: str, score: float, attributes: Optional[Dict[str, Any]
     """
     try:
         tracer: DocentTracer = get_tracer()
+        if tracer.is_disabled():
+            return
         agent_run_id = tracer.get_current_agent_run_id()
 
         if not agent_run_id:
@@ -1216,6 +1229,8 @@ def agent_run_metadata(metadata: Dict[str, Any]) -> None:
     """
     try:
         tracer = get_tracer()
+        if tracer.is_disabled():
+            return
         agent_run_id = tracer.get_current_agent_run_id()
         if not agent_run_id:
             logger.warning("No active agent run context. Metadata will not be sent.")
@@ -1248,6 +1263,8 @@ def transcript_metadata(
     """
     try:
         tracer = get_tracer()
+        if tracer.is_disabled():
+            return
         transcript_id = tracer.get_current_transcript_id()
         if not transcript_id:
             logger.warning("No active transcript context. Metadata will not be sent.")

{docent_python-0.1.8a0 → docent_python-0.1.9a0}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "docent-python"
 description = "Docent SDK"
-version = "0.1.8-alpha"
+version = "0.1.9-alpha"
 authors = [
   { name="Transluce", email="info@transluce.org" },
 ]