fiddler-langgraph 1.0.0__tar.gz → 1.1.0__tar.gz

{fiddler_langgraph-1.0.0/fiddler_langgraph.egg-info → fiddler_langgraph-1.1.0}/PKG-INFO

@@ -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
 Home-page: https://fiddler.ai
 Author: Fiddler AI

fiddler_langgraph-1.1.0/fiddler_langgraph/VERSION

@@ -0,0 +1 @@
+1.1.0

{fiddler_langgraph-1.0.0 → fiddler_langgraph-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-1.0.0 → fiddler_langgraph-1.1.0}/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-1.0.0 → fiddler_langgraph-1.1.0}/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 → fiddler_langgraph-1.1.0/fiddler_langgraph.egg-info}/PKG-INFO

@@ -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
 Home-page: https://fiddler.ai
 Author: Fiddler AI

fiddler_langgraph-1.0.0/fiddler_langgraph/VERSION

@@ -1 +0,0 @@
-1.0.0