fairo 25.7.2__tar.gz → 25.7.4__tar.gz

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

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

fairo-25.7.4/fairo/__init__.py

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

{fairo-25.7.2 → fairo-25.7.4}/fairo/core/chat/chat.py

@@ -205,6 +205,45 @@ class ChatFairo(ChatMlflow):
         self.target_uri = self._target_uri
         self._client = FairoDeploymentClient(self.target_uri, self.endpoint)
         return super().invoke(*args, **kwargs)
+    
+    def stream(self, *args, **kwargs):
+        # Override stream to use dynamic target_uri
+        self.target_uri = self._target_uri
+        self._client = FairoDeploymentClient(self.target_uri, self.endpoint)
+        return super().stream(*args, **kwargs)
+    
+    def bind_tools(self, tools, **kwargs):
+        result = super().bind_tools(tools, **kwargs)
+        result._uses_tools = True
+        return result
+    
+    def _stream(self, *args, **kwargs):
+        response = self.invoke(*args, **kwargs)
+        
+        from langchain_core.messages import AIMessage, AIMessageChunk
+        from langchain_core.outputs import ChatGenerationChunk
+        
+        if isinstance(response, AIMessage):
+            initial_chunk = AIMessageChunk(content="", role="assistant")
+            yield ChatGenerationChunk(message=initial_chunk)
+            
+            if response.content:
+                content_chunk = AIMessageChunk(content=response.content, role="assistant")
+                yield ChatGenerationChunk(message=content_chunk)
+            
+            if hasattr(response, 'tool_calls') and response.tool_calls:
+                tool_chunk = AIMessageChunk(
+                    content="", 
+                    role="assistant",
+                    tool_calls=response.tool_calls
+                )
+                yield ChatGenerationChunk(message=tool_chunk)
+            else:
+                final_chunk = AIMessageChunk(content="", role="assistant")
+                yield ChatGenerationChunk(message=final_chunk)
+        else:
+            chunk = AIMessageChunk(content=str(response), role="assistant")
+            yield ChatGenerationChunk(message=chunk)
 
 
 class FairoChat(ChatMlflow):

{fairo-25.7.2 → fairo-25.7.4}/fairo.egg-info/PKG-INFO

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

fairo-25.7.2/fairo/__init__.py

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