agenta 0.14.6a0__py3-none-any.whl → 0.14.7a0__py3-none-any.whl

agenta/cli/variant_commands.py

@@ -104,10 +104,10 @@ def add_variant(
     )
 
     if variant_name in config["variants"] and not overwrite:
-        overwrite = questionary.confirm(
+        overwrite_question = questionary.confirm(
             "This variant already exists. Do you want to overwrite it?"
         ).ask()
-        if not overwrite:
+        if not overwrite_question:
             click.echo("Operation cancelled.")
             sys.exit(0)
 

agenta/sdk/decorators/llm_entrypoint.py

@@ -53,7 +53,7 @@ app.include_router(router, prefix="")
 
 
 class entrypoint(BaseDecorator):
-    """Decorator class to wrap a function for HTTP POST, and terminal exposure.
+    """Decorator class to wrap a function for HTTP POST, terminal exposure and enable tracing.
 
     Args:
         BaseDecorator (object): base decorator class
@@ -62,18 +62,26 @@ class entrypoint(BaseDecorator):
     ```python
         import agenta as ag
 
-        @ag.entrypoint()
+        @ag.entrypoint(enable_tracing=True) # Defaults to False
         async def chain_of_prompts_llm(prompt: str):
             return ...
     ```
     """
 
-    def __call__(self, func: Callable[..., Any]) -> Callable[..., Any]:
+    def __init__(self, enable_tracing: bool = False):
+        self.enable_tracing = enable_tracing
+
+    def __call__(self, func: Callable[..., Any]):
         endpoint_name = "generate"
         func_signature = inspect.signature(func)
         config_params = agenta.config.all()
         ingestible_files = self.extract_ingestible_files(func_signature)
 
+        if self.enable_tracing:
+            from .tracing import trace
+
+            func = trace()(func)
+
         @functools.wraps(func)
         async def wrapper(*args, **kwargs) -> Any:
             func_params, api_config_params = self.split_kwargs(kwargs, config_params)
@@ -123,14 +131,13 @@ class entrypoint(BaseDecorator):
         )
 
         if self.is_main_script(func):
-            self.handle_terminal_run(
+            result = self.handle_terminal_run(
                 func,
                 func_signature.parameters,  # type: ignore
                 config_params,
                 ingestible_files,
             )
-
-        return wrapper
+            return result
 
     def extract_ingestible_files(
         self,
@@ -325,7 +332,7 @@ class entrypoint(BaseDecorator):
         func_params: Dict[str, inspect.Parameter],
         config_params: Dict[str, Any],
         ingestible_files: Dict,
-    ) -> None:
+    ):
         """
         Parses command line arguments and sets configuration when script is run from the terminal.
 
@@ -384,6 +391,7 @@ class entrypoint(BaseDecorator):
         print(
             f"\n========== Result ==========\n\nMessage: {result.message}\nCost: {result.cost}\nToken Usage: {result.usage}"
         )
+        return result
 
     def override_schema(
         self, openapi_schema: dict, func_name: str, endpoint: str, params: dict

agenta/sdk/decorators/tracing.py

@@ -84,7 +84,7 @@ class trace(BaseDecorator):
                 name=func.__name__,
                 inputs=kwargs,
                 config=ag.config.all(),
-                environment=os.environ.get("AGENTA_LLM_RUN_PLAYGROUND"),  # type: ignore
+                environment=os.environ.get("AGENTA_LLM_RUN_ENVIRONMENT"),  # type: ignore
             )
             try:
                 is_coroutine_function = inspect.iscoroutinefunction(func)

{agenta-0.14.6a0.dist-info → agenta-0.14.7a0.dist-info}/METADATA

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

{agenta-0.14.6a0.dist-info → agenta-0.14.7a0.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=GgYu6UsrnHbqPV7zPlO14b61IyaDiTIjGMYQS9DlqC4,9551
 agenta/cli/telemetry.py,sha256=GaFFRsE_NtrcSSJ10r2jhgFs5Sk8gf2C09Ox3gOr3eU,1317
-agenta/cli/variant_commands.py,sha256=9Ac2gB_2wokabxEgtY85xbYMt078JFfOXyi6GSkZ3rI,17268
+agenta/cli/variant_commands.py,sha256=clqmv-UL2UKjOSVGn1icOXSEJ3EMf43-OphuHNWknCQ,17286
 agenta/cli/variant_configs.py,sha256=PLiuMKadVzs6Gi2uYaT0pZzyULNHDXaTMDWboqpwWdU,1293
 agenta/client/Readme.md,sha256=zWJ6VMYCG124op5RcqgWBdJdlGkGQ2rPLk9F32rWvqo,2756
 agenta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -131,8 +131,8 @@ agenta/sdk/agenta_init.py,sha256=r28xgllqHe0Zs3Xz4UU7DoBMLF1tbPEbaqDi5irSiC8,863
 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=VU41C20ocPWm2pT5RCGpgj4cpT_ZRsgLmVteRgCx0sg,18586
-agenta/sdk/decorators/tracing.py,sha256=5Y2Jb0v07BUDlJyBuIDCZ3oIcF1Mp6rNCqQzDIW3Skk,3299
+agenta/sdk/decorators/llm_entrypoint.py,sha256=RU_Se5RcS_JBrpd4Sst54WLrEzxxVHhmkrXwv6nZxW8,18849
+agenta/sdk/decorators/tracing.py,sha256=9-NZPo52pC9i6vbGaoAcmU4J-fFYA1U6eyPWdbm6esU,3300
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/tracing/context_manager.py,sha256=HskDaiORoOhjeN375gm05wYnieQzh5UnoIsnSAHkAyc,252
 agenta/sdk/tracing/llm_tracing.py,sha256=s8sHojV1ncYEg1uAWcm3BsfPd_6yaOPvB03Tq1FnEMU,8709
@@ -157,7 +157,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.14.6a0.dist-info/METADATA,sha256=ckpAYSrWyEdhVcfNPPTus0gpwFqU1boT3YslWPUncM4,26472
-agenta-0.14.6a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-agenta-0.14.6a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.14.6a0.dist-info/RECORD,,
+agenta-0.14.7a0.dist-info/METADATA,sha256=FZ1CBOL2QI1sj6R650zq7V9fbHfIGSs7zGE79opaooU,26472
+agenta-0.14.7a0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+agenta-0.14.7a0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.14.7a0.dist-info/RECORD,,