agenta 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

agenta/client/client.py

@@ -134,7 +134,9 @@ def send_docker_tar(
     with tar_path.open("rb") as tar_file:
         response = requests.post(
             f"{host}/{BACKEND_URL_SUFFIX}/containers/build_image/?app_name={app_name}&variant_name={variant_name}",
-            files={"tar_file": tar_file,},
+            files={
+                "tar_file": tar_file,
+            },
             timeout=1200,
         )
 

agenta/docker/docker_utils.py

@@ -106,7 +106,8 @@ def build_and_upload_docker_image(
         # TODO: Later do this in the temp dir
         dockerfile_path = create_dockerfile(folder)
         shutil.copytree(
-            Path(__file__).parent.parent / "sdk", folder / "agenta",
+            Path(__file__).parent.parent / "sdk",
+            folder / "agenta",
         )
         shutil.copy(Path(__file__).parent / "docker-assets" / "main.py", folder)
         shutil.copy(Path(__file__).parent / "docker-assets" / "entrypoint.sh", folder)

agenta/sdk/agenta.py

@@ -83,7 +83,8 @@ def ingest(func: Callable[..., Any]):
                 traceback.format_exception(None, e, e.__traceback__)
             )
             return JSONResponse(
-                status_code=500, content={"error": str(e), "traceback": traceback_str},
+                status_code=500,
+                content={"error": str(e), "traceback": traceback_str},
             )
 
     new_params = []
@@ -127,7 +128,9 @@ def ingest(func: Callable[..., Any]):
             if name in app_params:
                 # For optional parameters, we add them as options
                 parser.add_argument(
-                    f"--{name}", type=type(param.default), default=param.default,
+                    f"--{name}",
+                    type=type(param.default),
+                    default=param.default,
                 )
             elif name in ingestible_files:
                 parser.add_argument(name, type=str)
@@ -177,7 +180,8 @@ def post(func: Callable[..., Any]):
                     traceback.format_exception(e, value=e, tb=e.__traceback__)
                 )
             return JSONResponse(
-                status_code=500, content={"error": str(e), "traceback": traceback_str},
+                status_code=500,
+                content={"error": str(e), "traceback": traceback_str},
             )
 
     new_params = []
@@ -230,7 +234,9 @@ def post(func: Callable[..., Any]):
                     )
                 else:
                     parser.add_argument(
-                        f"--{name}", type=type(param.default), default=param.default,
+                        f"--{name}",
+                        type=type(param.default),
+                        default=param.default,
                     )
             else:
                 # For required parameters, we add them as arguments

agenta/sdk/types.py

@@ -39,7 +39,12 @@ class IntParam(int):
     @classmethod
     def __modify_schema__(cls, field_schema):
         field_schema.update(
-            {"x-parameter": "int", "type": "integer", "minimum": 1, "maximum": 10,}
+            {
+                "x-parameter": "int",
+                "type": "integer",
+                "minimum": 1,
+                "maximum": 10,
+            }
         )
 
 
@@ -53,7 +58,12 @@ class FloatParam(float):
     @classmethod
     def __modify_schema__(cls, field_schema):
         field_schema.update(
-            {"x-parameter": "float", "type": "number", "minimum": 0.0, "maximum": 1.0,}
+            {
+                "x-parameter": "float",
+                "type": "number",
+                "minimum": 0.0,
+                "maximum": 1.0,
+            }
         )
 
 
@@ -79,7 +89,11 @@ class MultipleChoiceParam(str):
     @classmethod
     def __modify_schema__(cls, field_schema: dict[str, Any]):
         field_schema.update(
-            {"x-parameter": "choice", "type": "string", "enum": [],}
+            {
+                "x-parameter": "choice",
+                "type": "string",
+                "enum": [],
+            }
         )
 
 

agenta/templates/simple_prompt/app.py

@@ -13,7 +13,10 @@ def generate(
     prompt_template: ag.TextParam = default_prompt,
 ) -> str:
     llm = OpenAI(temperature=temperature)
-    prompt = PromptTemplate(input_variables=["product"], template=prompt_template,)
+    prompt = PromptTemplate(
+        input_variables=["product"],
+        template=prompt_template,
+    )
     chain = LLMChain(llm=llm, prompt=prompt)
     output = chain.run(product=product)
 

{agenta-0.2.0.dist-info → agenta-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.2.0
+Version: 0.2.2
 Summary: Code for the SDK and CLI for the Agenta Lab platform
 Author: Mahmoud Mabrouk
 Author-email: mahmoudmabrouk.mail@gmail.com

{agenta-0.2.0.dist-info → agenta-0.2.2.dist-info}/RECORD

@@ -5,26 +5,26 @@ agenta/cli/variant_commands.py,sha256=Rw45rhEOWMar7qnPrb_yAiDtRIsCgHOf8Y-FVNxsYv
 agenta/client/Readme.md,sha256=umhMce1Gq_der9pH4M_pP4NQ8rHa3MENJLM9OrdUZ50,131
 agenta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/client/api_models.py,sha256=gqIIZay_A7kg2d451gO9QTgaKxGuAk3YKtWtArIzMNw,325
-agenta/client/client.py,sha256=OAeTERiYjC5-snjfyI52gZYKjO45Ikx60_mIT1XYQ3Y,5416
+agenta/client/client.py,sha256=iFQKVq2zZ0sw7Ph246byZdwY0GhKXFM0NaQfF07i0CI,5446
 agenta/config.py,sha256=toEpBx8PjBC6P_22vLRlyMn0gu-P4CR0qPTgrUMFoiw,630
 agenta/config.toml,sha256=un4IS3NUcJczZS2LrjPCOShx8W9aw0XIoabZKfqZAWw,198
 agenta/docker/docker-assets/Dockerfile.template,sha256=hIEfOD58RdQ56GbyNZGu3vsDSAknbA44lQWxWLRVMbo,282
 agenta/docker/docker-assets/README.md,sha256=XHxwh2ks_ozrtAU7SLbL3J14SB2holG6buoTxwmMiZM,102
 agenta/docker/docker-assets/entrypoint.sh,sha256=29XK8VQjQsx4hN2j-4JDy-6kQb5y4LCqZEa7PD4eqCQ,74
 agenta/docker/docker-assets/main.py,sha256=BbmbFByRQ8MzL8402pJryEF34t6ba1i_JrMtWMAQDQ4,327
-agenta/docker/docker_utils.py,sha256=3jr1EiDX2wyN6--9h2zwyDdWMxH68-L14RIiv6fZ_Qc,4949
+agenta/docker/docker_utils.py,sha256=N-tmbHp6XpH1azevJanxf8qVEOE7wm4F5cZs4-5zoH4,4961
 agenta/sdk/__init__.py,sha256=L1gRP0s2LPgCokt1NSg80pY8nTClQvBH0Fv6FCIo2wc,316
-agenta/sdk/agenta.py,sha256=MHiYeir0-QeRoWa5Xn5geLu2LD6wEF2vif42wWWk-zQ,10630
+agenta/sdk/agenta.py,sha256=6cx5aJQyxlQqvmeoLqeJ3-tforqL1ItFTo5jDAEj3eg,10750
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/init.py,sha256=0CuErDz4uFEWN8Av_yIIGltHR4KpWbnHMucMZ7uSgRk,87
 agenta/sdk/router.py,sha256=zBuwaFCNDnTltzK_edrb19BeZfLad1smCiB2zIBOd5U,202
-agenta/sdk/types.py,sha256=mXLkt5CqD2Nn-e3GedLnRG313Meox4WlubRGwZ2t2y8,2860
+agenta/sdk/types.py,sha256=79vEANLaygI6PtXoNWnC1Y4ieT8qXDoRdTaE-fZeypQ,3078
 agenta/templates/simple_prompt/README.md,sha256=ss7vZPpI1Hg0VmYtFliwq_r5LnqbCy_S5OQDXg8UoIA,308
-agenta/templates/simple_prompt/app.py,sha256=Z7pjNxDoP2xbzM8FRpMXf5Ha_nmDhIdqbrD1oQTLLo0,586
+agenta/templates/simple_prompt/app.py,sha256=0Q-h-GQ4gOU_uBKJnT7xo-j3EUAqb7rHaGinI-Wdh9I,608
 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.2.0.dist-info/METADATA,sha256=7zXYqmQnepaFnZOaTIYaLDaE5h2GSwRMTGsbqWOWavE,1026
-agenta-0.2.0.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-agenta-0.2.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.2.0.dist-info/RECORD,,
+agenta-0.2.2.dist-info/METADATA,sha256=Dp-P5rVQxs0Txi1wxeHEcjto90fZSoQ3UAdlkbl9XDM,1026
+agenta-0.2.2.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+agenta-0.2.2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.2.2.dist-info/RECORD,,