agenta 0.29.0__py3-none-any.whl → 0.30.0a1__py3-none-any.whl

agenta/sdk/decorators/routing.py

@@ -74,10 +74,13 @@ class route:
         self.route_path = path
 
     def __call__(self, f):
+        # If not running in Agenta, return the original function unchanged
+        if environ.get("AGENTA_RUNTIME") != "true":
+            return f
+
         self.e = entrypoint(
             f, route_path=self.route_path, config_schema=self.config_schema
         )
-
         return f
 
 
@@ -120,6 +123,23 @@ class entrypoint:
         route_path="",
         config_schema: Optional[BaseModel] = None,
     ):
+        self.func = func
+        self.route_path = route_path
+        self.config_schema = config_schema
+
+    def __call__(self, func=None):
+        if func is None:
+            func = self.func
+            route_path = self.route_path
+            config_schema = self.config_schema
+        else:
+            route_path = ""
+            config_schema = None
+
+        # If not running in Agenta, return the original function unchanged
+        if environ.get("AGENTA_RUNTIME") != "true":
+            return func
+
         ### --- Update Middleware --- #
         try:
             global _MIDDLEWARES  # pylint: disable=global-statement
@@ -172,6 +192,11 @@ class entrypoint:
         ### --- Playground  --- #
         @wraps(func)
         async def wrapper(*args, **kwargs) -> Any:
+            if environ.get("AGENTA_RUNTIME") != "true":
+                raise HTTPException(
+                    status_code=403,
+                    detail="This endpoint is only available when running in Agenta environment",
+                )
             func_params, api_config_params = self.split_kwargs(kwargs, config_params)
             self.ingest_files(func_params, ingestible_files)
             if not config_schema:
@@ -234,6 +259,11 @@ class entrypoint:
         ### --- Deployed --- #
         @wraps(func)
         async def wrapper_deployed(*args, **kwargs) -> Any:
+            if environ.get("AGENTA_RUNTIME") != "true":
+                raise HTTPException(
+                    status_code=403,
+                    detail="This endpoint is only available when running in Agenta environment",
+                )
             func_params = {
                 k: v
                 for k, v in kwargs.items()
@@ -306,14 +336,6 @@ class entrypoint:
                     config=route["config"],
                 )
 
-        if self.is_main_script(func) and route_path == "":
-            self.handle_terminal_run(
-                func,
-                func_signature.parameters,  # type: ignore
-                config_params,
-                ingestible_files,
-            )
-
     def extract_ingestible_files(
         self,
         func_signature: Signature,
@@ -615,99 +637,6 @@ class entrypoint:
         """
         return func.__module__ == "__main__"
 
-    def handle_terminal_run(
-        self,
-        func: Callable,
-        func_params: Dict[str, Parameter],
-        config_params: Dict[str, Any],
-        ingestible_files: Dict,
-    ):
-        """
-        Parses command line arguments and sets configuration when script is run from the terminal.
-
-        Args:
-            func_params (dict): A dictionary containing the function parameters and their annotations.
-            config_params (dict): A dictionary containing the configuration parameters.
-            ingestible_files (dict): A dictionary containing the files that should be ingested.
-        """
-
-        # For required parameters, we add them as arguments
-        parser = ArgumentParser()
-        for name, param in func_params.items():
-            if name in ingestible_files:
-                parser.add_argument(name, type=str)
-            else:
-                parser.add_argument(name, type=param.annotation)
-
-        for name, param in config_params.items():
-            if type(param) is MultipleChoiceParam:
-                parser.add_argument(
-                    f"--{name}",
-                    type=str,
-                    default=param.default,
-                    choices=param.choices,  # type: ignore
-                )
-            else:
-                parser.add_argument(
-                    f"--{name}",
-                    type=type(param),
-                    default=param,
-                )
-
-        args = parser.parse_args()
-
-        # split the arg list into the arg in the app_param and
-        # the args from the sig.parameter
-        args_config_params = {k: v for k, v in vars(args).items() if k in config_params}
-        args_func_params = {
-            k: v for k, v in vars(args).items() if k not in config_params
-        }
-        for name in ingestible_files:
-            args_func_params[name] = InFile(
-                file_name=Path(args_func_params[name]).stem,
-                file_path=args_func_params[name],
-            )
-
-        # Update args_config_params with default values from config_params if not provided in command line arguments
-        args_config_params.update(
-            {
-                key: value
-                for key, value in config_params.items()
-                if key not in args_config_params
-            }
-        )
-
-        loop = get_event_loop()
-
-        with routing_context_manager(config=args_config_params):
-            result = loop.run_until_complete(
-                self.execute_function(
-                    func,
-                    True,  # inline trace: True
-                    **{"params": args_func_params, "config_params": args_config_params},
-                )
-            )
-
-        if result.trace:
-            log.info("\n========= Result =========\n")
-
-            log.info(f"trace_id: {result.trace['trace_id']}")
-            log.info(f"latency:  {result.trace.get('latency')}")
-            log.info(f"cost:     {result.trace.get('cost')}")
-            log.info(f"usage:   {list(result.trace.get('usage', {}).values())}")
-
-            log.info(" ")
-            log.info("data:")
-            log.info(dumps(result.data, indent=2))
-
-            log.info(" ")
-            log.info("trace:")
-            log.info("----------------")
-            log.info(dumps(result.trace.get("spans", []), indent=2))
-            log.info("----------------")
-
-            log.info("\n==========================\n")
-
     def override_config_in_schema(
         self,
         openapi_schema: dict,

{agenta-0.29.0.dist-info → agenta-0.30.0a1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.29.0
+Version: 0.30.0a1
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -13,8 +13,7 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.9
 Classifier: Topic :: Software Development :: Libraries
 Requires-Dist: cachetools (>=5.3.3,<6.0.0)
 Requires-Dist: click (>=8.1.3,<9.0.0)

{agenta-0.29.0.dist-info → agenta-0.30.0a1.dist-info}/RECORD

@@ -181,7 +181,7 @@ agenta/sdk/context/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuF
 agenta/sdk/context/routing.py,sha256=ycUgmJZyWhL4bHjKtUSAsTlt_0Fujr_6OpoaEH1lAN0,683
 agenta/sdk/context/tracing.py,sha256=UmmW15UFFsvxS0myS6aD9wBk5iNepNlQi4tEQ_ejfYM,96
 agenta/sdk/decorators/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agenta/sdk/decorators/routing.py,sha256=LN9UJO30rJESnw3yokISKjYHbkiMmycbqa8jonEz5mM,36712
+agenta/sdk/decorators/routing.py,sha256=wtBLzk075DlMsVfsAEDg-LU1lkqgqAm1Fc6bFBg7jXM,34190
 agenta/sdk/decorators/tracing.py,sha256=vL5e6TVX6TQwO0t9raZwnzXHV3vElVT0pHS1vD-vzEo,8523
 agenta/sdk/litellm/__init__.py,sha256=Bpz1gfHQc0MN1yolWcjifLWznv6GjHggvRGQSpxpihM,37
 agenta/sdk/litellm/litellm.py,sha256=Ke0It-jA1z0KQ2770gIlWIEgramZGmt1k0GjmpEnFV4,8793
@@ -227,7 +227,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.29.0.dist-info/METADATA,sha256=uOLDR3yhSz76rdYHC5gJqveqkpbqOheJb6exYHKPUeA,29042
-agenta-0.29.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-agenta-0.29.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.29.0.dist-info/RECORD,,
+agenta-0.30.0a1.dist-info/METADATA,sha256=155SCLF7rcKPbx6G51p_Lm1pk3xj6OlHbTRN_U0Q6FU,28992
+agenta-0.30.0a1.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+agenta-0.30.0a1.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.30.0a1.dist-info/RECORD,,

{agenta-0.29.0.dist-info → agenta-0.30.0a1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 1.5.2
 Root-Is-Purelib: true
 Tag: py3-none-any