fairo 25.8.1__tar.gz → 25.8.2__tar.gz

{fairo-25.8.1 → fairo-25.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.8.1
+Version: 25.8.2
 Summary: SDK for interfacing with Fairo SaaS platform.
 Author-email: "Fairo Systems, Inc." <support@fairo.ai>
 License: Apache-2.0
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Description-Content-Type: text/markdown
-Requires-Dist: mlflow<=3.1.1,>=3.1.0
+Requires-Dist: mlflow<=3.2.0,>=3.1.0
 Requires-Dist: langchain<0.4.0,>=0.3.20
 Requires-Dist: langchain-aws<0.3.0,>=0.2.18
 Requires-Dist: langchain-community<0.4.0,>=0.3.20

fairo-25.8.2/fairo/__init__.py

@@ -0,0 +1 @@
+__version__ = "25.8.2"

{fairo-25.8.1 → fairo-25.8.2}/fairo/core/execution/executor.py

@@ -23,6 +23,7 @@ logger = logging.getLogger(__name__)
 class FairoExecutor:
     def __init__(
         self,
+        input_schema,
         agent_type: str = "Langchain",
         agents: List[Any] = [],
         verbose: bool = False,
@@ -37,6 +38,7 @@ class FairoExecutor:
             raise ValueError("FairoExecutor cannot be initialized with both 'agents' and 'runnable'. Please provide only one.")
         if not input_fields:
             raise ValueError("Missing input_fields")
+        self.input_schema = input_schema
         self.agents = agents
         self.agent_type = agent_type
         self.verbose = verbose
@@ -67,8 +69,26 @@ class FairoExecutor:
             pipeline = mlflow.pyfunc.load_model(self.runnable.artifact_path)
         else:
             pipeline = AgentChainWrapper(chain_class=self.chain_class, agent_functions_list=self.agents)
-            cols = [ColSpec(type="string", name=field) for field in self.input_fields]
-            input_schema = Schema(cols)
+            # Convert Pydantic schema to MLflow Schema
+            if hasattr(self.input_schema, 'model_json_schema'):
+                # Extract field names from Pydantic schema
+                pydantic_schema = self.input_schema.model_json_schema()
+                properties = pydantic_schema.get('properties', {})
+                cols = []
+                for field_name, field_info in properties.items():
+                    field_type = field_info.get('type', 'string')
+                    # Map Pydantic types to MLflow types
+                    mlflow_type = 'string'  # Default to string
+                    if field_type in ['integer', 'number']:
+                        mlflow_type = 'double'
+                    elif field_type == 'boolean':
+                        mlflow_type = 'boolean'
+                    cols.append(ColSpec(type=mlflow_type, name=field_name))
+                input_schema = Schema(cols)
+            else:
+                # Fallback to input_fields if schema is not Pydantic
+                cols = [ColSpec(type="string", name=field) for field in self.input_fields]
+                input_schema = Schema(cols)
 
             output_schema = Schema([
                 ColSpec(type="string", name="output"),
@@ -97,7 +117,8 @@ class FairoExecutor:
             fairo_settings = {
                 "type": type,
                 "process_graph": process_graph,
-                "input_schema": self.input_fields
+                "schema": self.input_schema.model_json_schema(),
+                "input_fields": list(self.input_schema.model_fields.keys()),
             }
             if process_graph:
                 mlflow.log_text(json.dumps(fairo_settings, ensure_ascii=False, indent=2), artifact_file="fairo_settings.txt")

{fairo-25.8.1 → fairo-25.8.2}/fairo/core/workflow/utils.py

@@ -278,6 +278,7 @@ def output_langchain_process_graph(agents):
                 "prompt": get_agent_prompt(agent),
                 "prefix": agent_kwargs.get("prefix", ""),
                 "suffix": agent_kwargs.get("suffix", ""),
+                "schema": get_agent_schema(agent),
             }
         )
 
@@ -457,4 +458,31 @@ def get_agent_prompt(agent) -> str:
                     search_queue.append(child)
 
     # Nothing found – return empty string
-    return ""
+    return ""
+
+
+def get_agent_schema(agent):
+    """
+    Returns a JSON schema dict when available, otherwise None.
+    Checks common attributes across SimpleAgent and LangChain agents.
+    """
+    candidate_attrs = ("args_schema", "input_schema", "agent_schema", "schema")
+    for attr in candidate_attrs:
+        obj = getattr(agent, attr, None)
+        if obj is None:
+            continue
+        try:
+            if isinstance(obj, dict):
+                return obj
+            if hasattr(obj, "model_json_schema") and callable(getattr(obj, "model_json_schema")):
+                return obj.model_json_schema()
+            if hasattr(obj, "schema") and callable(getattr(obj, "schema")):
+                return obj.schema()
+        except Exception as e:
+            pass
+
+    inner = getattr(agent, "agent", None)
+    if inner is not None and inner is not agent:
+        return get_agent_schema(inner)
+
+    return None

{fairo-25.8.1 → fairo-25.8.2}/fairo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.8.1
+Version: 25.8.2
 Summary: SDK for interfacing with Fairo SaaS platform.
 Author-email: "Fairo Systems, Inc." <support@fairo.ai>
 License: Apache-2.0
@@ -11,7 +11,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Operating System :: OS Independent
 Description-Content-Type: text/markdown
-Requires-Dist: mlflow<=3.1.1,>=3.1.0
+Requires-Dist: mlflow<=3.2.0,>=3.1.0
 Requires-Dist: langchain<0.4.0,>=0.3.20
 Requires-Dist: langchain-aws<0.3.0,>=0.2.18
 Requires-Dist: langchain-community<0.4.0,>=0.3.20

{fairo-25.8.1 → fairo-25.8.2}/fairo.egg-info/requires.txt

@@ -1,4 +1,4 @@
-mlflow<=3.1.1,>=3.1.0
+mlflow<=3.2.0,>=3.1.0
 langchain<0.4.0,>=0.3.20
 langchain-aws<0.3.0,>=0.2.18
 langchain-community<0.4.0,>=0.3.20

{fairo-25.8.1 → fairo-25.8.2}/pyproject.toml

@@ -20,7 +20,7 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-    "mlflow>=3.1.0,<=3.1.1",
+    "mlflow>=3.1.0,<=3.2.0",
     "langchain>=0.3.20,<0.4.0",
     "langchain-aws>=0.2.18,<0.3.0",
     "langchain-community>=0.3.20,<0.4.0",

fairo-25.8.1/fairo/__init__.py

@@ -1 +0,0 @@
-__version__ = "25.8.1"