agenta 0.2.0__tar.gz → 0.2.2__tar.gz

{agenta-0.2.0 → agenta-0.2.2}/PKG-INFO

@@ -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 → agenta-0.2.2}/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-0.2.0 → agenta-0.2.2}/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-0.2.0 → agenta-0.2.2}/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-0.2.0 → agenta-0.2.2}/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-0.2.0 → agenta-0.2.2}/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 → agenta-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.2.0"
+version = "0.2.2"
 description = "Code for the SDK and CLI for the Agenta Lab platform"
 authors = ["Mahmoud Mabrouk <mahmoudmabrouk.mail@gmail.com>"]
 readme = "README.md"