agenta 0.19.6a0__py3-none-any.whl → 0.19.8__py3-none-any.whl

agenta/cli/variant_commands.py

@@ -139,7 +139,7 @@ def add_variant(
 
     except Exception as ex:
         click.echo(click.style(f"Error while building image: {ex}", fg="red"))
-        return None
+        raise
     try:
         if overwrite:
             click.echo(
@@ -166,7 +166,7 @@ def add_variant(
             click.echo(click.style(f"Error while updating variant: {ex}", fg="red"))
         else:
             click.echo(click.style(f"Error while adding variant: {ex}", fg="red"))
-        return None
+        raise
 
     agenta_dir = Path.home() / ".agenta"
     global_toml_file = toml.load(agenta_dir / "config.toml")
@@ -461,28 +461,28 @@ def serve_cli(ctx, app_folder: str, file_name: str, overwrite: bool):
             error_msg += "or\n"
             error_msg += ">>> agenta variant serve <filename>.py"
             click.echo(click.style(f"{error_msg}", fg="red"))
-            sys.exit(0)
+            sys.exit(1)
 
     try:
         config_check(app_folder)
     except Exception as e:
         click.echo(click.style("Failed during configuration check.", fg="red"))
         click.echo(click.style(f"Error message: {str(e)}", fg="red"))
-        return
+        sys.exit(1)
 
     try:
         host = get_host(app_folder)
     except Exception as e:
         click.echo(click.style("Failed to retrieve the host.", fg="red"))
         click.echo(click.style(f"Error message: {str(e)}", fg="red"))
-        return
+        sys.exit(1)
 
     try:
         api_key = helper.get_global_config("api_key")
     except Exception as e:
         click.echo(click.style("Failed to retrieve the api key.", fg="red"))
         click.echo(click.style(f"Error message: {str(e)}", fg="red"))
-        return
+        sys.exit(1)
 
     try:
         variant_id = add_variant(
@@ -491,7 +491,7 @@ def serve_cli(ctx, app_folder: str, file_name: str, overwrite: bool):
     except Exception as e:
         click.echo(click.style("Failed to add variant.", fg="red"))
         click.echo(click.style(f"Error message: {str(e)}", fg="red"))
-        return
+        sys.exit(1)
 
     if variant_id:
         try:
@@ -503,9 +503,11 @@ def serve_cli(ctx, app_folder: str, file_name: str, overwrite: bool):
                 "- Second, try restarting the containers (if using Docker Compose)."
             )
             click.echo(click.style(f"{error_msg}", fg="red"))
+            sys.exit(1)
         except Exception as e:
             click.echo(click.style("Failed to start container with LLM app.", fg="red"))
             click.echo(click.style(f"Error message: {str(e)}", fg="red"))
+            sys.exit(1)
 
 
 @variant.command(name="list")

agenta/sdk/agenta_init.py

@@ -240,11 +240,6 @@ def init(
 
     singleton.init(app_id=app_id, host=host, api_key=api_key, config_fname=config_fname)
 
-    if os.environ.get("AGENTA_LOCAL", False):
-        singleton.host = singleton.host.replace(
-            "http://localhost", "http://host.docker.internal"
-        )
-
     tracing = Tracing(
         host=singleton.host,  # type: ignore
         app_id=singleton.app_id,  # type: ignore

agenta/sdk/decorators/tracing.py

@@ -56,7 +56,6 @@ class instrument(BaseDecorator):
             input_dict.update(kwargs)
 
             async def wrapped_func(*args, **kwargs):
-
                 # logging.debug(" ".join([">..", str(tracing_context.get())]))
 
                 token = None
@@ -124,7 +123,6 @@ class instrument(BaseDecorator):
             input_dict.update(kwargs)
 
             def wrapped_func(*args, **kwargs):
-
                 # logging.debug(" ".join([">..", str(tracing_context.get())]))
 
                 token = None

agenta/sdk/tracing/callbacks.py

@@ -32,12 +32,13 @@ def litellm_handler():
             span_kind = (
                 "llm" if call_type in ["completion", "acompletion"] else "embedding"
             )
-            self._trace.start_span(
+
+            ag.tracing.start_span(
                 name=f"{span_kind}_call",
                 input={"messages": kwargs["messages"]},
                 spankind=span_kind,
             )
-            self._trace.set_span_attribute(
+            ag.tracing.set_attributes(
                 {
                     "model_config": {
                         "model": kwargs.get("model"),
@@ -49,15 +50,17 @@ def litellm_handler():
             )
 
         def log_stream_event(self, kwargs, response_obj, start_time, end_time):
-            self._trace.update_span_status(span=self._trace.active_span, value="OK")
-            self._trace.end_span(
+            ag.tracing.set_status(status="OK")
+            ag.tracing.end_span(
                 outputs={
                     "message": kwargs.get(
                         "complete_streaming_response"
                     ),  # the complete streamed response (only set if `completion(..stream=True)`)
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -67,13 +70,15 @@ def litellm_handler():
         def log_success_event(
             self, kwargs, response_obj: ModelResponse, start_time, end_time
         ):
-            self._trace.update_span_status(span=self._trace.active_span, value="OK")
-            self._trace.end_span(
+            ag.tracing.set_status(status="OK")
+            ag.tracing.end_span(
                 outputs={
                     "message": response_obj.choices[0].message.content,
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -83,23 +88,25 @@ def litellm_handler():
         def log_failure_event(
             self, kwargs, response_obj: ModelResponse, start_time, end_time
         ):
-            self._trace.update_span_status(span=self._trace.active_span, value="ERROR")
-            self._trace.set_span_attribute(
+            ag.tracing.set_status(status="ERROR")
+            ag.tracing.set_attributes(
                 {
-                    "traceback_exception": kwargs[
-                        "traceback_exception"
-                    ],  # the traceback generated via `traceback.format_exc()`
+                    "traceback_exception": repr(
+                        kwargs["traceback_exception"]
+                    ),  # the traceback generated via `traceback.format_exc()`
                     "call_end_time": kwargs[
                         "end_time"
                     ],  # datetime object of when call was completed
                 },
             )
-            self._trace.end_span(
+            ag.tracing.end_span(
                 outputs={
                     "message": kwargs["exception"],  # the Exception raised
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -109,15 +116,17 @@ def litellm_handler():
         async def async_log_stream_event(
             self, kwargs, response_obj, start_time, end_time
         ):
-            self._trace.update_span_status(span=self._trace.active_span, value="OK")
-            self._trace.end_span(
+            ag.tracing.set_status(status="OK")
+            ag.tracing.end_span(
                 outputs={
                     "message": kwargs.get(
                         "complete_streaming_response"
                     ),  # the complete streamed response (only set if `completion(..stream=True)`)
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -127,13 +136,15 @@ def litellm_handler():
         async def async_log_success_event(
             self, kwargs, response_obj, start_time, end_time
         ):
-            self._trace.update_span_status(span=self._trace.active_span, value="OK")
-            self._trace.end_span(
+            ag.tracing.set_status(status="OK")
+            ag.tracing.end_span(
                 outputs={
                     "message": response_obj.choices[0].message.content,
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost
@@ -143,8 +154,8 @@ def litellm_handler():
         async def async_log_failure_event(
             self, kwargs, response_obj, start_time, end_time
         ):
-            self._trace.update_span_status(span=self._trace.active_span, value="ERROR")
-            self._trace.set_span_attribute(
+            ag.tracing.set_status(status="ERROR")
+            ag.tracing.set_attributes(
                 {
                     "traceback_exception": kwargs[
                         "traceback_exception"
@@ -154,12 +165,14 @@ def litellm_handler():
                     ],  # datetime object of when call was completed
                 },
             )
-            self._trace.end_span(
+            ag.tracing.end_span(
                 outputs={
-                    "message": kwargs["exception"],  # the Exception raised
-                    "usage": response_obj.usage.dict()
-                    if hasattr(response_obj, "usage")
-                    else None,  # litellm calculates usage
+                    "message": repr(kwargs["exception"]),  # the Exception raised
+                    "usage": (
+                        response_obj.usage.dict()
+                        if hasattr(response_obj, "usage")
+                        else None
+                    ),  # litellm calculates usage
                     "cost": kwargs.get(
                         "response_cost"
                     ),  # litellm calculates response cost

agenta/sdk/tracing/llm_tracing.py

@@ -196,7 +196,6 @@ class Tracing(metaclass=SingletonMeta):
         config: Optional[Dict[str, Any]] = None,
         **kwargs,
     ) -> CreateSpan:
-
         tracing = tracing_context.get()
 
         span_id = self._create_span_id()
@@ -415,7 +414,8 @@ class Tracing(metaclass=SingletonMeta):
             "trace",
             # mock_create_traces(
             self.client.create_traces(
-                trace=tracing.trace_id, spans=tracing.closed_spans  # type: ignore
+                trace=tracing.trace_id,
+                spans=tracing.closed_spans,  # type: ignore
             ),
             self.client,
         )

{agenta-0.19.6a0.dist-info → agenta-0.19.8.dist-info}/METADATA

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

{agenta-0.19.6a0.dist-info → agenta-0.19.8.dist-info}/RECORD

@@ -3,7 +3,7 @@ agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxu
 agenta/cli/helper.py,sha256=vRxHyeNaltzNIGrfU2vO0H28_rXDzx9QqIZ_S-W6zL4,6212
 agenta/cli/main.py,sha256=Wz0ODhoeKK3Qg_CFUhu6D909szk05tc8ZVBB6H1-w7k,9763
 agenta/cli/telemetry.py,sha256=GaFFRsE_NtrcSSJ10r2jhgFs5Sk8gf2C09Ox3gOr3eU,1317
-agenta/cli/variant_commands.py,sha256=NjTXOMaJWyZtwtJiU7Ldl47jTYeC7-vktar1GNO6FVY,17294
+agenta/cli/variant_commands.py,sha256=u2ThjbnE6qnNxNm1vZIXWHfHc-0JPmYnJ-AqpwtM8Ns,17350
 agenta/cli/variant_configs.py,sha256=PLiuMKadVzs6Gi2uYaT0pZzyULNHDXaTMDWboqpwWdU,1293
 agenta/client/Readme.md,sha256=zWJ6VMYCG124op5RcqgWBdJdlGkGQ2rPLk9F32rWvqo,2756
 agenta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -127,17 +127,17 @@ agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4J
 agenta/docker/docker-assets/main.py,sha256=7MI-21n81U7N7A0GxebNi0cmGWtJKcR2sPB6FcH2QfA,251
 agenta/docker/docker_utils.py,sha256=5uHMCzXkCvIsDdEiwbnnn97KkzsFbBvyMwogCsv_Z5U,3509
 agenta/sdk/__init__.py,sha256=cF0de6DiH-NZWEm0XvPN8_TeC1whPBnDf1WYYE1qK2g,762
-agenta/sdk/agenta_init.py,sha256=xZEuuSnKm1zgG7YFpMCMXxabbgVyPqnW6am0Qu3n3TU,9958
+agenta/sdk/agenta_init.py,sha256=8MfDuypxohd0qRTdtGjX7L17KW-1UGmzNVdiqF15_ak,9790
 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=umcniOOQfKVdPgHVb_jRhoGKei14Yc3PIZmEC8CU2Wg,22996
-agenta/sdk/decorators/tracing.py,sha256=_RCiT6VqOshUtFm7rZhv5mQGl5PioXPDJxb3N8EDOUY,6174
+agenta/sdk/decorators/tracing.py,sha256=c9LwQJkhJcyO7Uq-sNpDSwfwOUTAmlqNuJjz4bSx-k0,6172
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/tracing/callbacks.py,sha256=8XYcGC4EZQ0lBn2IWmtPH_yC5Hu6Yu-NkdZRuY7CoTw,6816
+agenta/sdk/tracing/callbacks.py,sha256=0rqkW-PGZSaNSMp_t4bGI7R9HQRgTyIy0gxmpGVJWpE,6915
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
-agenta/sdk/tracing/llm_tracing.py,sha256=k8SfZ7mo2oDUijne00lGkHh1AryzWydZEefT4RtnnWU,13690
+agenta/sdk/tracing/llm_tracing.py,sha256=OL9OZ9sKv58hAsSxf33E-VYFx2ZeJpr9jqe3Hvz1CX8,13706
 agenta/sdk/tracing/logger.py,sha256=GfH7V-jBHcn7h5dbdrnkDMe_ml3wkXFBeoQiqR4KVRc,474
 agenta/sdk/tracing/tasks_manager.py,sha256=ROrWIaqS2J2HHiJtRWiHKlLY8CCsqToP5VeXu7mamck,3748
 agenta/sdk/tracing/tracing_context.py,sha256=EOi1zfqpb2cBjhBtHIphUkVHi4jiWes-RRNdbgk1kMc,906
@@ -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.19.6a0.dist-info/METADATA,sha256=0uvnsMgJUeCyZC51VbA-_C67WR4gpzEYNfKQ5onan1E,26462
-agenta-0.19.6a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.19.6a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.19.6a0.dist-info/RECORD,,
+agenta-0.19.8.dist-info/METADATA,sha256=OAjZGmsi3_b62yrLWwmmtJqRMi_wa9d1vGBRuN_6NRg,26460
+agenta-0.19.8.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.19.8.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.19.8.dist-info/RECORD,,