agenta 0.20.0a10__py3-none-any.whl → 0.20.0a11__py3-none-any.whl

agenta/sdk/decorators/tracing.py

@@ -79,7 +79,7 @@ class instrument(BaseDecorator):
                             and "cost" in result.keys()
                             and "usage" in result.keys()
                         ):
-                            outputs = {"message": result["message"]}
+                            outputs = {TRACE_DEFAULT_KEY: result["message"]}
 
                             ag.tracing.store_cost(result["cost"])
                             ag.tracing.store_usage(result["usage"])

agenta/sdk/tracing/callbacks.py

@@ -5,6 +5,8 @@ from agenta.sdk.tracing.logger import llm_logger as logging
 
 from agenta.sdk.utils.debug import debug
 
+TRACE_DEFAULT_KEY = "__default__"
+
 
 def litellm_handler():
     try:
@@ -70,7 +72,7 @@ def litellm_handler():
             )
             ag.tracing.store_outputs(
                 # the complete streamed response (only set if `completion(..stream=True)`
-                outputs={"message": kwargs.get("complete_streaming_response")},
+                outputs={TRACE_DEFAULT_KEY: kwargs.get("complete_streaming_response")},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)
@@ -84,7 +86,7 @@ def litellm_handler():
                 response_obj.usage.dict() if hasattr(response_obj, "usage") else None
             )
             ag.tracing.store_outputs(
-                outputs={"message": response_obj.choices[0].message.content},
+                outputs={TRACE_DEFAULT_KEY: response_obj.choices[0].message.content},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)
@@ -110,7 +112,7 @@ def litellm_handler():
             )
             ag.tracing.store_outputs(
                 # the Exception raised
-                outputs={"message": repr(kwargs["exception"])},
+                outputs={TRACE_DEFAULT_KEY: repr(kwargs["exception"])},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)
@@ -127,7 +129,7 @@ def litellm_handler():
             )
             ag.tracing.store_outputs(
                 # the complete streamed response (only set if `completion(..stream=True)`)
-                outputs={"message": kwargs.get("complete_streaming_response")},
+                outputs={TRACE_DEFAULT_KEY: kwargs.get("complete_streaming_response")},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)
@@ -143,7 +145,7 @@ def litellm_handler():
                 response_obj.usage.dict() if hasattr(response_obj, "usage") else None
             )
             ag.tracing.store_outputs(
-                outputs={"message": response_obj.choices[0].message.content},
+                outputs={TRACE_DEFAULT_KEY: response_obj.choices[0].message.content},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)
@@ -171,7 +173,7 @@ def litellm_handler():
             )
             ag.tracing.store_outputs(
                 # the Exception raised
-                outputs={"message": repr(kwargs["exception"])},
+                outputs={TRACE_DEFAULT_KEY: repr(kwargs["exception"])},
                 span_id=self.span.id,
             )
             ag.tracing.close_span(span_id=self.span.id)

{agenta-0.20.0a10.dist-info → agenta-0.20.0a11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.20.0a10
+Version: 0.20.0a11
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.20.0a10.dist-info → agenta-0.20.0a11.dist-info}/RECORD

@@ -132,10 +132,10 @@ agenta/sdk/client.py,sha256=trKyBOYFZRk0v5Eptxvh87yPf50Y9CqY6Qgv4Fy-VH4,2142
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/decorators/base.py,sha256=9aNdX5h8a2mFweuhdO-BQPwXGKY9ONPIdLRhSGAGMfY,217
 agenta/sdk/decorators/llm_entrypoint.py,sha256=9GFT6dpVWY2_1Ckh1Y_N0xSXwosoSjtNtqMEM7u55uY,28224
-agenta/sdk/decorators/tracing.py,sha256=TwzhG3ZoyvmS1hCy3Z4S_6ZJGu1J_WHHgZm8Lc9Vkes,4371
+agenta/sdk/decorators/tracing.py,sha256=J9MG04g0OoN_nGB1JkkVzCX1XPaouQr93SR54-TSRTA,4379
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/tracing/callbacks.py,sha256=jX4nEmmQ9VX5NLQz6b0Crwv_MWzNjCweqW2DmmG7Nks,7250
+agenta/sdk/tracing/callbacks.py,sha256=JD6Q7_geFzZSwCnW8u3ISw3cWCYc2TEjGg-rf-colf8,7333
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
 agenta/sdk/tracing/llm_tracing.py,sha256=FfZQcID-8E4G0R34H29SA05yFUNDMEPccb4MqVcNDLM,18064
 agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
@@ -161,7 +161,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.20.0a10.dist-info/METADATA,sha256=xKQS1zITOgYRQWrUuQjqur7I5G6pGVA1FdM_5vWA2_U,26463
-agenta-0.20.0a10.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.20.0a10.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.20.0a10.dist-info/RECORD,,
+agenta-0.20.0a11.dist-info/METADATA,sha256=ehpOsYlSZ6gwcRxA9_kBHrCfGvaAIp9uE3aUFmOfYnQ,26463
+agenta-0.20.0a11.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.20.0a11.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.20.0a11.dist-info/RECORD,,