fairo 25.6.1__tar.gz → 25.6.2__tar.gz

{fairo-25.6.1 → fairo-25.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.6.1
+Version: 25.6.2
 Summary: SDK for interfacing with Fairo SaaS platform.
 Author-email: "Fairo Systems, Inc." <support@fairo.ai>
 License: Apache-2.0

{fairo-25.6.1 → fairo-25.6.2}/fairo/core/execution/executor.py

@@ -7,9 +7,13 @@ import logging
 
 import mlflow
 
+from mlflow.models.signature import ModelSignature
+from mlflow.types.schema import Schema, ColSpec
+from fairo.core.agent.base_agent import SimpleAgent
 from fairo.core.client.client import BaseClient
 from fairo.core.execution.env_finder import read_variables
 from fairo.core.runnable.runnable import Runnable
+from fairo.core.workflow.utils import output_workflow_process_graph
 from fairo.settings import get_fairo_api_key, get_fairo_api_secret, get_mlflow_experiment_name, get_mlflow_server, get_fairo_base_url
 
 # Optional interfaces/types
@@ -95,25 +99,51 @@ class FairoExecutor:
             runnables.append(
                 self._wrap_agent_runnable(agent, agent.input_key, agent.output_key)
             )
-
+        first_input_key = self.agents[0].input_key
+        last_output_key = self.agents[-1].output_key
         # Build RunnableSequence from all steps
         pipeline = runnables[0]
         for r in runnables[1:]:
             pipeline = pipeline | r  # chaining
         
+        def save_process_graph():
+            process_graph = (
+                output_workflow_process_graph(self.agents)
+                if all(isinstance(agent, SimpleAgent) for agent in self.agents)
+                else None
+            )
+            if len(self.agents) > 1:
+                type = "Workflow"
+            else:
+                type = "Agent"
+            fairo_settings = {
+                "type": type,
+                "process_graph": process_graph,
+            }
+            if process_graph:
+                mlflow.log_text(json.dumps(fairo_settings, ensure_ascii=False, indent=2), artifact_file="fairo_settings.txt")
+        save_process_graph()
         try:
             # Find environment variables used in the project
             all_env_vars = read_variables()
             # Log the file as an artifact
-            mlflow.log_text(all_env_vars, artifact_file="environment/variables.txt")         
+            mlflow.log_text(all_env_vars, artifact_file="environment/variables.txt")
             if self.verbose:
                 logger.info(f"Logged {len(all_env_vars)} environment variables as artifact")
         except Exception as e:
             logger.warning(f"Failed to log environment variables: {str(e)}")
 
+        input_schema = Schema([
+            ColSpec(type="string", name=first_input_key),
+        ])
+
+        output_schema = Schema([
+            ColSpec(type="string", name=last_output_key),
+        ])
         model_info = mlflow.langchain.log_model(
             pipeline,
-            artifact_path="agent",
+            artifact_path="",
+            signature=ModelSignature(inputs=input_schema, outputs=output_schema)
         )
         # If runnable object was added, check if it exists, if yes, just set tags for the trace, otherwise create it
         if self.runnable:

{fairo-25.6.1 → fairo-25.6.2}/fairo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.6.1
+Version: 25.6.2
 Summary: SDK for interfacing with Fairo SaaS platform.
 Author-email: "Fairo Systems, Inc." <support@fairo.ai>
 License: Apache-2.0