docent-python 0.1.7a0__py3-none-any.whl → 0.1.9a0__py3-none-any.whl

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/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:
@@ -626,7 +630,7 @@ class DocentTracer:
             raise RuntimeError("API endpoint base is not configured")
         url = f"{self._api_endpoint_base}{path}"
         try:
-            resp = requests.post(url, json=data, headers=self._api_headers(), timeout=10)
+            resp = requests.post(url, json=data, headers=self._api_headers(), timeout=(10, 60))
             resp.raise_for_status()
         except requests.exceptions.RequestException as e:
             logger.error(f"Failed POST {url}: {e}")
@@ -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.7a0.dist-info → docent_python-0.1.9a0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: docent-python
-Version: 0.1.7a0
+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.7a0.dist-info → docent_python-0.1.9a0.dist-info}/RECORD

@@ -1,6 +1,6 @@
 docent/__init__.py,sha256=J2BbO6rzilfw9WXRUeolr439EGFezqbMU_kCpCCryRA,59
 docent/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-docent/trace.py,sha256=_0grQFHEBAL98ZtEcDncoCWL23cJ_1j7DHrd2Fs1lPI,66136
+docent/trace.py,sha256=-3QbmAjZJfKZp9pmhLne0EIB-akbz2XbEDNqpsROEco,66513
 docent/trace_temp.py,sha256=Z0lAPwVzXjFvxpiU-CuvfWIslq9Q4alNkZMoQ77Xudk,40711
 docent/_log_util/__init__.py,sha256=3HXXrxrSm8PxwG4llotrCnSnp7GuroK1FNHsdg6f7aE,73
 docent/_log_util/logger.py,sha256=kwM0yRW1IJd6-XTorjWn48B4l8qvD2ZM6VDjY5eskQI,4422
@@ -12,7 +12,7 @@ docent/data_models/metadata.py,sha256=r0SYC4i2x096dXMLfw_rAMtcJQCsoV6EOMPZuEngbG
 docent/data_models/regex.py,sha256=0ciIerkrNwb91bY5mTcyO5nDWH67xx2tZYObV52fmBo,1684
 docent/data_models/shared_types.py,sha256=jjm-Dh5S6v7UKInW7SEqoziOsx6Z7Uu4e3VzgCbTWvc,225
 docent/data_models/transcript.py,sha256=0iF2ujcWhTss8WkkpNMeIKJyKOfMEsiMoAQMGwY4ing,15753
-docent/data_models/chat/__init__.py,sha256=O04XQ2NmO8GTWqkkB_Iydj8j_CucZuLhoyMVTxJN_cs,570
+docent/data_models/chat/__init__.py,sha256=GleyRzYqKRkwwSRm_tQJw5BudCbgu9WRSa71Fntz0L0,610
 docent/data_models/chat/content.py,sha256=Co-jO8frQa_DSP11wJuhPX0s-GpJk8yqtKqPeiAIZ_U,1672
 docent/data_models/chat/message.py,sha256=iAo38kbV6wYbFh8S23cxLy6HY4C_i3PzQ6RpSQG5dxM,3861
 docent/data_models/chat/tool.py,sha256=x7NKINswPe0Kqvcx4ubjHzB-n0-i4DbFodvaBb2vitk,3042
@@ -23,7 +23,7 @@ docent/samples/log.eval,sha256=orrW__9WBfANq7NwKsPSq9oTsQRcG6KohG5tMr_X_XY,39770
 docent/samples/tb_airline.json,sha256=eR2jFFRtOw06xqbEglh6-dPewjifOk-cuxJq67Dtu5I,47028
 docent/sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 docent/sdk/client.py,sha256=fLdniy8JzMLoZpaS9SP2pHban_ToavgtI8VeHZLMNZo,12773
-docent_python-0.1.7a0.dist-info/METADATA,sha256=k5o8vcR52Rp7GmJwz2o7ao9F2nw0g35npQ5bQOKbsPI,1037
-docent_python-0.1.7a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-docent_python-0.1.7a0.dist-info/licenses/LICENSE.md,sha256=vOHzq3K4Ndu0UV9hPrtXvlD7pHOjyDQmGjHuLSIkRQY,1087
-docent_python-0.1.7a0.dist-info/RECORD,,
+docent_python-0.1.9a0.dist-info/METADATA,sha256=fgAhTw2bXGNLlU2Y6XFq2rvg7lloXipHXWRXXHLq4gw,1037
+docent_python-0.1.9a0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+docent_python-0.1.9a0.dist-info/licenses/LICENSE.md,sha256=vOHzq3K4Ndu0UV9hPrtXvlD7pHOjyDQmGjHuLSIkRQY,1087
+docent_python-0.1.9a0.dist-info/RECORD,,