fairo 25.7.4__tar.gz → 25.8.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.7.4
+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.7.4 → fairo-25.8.2}/fairo/core/chat/chat.py

@@ -122,6 +122,9 @@ class FairoDeploymentClient(BaseDeploymentClient):
                 os.environ.get('MLFLOW_TRACKING_PASSWORD')
             )
         
+        if os.environ.get('MLFLOW_TRACKING_TOKEN'):
+            headers['Authorization'] = f"Bearer {os.environ.get('MLFLOW_TRACKING_TOKEN')}"
+        
         # Make request
         response = requests.post(
             gateway_url,

{fairo-25.7.4 → fairo-25.8.2}/fairo/core/client/client.py

@@ -3,10 +3,17 @@ import requests
 from requests.auth import HTTPBasicAuth
 
 class BaseClient:
-    def __init__(self, base_url: str, username: str, password: str):
+    def __init__(self, base_url: str, username: str = None, password: str = None, fairo_auth_token: str = None):
         self.base_url = base_url
         self.session = requests.Session()
-        self.session.auth = HTTPBasicAuth(username, password)
+        if username is not None and password is not None:
+            self.session.auth = HTTPBasicAuth(username, password)
+        elif fairo_auth_token is not None:
+            self.session.headers.update({
+                "Authorization": f"Bearer {fairo_auth_token}"
+            })
+        else:
+            raise ValueError("Must provide either username/password or fairo_auth_token")
         self.session.headers.update({
             "Content-Type": "application/json",
         })

{fairo-25.7.4 → 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.7.4 → fairo-25.8.2}/fairo/core/workflow/dependency.py

@@ -185,15 +185,17 @@ class FairoVectorStore(BaseVectorStore):
         self.username = username or os.environ.get("FAIRO_API_ACCESS_KEY_ID")
         self.password = password or os.environ.get("FAIRO_API_SECRET")
         self.api_url = api_url or os.environ.get("FAIRO_BASE_URL", "https://api.fairo.ai")
+        self.fairo_auth_token = os.environ.get("FAIRO_AUTH_TOKEN")
 
-        if not self.username or not self.password:
-            raise ValueError("Fairo API credentials must be provided either as parameters or in the FAIRO_USERNAME and FAIRO_PASSWORD environment variables")
+        if not self.fairo_auth_token and (not self.username or not self.password):
+            raise ValueError("Fairo API credentials must be provided either as FAIRO_AUTH_TOKEN or as parameters or in the FAIRO_USERNAME and FAIRO_PASSWORD environment variables")
 
         # Initialize API client
         self.client = BaseClient(
             base_url=self.api_url.rstrip('/'),
             username=self.username,
-            password=self.password
+            password=self.password,
+            fairo_auth_token=self.fairo_auth_token
         )
 
         # Set up embeddings
@@ -258,11 +260,10 @@ class FairoVectorStore(BaseVectorStore):
             # Generate embeddings for the document content
             embedding = self.embeddings.embed_query(doc.page_content)
 
-            # Create doc entry
+            # Create doc entry (let Fairo API generate document IDs)
             doc_entry = {
                 "page_content": doc.page_content,
-                "metadata": doc.metadata,
-                "uuid": str(uuid.uuid4())  # Generate a UUID for this document
+                "metadata": doc.metadata
             }
             docs_data.append(doc_entry)
 

{fairo-25.7.4 → 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.7.4 → fairo-25.8.2}/fairo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fairo
-Version: 25.7.4
+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.7.4 → 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.7.4 → 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.7.4/fairo/__init__.py

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