fairo 25.9.7__tar.gz → 25.10.1__tar.gz

{fairo-25.9.7 → fairo-25.10.1}/PKG-INFO

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

fairo-25.10.1/fairo/__init__.py

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

{fairo-25.9.7 → fairo-25.10.1}/fairo/core/execution/agent_serializer.py

@@ -1,5 +1,6 @@
 from typing import Any, Dict, Optional
 import mlflow
+import json
 from langchain.callbacks.base import BaseCallbackHandler
 from langchain_core.callbacks import CallbackManagerForChainRun
 from pathlib import Path
@@ -77,12 +78,12 @@ class CustomPythonModel(mlflow.pyfunc.PythonModel):
         except ImportError as e:
             raise ImportError(f"Failed to import agent_code: {e}")
     
-    def predict(self, context, model_input):
+    def predict(self, context, model_input: list[str]):
         if isinstance(model_input, list):
-            return [self.agent.run(query) for query in model_input]
+            parsed_data = json.loads(model_input[0])
+            return self.run(parsed_data, callback_enabled=True)
         else:
-            return self.agent.run(model_input)
-
+            return self.run(model_input)
 class AgentChainWrapper:
     def __init__(self, chain_class = SimpleSequentialChain, agent_functions_list = [], callback_enabled = False):
         self.chain_class = chain_class
@@ -182,9 +183,12 @@ class AgentChainWrapper:
         )
         return chain.run(result)
     
-    def predict(self, context = "", model_input = "", callback_enabled: Optional[bool] = False):
-        self.callback_enabled = callback_enabled
-        return self.run(model_input)
+    def predict(self, context = "", model_input: list[str] = [""]):
+        if isinstance(model_input, list):
+            parsed_data = json.loads(model_input[0])
+            return self.run(parsed_data, callback_enabled=True)
+        else:
+            return self.run(model_input)
 
 class CustomChainModel(mlflow.pyfunc.PythonModel):
     def __init__(self):
@@ -253,9 +257,10 @@ class CustomChainModel(mlflow.pyfunc.PythonModel):
         # Create the agent chain
         self.agent_chain = AgentChainWrapper(agent_functions_list=agent_functions, callback_enabled=True)
     
-    def predict(self, context, model_input, callback_enabled: Optional[bool] = False):
+    def predict(self, context, model_input: list[str]):
         if isinstance(model_input, list):
-            return [self.agent_chain.run(query) for query in model_input]
+            parsed_data = json.loads(model_input[0])
+            return self.agent_chain.run(parsed_data)
         else:
             return self.agent_chain.run(model_input)
 
@@ -298,7 +303,7 @@ class CrewAgentWrapper:
             print(f"Error running CrewAI crew: {e}")
             return f"Error executing query '{query}': {str(e)}"
     
-    def predict(self, context, model_input):
+    def predict(self, context, model_input: list[str]):
         return self.run(model_input)
 
 class CustomCrewModel(mlflow.pyfunc.PythonModel):
@@ -341,7 +346,7 @@ class CustomCrewModel(mlflow.pyfunc.PythonModel):
         except ImportError as e:
             raise ImportError(f"Failed to import CrewAI agent_code: {e}")
     
-    def predict(self, context, model_input):
+    def predict(self, context, model_input: list[str]):
         if isinstance(model_input, list):
             return [self.agent.run(query) for query in model_input]
         else:

{fairo-25.9.7 → fairo-25.10.1}/fairo/core/execution/executor.py

@@ -35,7 +35,8 @@ class FairoExecutor:
         chain_class = SimpleSequentialChain,
         input_fields: List[str] = [],
         input_schema: Optional[Type[BaseModel]] = None,
-        chat_suggestions: Optional[ChatSuggestions] = None
+        chat_suggestions: Optional[ChatSuggestions] = None,
+        debug_mode: bool = False
     ):
         if agents and runnable:
             raise ValueError("FairoExecutor cannot be initialized with both 'agents' and 'runnable'. Please provide only one.")
@@ -45,10 +46,12 @@ class FairoExecutor:
         self.agents = agents
         self.agent_type = agent_type
         self.verbose = verbose
+        self.debug_mode = debug_mode
         self.patch_run_output_json = patch_run_output_json
         self.workflow_run_id = workflow_run_id
         self.runnable = runnable
         self.experiment_name = experiment_name if experiment_name else get_mlflow_experiment_name()
+        self._setup_logging()
         self.setup_mlflow()
         self.chain_class = chain_class
         self.client = BaseClient(
@@ -65,6 +68,14 @@ class FairoExecutor:
             if hasattr(agent, 'verbose'):
                 agent.verbose = self.verbose
 
+    def _setup_logging(self):
+        """Configure MLflow logging level based on debug_mode."""
+        mlflow_logger = logging.getLogger('mlflow')
+        if not self.debug_mode:
+            import warnings
+            warnings.filterwarnings('ignore', category=UserWarning)
+            mlflow_logger.setLevel(logging.ERROR)
+
     def _build_pipeline(self) -> RunnableSequence:
         if not self.agents and not self.runnable:
             raise ValueError("At least one agent or runnable must be provided.")

{fairo-25.9.7 → fairo-25.10.1}/fairo/core/execution/model_log_helper.py

@@ -2,7 +2,7 @@ import mlflow
 import mlflow.pyfunc
 import os
 import sys
-import pkg_resources
+import importlib.metadata
 import ast
 import importlib.util
 import shutil
@@ -60,9 +60,9 @@ class ModelLogHelper:
         pip_packages = []
         for package in required_packages:
             try:
-                dist = pkg_resources.get_distribution(package)
-                pip_packages.append(f"{package}=={dist.version}")
-            except pkg_resources.DistributionNotFound:
+                version = importlib.metadata.version(package)
+                pip_packages.append(f"{package}=={version}")
+            except importlib.metadata.PackageNotFoundError:
                 pip_packages.append(package)
         
         conda_env = {

{fairo-25.9.7 → fairo-25.10.1}/fairo.egg-info/PKG-INFO

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

fairo-25.9.7/fairo/__init__.py

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