fiddler-langgraph 1.0.0__py3-none-any.whl → 1.1.0__py3-none-any.whl

fiddler_langgraph/VERSION

@@ -1 +1 @@
-1.0.0
+1.1.0

fiddler_langgraph/core/attributes.py

@@ -39,6 +39,7 @@ class FiddlerSpanAttributes:  # pylint: disable=too-few-public-methods
     TOOL_INPUT = 'gen_ai.tool.input'
     TOOL_OUTPUT = 'gen_ai.tool.output'
     TOOL_NAME = 'gen_ai.tool.name'
+    TOOL_DEFINITIONS = 'gen_ai.tool.definitions'
 
 
 class FiddlerResourceAttributes:

fiddler_langgraph/tracing/callback.py

@@ -138,6 +138,32 @@ def _set_token_usage_attributes(span: trace.Span, response: LLMResult) -> None:
         logger.warning('Failed to extract token usage: %s', e)
 
 
+def _set_tool_definitions(span: trace.Span, kwargs: dict[str, Any]) -> None:
+    """Extract and set tool definitions on the span.
+
+    Retrieves tool definitions from invocation params and stores them as a
+    JSON-serialized string attribute on the span.
+
+    Parameters
+    ----------
+    span : trace.Span
+        The OpenTelemetry span to set attributes on
+    kwargs : dict[str, Any]
+        Callback kwargs containing invocation_params
+
+    """
+    try:
+        invocation_params = kwargs.get('invocation_params', {})
+        tools = invocation_params.get('tools')
+
+        if tools and isinstance(tools, list) and len(tools) > 0:
+            # Store tool definitions as-is in OpenAI native format
+            tool_definitions_json = json.dumps(tools, cls=_LanggraphJSONEncoder)
+            span.set_attribute(FiddlerSpanAttributes.TOOL_DEFINITIONS, tool_definitions_json)
+    except Exception as e:
+        logger.warning('Failed to extract tool definitions: %s', e)
+
+
 class _CallbackHandler(BaseCallbackHandler):
     """A LangChain callback handler that creates OpenTelemetry spans for Fiddler.
 
@@ -641,6 +667,9 @@ class _CallbackHandler(BaseCallbackHandler):
         # Set model attributes
         _set_model_attributes(child_span, metadata)
 
+        # Extract and set tool definitions
+        _set_tool_definitions(child_span, kwargs)
+
         # We are only taking the 1st system message and 1st user message
         # as we are not supporting multiple system messages or multiple user messages
         # To support multiple system messages, we would need to add a new attribute with indexing
@@ -703,6 +732,9 @@ class _CallbackHandler(BaseCallbackHandler):
         # Set model attributes
         _set_model_attributes(child_span, metadata)
 
+        # Extract and set tool definitions
+        _set_tool_definitions(child_span, kwargs)
+
         # LLM model is more generic than a chat model, it only has a list on prompts
         # we are using the first prompt as both the system message and the user message
         # to capture all the prompts, we would need to add a new attribute with indexing

fiddler_langgraph/tracing/jsonl_capture.py

@@ -103,6 +103,7 @@ class JSONLSpanCapture:
         span_data['tool_name'] = attributes.get(FiddlerSpanAttributes.TOOL_NAME, '')
         span_data['tool_input'] = attributes.get(FiddlerSpanAttributes.TOOL_INPUT, '')
         span_data['tool_output'] = attributes.get(FiddlerSpanAttributes.TOOL_OUTPUT, '')
+        span_data['tool_definitions'] = attributes.get(FiddlerSpanAttributes.TOOL_DEFINITIONS, '')
 
         # Library versions (from resource if available)
         resource_attributes = (

{fiddler_langgraph-1.0.0.dist-info → fiddler_langgraph-1.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fiddler-langgraph
-Version: 1.0.0
+Version: 1.1.0
 Summary: Python SDK for instrumenting GenAI Applications with Fiddler
 Author-email: Fiddler AI <support@fiddler.ai>
 License-Expression: Apache-2.0

{fiddler_langgraph-1.0.0.dist-info → fiddler_langgraph-1.1.0.dist-info}/RECORD

@@ -1,15 +1,15 @@
-fiddler_langgraph/VERSION,sha256=WYVJhIUxBN9cNT4vaBoV_HkkdC-aLkaMKa8kjc5FzgM,6
+fiddler_langgraph/VERSION,sha256=FXXhr0qV8S9wtO5qatzoFglT2T6hfcJhG5CIPMw607g,6
 fiddler_langgraph/__init__.py,sha256=cqomWAmuY-2KvwJTvo7c7cecCPoe31pv4vgyBk_E8oQ,315
 fiddler_langgraph/core/__init__.py,sha256=HXPZt8YpmVrvwEEukoWR78LufMKtl7lVjLtcl9UNSoc,42
-fiddler_langgraph/core/attributes.py,sha256=TbTwr8WtmJpDDRgTP8S3wD3ZHc3wwGJK1lu3vHal3mY,2169
+fiddler_langgraph/core/attributes.py,sha256=kVu2F7XcdOA6216owWUrb49tps0DXAgM-DhNNgkk0E8,2218
 fiddler_langgraph/core/client.py,sha256=Fn4sPiTfAF1i375Qg_9DYwD0wAT65K6j3M9luyTVKPE,14948
 fiddler_langgraph/core/span_processor.py,sha256=ODYmdo0FUYEFbIWS_VaR9L6qHUVvpnuk-RSIhgRxyb0,1164
 fiddler_langgraph/tracing/__init__.py,sha256=Kw8VUB7RDffBq4ss0v6vNQYi4KDQOM0J1elbMrqJpsU,49
-fiddler_langgraph/tracing/callback.py,sha256=oI6Zpbfe50mGu85NArd8rXxOyYvOKUV4fY--EBnlQlo,31264
+fiddler_langgraph/tracing/callback.py,sha256=h75rSCl4XLNAScDt0Bh6JmluYxw7_-GQbF6FWok5Xxc,32428
 fiddler_langgraph/tracing/instrumentation.py,sha256=AlCM9GWp3qN_Fa9cl-USJjrUhuvKmrLokzkLMgah-CY,21142
-fiddler_langgraph/tracing/jsonl_capture.py,sha256=7Sy0nbxRftV5y64UCovSlm07qXn-EfG3uHpBu9H2ZiU,8174
+fiddler_langgraph/tracing/jsonl_capture.py,sha256=ju5JHGZKjUsRduAjlJMjHj_jQyLP1Tr9jzRR0j_Yatc,8273
 fiddler_langgraph/tracing/util.py,sha256=RKMrrmdCXSRJrTIHngdhRsotPLEY_LR1SKnUXAJC40Y,2678
-fiddler_langgraph-1.0.0.dist-info/METADATA,sha256=tfc_l2tf5HwODWFknx1qPfoVOFtfqHQ4GrqvzKPAxgI,10424
-fiddler_langgraph-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-fiddler_langgraph-1.0.0.dist-info/top_level.txt,sha256=hOKdR6_3AkS4dS6EfE9Ii7YrS_hApnyGfY-0v0DV0s4,18
-fiddler_langgraph-1.0.0.dist-info/RECORD,,
+fiddler_langgraph-1.1.0.dist-info/METADATA,sha256=b1ad_wYbBcEugKM3c1fkYEGtLI-WHqxA0diq4j66QwM,10424
+fiddler_langgraph-1.1.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+fiddler_langgraph-1.1.0.dist-info/top_level.txt,sha256=hOKdR6_3AkS4dS6EfE9Ii7YrS_hApnyGfY-0v0DV0s4,18
+fiddler_langgraph-1.1.0.dist-info/RECORD,,