flowcept 0.9.15__py3-none-any.whl → 0.9.17__py3-none-any.whl

flowcept/agents/agents_utils.py

@@ -166,6 +166,21 @@ def build_llm_model(
             from flowcept.agents.llms.gemini25 import Gemini25LLM
 
             llm = Gemini25LLM(**_model_kwargs)
+    elif _service_provider == "openai":
+        from langchain_openai import ChatOpenAI
+
+        api_key = os.environ.get("OPENAI_API_KEY", AGENT.get("api_key"))
+        base_url = os.environ.get("OPENAI_BASE_URL", AGENT.get("llm_server_url") or None)  # optional
+        org = os.environ.get("OPENAI_ORG_ID", AGENT.get("organization", None))  # optional
+
+        init_kwargs = {"api_key": api_key}
+        if base_url:
+            init_kwargs["base_url"] = base_url
+        if org:
+            init_kwargs["organization"] = org
+
+        # IMPORTANT: use the merged kwargs so `model` and temps flow through
+        llm = ChatOpenAI(**init_kwargs, **_model_kwargs)
 
     else:
         raise Exception("Currently supported providers are sambanova, openai, azure, and google.")

flowcept/flowceptor/adapters/mlflow/interception_event_handler.py

@@ -1,16 +1,19 @@
 """Event handler module."""
 
+from pathlib import Path
 from watchdog.events import FileSystemEventHandler
 
 
 class InterceptionEventHandler(FileSystemEventHandler):
     """Event handler class."""
 
-    def __init__(self, interceptor_instance, callback_function):
+    def __init__(self, interceptor_instance, file_path_to_watch, callback_function):
         super().__init__()
+        self.file_path_to_watch = file_path_to_watch
         self.callback_function = callback_function
         self.interceptor_instance = interceptor_instance
 
     def on_modified(self, event):
         """Get on modified."""
-        self.callback_function(self.interceptor_instance)
+        if Path(event.src_path).resolve() == Path(self.file_path_to_watch).resolve():
+            self.callback_function(self.interceptor_instance)

flowcept/flowceptor/adapters/mlflow/mlflow_interceptor.py

@@ -84,7 +84,7 @@ class MLFlowInterceptor(BaseInterceptor):
     def observe(self):
         """Observe it."""
         self.logger.debug("Observing")
-        event_handler = InterceptionEventHandler(self, self.__class__.callback)
+        event_handler = InterceptionEventHandler(self, self.settings.file_path, self.__class__.callback)
         while not os.path.isfile(self.settings.file_path):
             self.logger.warning(
                 f"I can't watch the file {self.settings.file_path},"
@@ -95,6 +95,7 @@ class MLFlowInterceptor(BaseInterceptor):
             sleep(self.settings.watch_interval_sec)
 
         self._observer = PollingObserver()
-        self._observer.schedule(event_handler, self.settings.file_path, recursive=True)
+        watch_dir = os.path.dirname(self.settings.file_path) or "."
+        self._observer.schedule(event_handler, watch_dir, recursive=True)
         self._observer.start()
-        self.logger.info(f"Watching {self.settings.file_path}")
+        self.logger.info(f"Watching directory {watch_dir} with file {self.settings.file_path} ")

flowcept/flowceptor/adapters/tensorboard/tensorboard_interceptor.py

@@ -1,6 +1,7 @@
 """Tensorboard interceptor module."""
 
 import os
+from pathlib import Path
 from time import sleep
 
 from tbparse import SummaryReader
@@ -29,6 +30,8 @@ class TensorboardInterceptor(BaseInterceptor):
     def __init__(self, plugin_key="tensorboard"):
         super().__init__(plugin_key)
         self._observer: PollingObserver = None
+        if not Path(self.settings.file_path).is_dir():
+            raise Exception("Tensorboard Observer must observe directories.")
         self.state_manager = InterceptorStateManager(self.settings)
         self.state_manager.reset()
         self.log_metrics = set(self.settings.log_metrics)
@@ -108,13 +111,15 @@ class TensorboardInterceptor(BaseInterceptor):
 
     def observe(self):
         """Observe it."""
-        event_handler = InterceptionEventHandler(self, self.__class__.callback)
+        self.logger.debug("Observing")
+        event_handler = InterceptionEventHandler(self, self.settings.file_path, self.__class__.callback)
         while not os.path.isdir(self.settings.file_path):
             self.logger.debug(f"I can't watch the file {self.settings.file_path}, as it does not exist.")
             self.logger.debug(f"\tI will sleep for {self.settings.watch_interval_sec} s to see if it appears.")
             sleep(self.settings.watch_interval_sec)
 
         self._observer = PollingObserver()
+
         self._observer.schedule(event_handler, self.settings.file_path, recursive=True)
         self._observer.start()
         self.logger.debug(f"Watching {self.settings.file_path}")

flowcept/version.py

@@ -4,4 +4,4 @@
 # The expected format is: <Major>.<Minor>.<Patch>
 # This file is supposed to be automatically modified by the CI Bot.
 # See .github/workflows/version_bumper.py
-__version__ = "0.9.15"
+__version__ = "0.9.17"

{flowcept-0.9.15.dist-info → flowcept-0.9.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowcept
-Version: 0.9.15
+Version: 0.9.17
 Summary: Capture and query workflow provenance data using data observability
 Author: Oak Ridge National Laboratory
 License-Expression: MIT
@@ -9,7 +9,7 @@ Keywords: agentic-ai,agentic-workflows,ai,big-data,dask,data-analytics,data-inte
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Requires-Dist: msgpack
 Requires-Dist: numpy
 Requires-Dist: omegaconf
@@ -169,6 +169,7 @@ Flowcept captures and queries workflow provenance at runtime with minimal code c
 
 
 [![Documentation](https://img.shields.io/badge/docs-readthedocs.io-green.svg)](https://flowcept.readthedocs.io/)
+[![Slack](https://img.shields.io/badge/Slack-%23flowcept%40Workflows%20Community-4A154B?logo=slack)](https://workflowscommunity.slack.com/archives/C06L5GYJKQS)
 [![Build](https://github.com/ORNL/flowcept/actions/workflows/create-release-n-publish.yml/badge.svg)](https://github.com/ORNL/flowcept/actions/workflows/create-release-n-publish.yml)
 [![PyPI](https://badge.fury.io/py/flowcept.svg)](https://pypi.org/project/flowcept)
 [![Tests](https://github.com/ORNL/flowcept/actions/workflows/run-tests.yml/badge.svg)](https://github.com/ORNL/flowcept/actions/workflows/run-tests.yml)

{flowcept-0.9.15.dist-info → flowcept-0.9.17.dist-info}/RECORD

@@ -1,10 +1,10 @@
 flowcept/__init__.py,sha256=tvVZKyymdqv3qOsgpAyDppBlUiBc0ag4QF21IcS-mVk,2449
 flowcept/cli.py,sha256=d3hogRpuMwQFIqw_fsYD4y074o3AFslBjlkRvTthOVM,25702
 flowcept/configs.py,sha256=DBkYx0CAaDSl8x2EJY1665PFY80eCp9PEriYH-BNwL4,8781
-flowcept/version.py,sha256=lpsttQT_qTX1Bsw13GeGYe5P2OBF2EKw2PC8dx5YBLw,307
+flowcept/version.py,sha256=CCisSYx_UPtct6ynMdnfePrYgGtXK-iQk7gTD4nsAVs,307
 flowcept/agents/__init__.py,sha256=8eeD2CiKBtHiDsWdrHK_UreIkKlTq4dUbhHDyzw372o,175
 flowcept/agents/agent_client.py,sha256=UiBQkC9WE2weLZR2OTkEOEQt9-zqQOkPwRA17HfI-jk,2027
-flowcept/agents/agents_utils.py,sha256=Az5lvWTsBHs_3sWWwy7jSdDjNn-PvZ7KmYd79wxvdyU,6666
+flowcept/agents/agents_utils.py,sha256=MXrzAFwCnBQF7s28QLN3OzWEIryOl-a0zeVQIbzNxD4,7330
 flowcept/agents/dynamic_schema_tracker.py,sha256=TsmXRRkyUkqB-0bEgmeqSms8xj1tMMJeYvjoaO2mtwI,6829
 flowcept/agents/flowcept_agent.py,sha256=1sidjnNMdG0S6lUKBvml7ZfIb6o3u7zc6HNogsJbl9g,871
 flowcept/agents/flowcept_ctx_manager.py,sha256=-WmkddzzFY2dnU9LbZaoY4-5RcSAQH4FziEJgcC5LEI,7083
@@ -74,13 +74,13 @@ flowcept/flowceptor/adapters/dask/dask_dataclasses.py,sha256=6LTG-kdcc6AUuVINvkq
 flowcept/flowceptor/adapters/dask/dask_interceptor.py,sha256=uBQpLluYXzlT1gBDfTe4_WueC_fWBEs5Xr8ntpOmljE,5869
 flowcept/flowceptor/adapters/dask/dask_plugins.py,sha256=s1ENAi9N61PC_6RiFvOYhJsgWzSm_lFWm3w87V-R1YY,2473
 flowcept/flowceptor/adapters/mlflow/__init__.py,sha256=3mzHrvh1XQOy68qx1A3so9Nq27tIb0i2mSXfv3F6gZg,25
-flowcept/flowceptor/adapters/mlflow/interception_event_handler.py,sha256=-SsIRdOcZjQUTzWgsZ41ouqpla4Qd32jIWXIAGU1pPw,494
+flowcept/flowceptor/adapters/mlflow/interception_event_handler.py,sha256=eR9vrPtSFEonJK7GQE1WTp1KVomV188uwRUqRRbFT9Q,682
 flowcept/flowceptor/adapters/mlflow/mlflow_dao.py,sha256=dPEgCduiw14_pzT5WCjuokwaN7p5Tu7UvWS2rtGh4qk,4589
 flowcept/flowceptor/adapters/mlflow/mlflow_dataclasses.py,sha256=vbijpDW6npHdsA9-28otXw94O4a9R-PWtq3xlJapsyY,690
-flowcept/flowceptor/adapters/mlflow/mlflow_interceptor.py,sha256=OLmVBdOCMS3GPcdxSdCD794RDbW6p4f8eBh1PXWcvHE,3799
+flowcept/flowceptor/adapters/mlflow/mlflow_interceptor.py,sha256=gufzv2OSVuLkFoo5In7aRzAZOFptwH6R9TTSsqO5vb8,3911
 flowcept/flowceptor/adapters/tensorboard/__init__.py,sha256=LrcR4WCIlBwwHIUSteQ8k8JBdCJTFqLvvgAfnoLeREw,30
 flowcept/flowceptor/adapters/tensorboard/tensorboard_dataclasses.py,sha256=lSfDd6TucVNzGxbm69BYyCVgMr2p9iUEQjnsS4jIfeI,554
-flowcept/flowceptor/adapters/tensorboard/tensorboard_interceptor.py,sha256=PUKGlCsYcybsk1HK573Brs6FiXQRoaj6MKgZ3Oyeec4,4881
+flowcept/flowceptor/adapters/tensorboard/tensorboard_interceptor.py,sha256=-uGtCDE8YJ-hbZKII_l3HVXNMKiw_MjrfAdcdt6ex70,5104
 flowcept/flowceptor/consumers/__init__.py,sha256=foxtVEb2ZEe9g1slfYIKM4tIFv-He1l7XS--SYs7nlQ,28
 flowcept/flowceptor/consumers/base_consumer.py,sha256=hrZ3VFV7pJBMXZsvh7Q2Y36b_ifcnbJkgwe2MiuZL70,3324
 flowcept/flowceptor/consumers/consumer_utils.py,sha256=E6R07zIKNXJTCxvL-OCrCKNYRpqtwRiXiZx0D2BKidk,5893
@@ -94,9 +94,9 @@ flowcept/instrumentation/flowcept_loop.py,sha256=nF7Sov-DCDapyYvS8zx-1ZFrnjc3CPg
 flowcept/instrumentation/flowcept_task.py,sha256=EmKODpjl8usNklKSVmsKYyCa6gC_QMqKhAr3DKaw44s,8199
 flowcept/instrumentation/flowcept_torch.py,sha256=kkZQRYq6cDBpdBU6J39_4oKRVkhyF3ODlz8ydV5WGKw,23455
 flowcept/instrumentation/task_capture.py,sha256=1g9EtLdqsTB0RHsF-eRmA2Xh9l_YqTd953d4v89IC24,8287
-resources/sample_settings.yaml,sha256=-j2tPycvT2xICEBeIX7Poye7R42SycEA6_hY3H6VeWE,6881
-flowcept-0.9.15.dist-info/METADATA,sha256=OU2QaLrjb7FWoIt7b4CbCYLU4qsnyCoLPHRC1uRXI2w,32896
-flowcept-0.9.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flowcept-0.9.15.dist-info/entry_points.txt,sha256=i8q67WE0201rVxYI2lyBtS52shvgl93x2Szp4q8zMlw,47
-flowcept-0.9.15.dist-info/licenses/LICENSE,sha256=r5-2P6tFTuRGWT5TiX32s1y0tnp4cIqBEC1QjTaXe2k,1086
-flowcept-0.9.15.dist-info/RECORD,,
+resources/sample_settings.yaml,sha256=4Hfj7qnthQdfWoCQYRLEF5Jm3zfOFmNbko249tIizPY,6881
+flowcept-0.9.17.dist-info/METADATA,sha256=8YvAYyHRt02C-TWstfldJrTL9pEX2ijrEn-ygpkVgJY,33056
+flowcept-0.9.17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flowcept-0.9.17.dist-info/entry_points.txt,sha256=i8q67WE0201rVxYI2lyBtS52shvgl93x2Szp4q8zMlw,47
+flowcept-0.9.17.dist-info/licenses/LICENSE,sha256=r5-2P6tFTuRGWT5TiX32s1y0tnp4cIqBEC1QjTaXe2k,1086
+flowcept-0.9.17.dist-info/RECORD,,

resources/sample_settings.yaml

@@ -1,4 +1,4 @@
-flowcept_version: 0.9.15 # Version of the Flowcept package. This setting file is compatible with this version.
+flowcept_version: 0.9.17 # Version of the Flowcept package. This setting file is compatible with this version.
 
 project:
   debug: true # Toggle debug mode. This will add a property `debug: true` to all saved data, making it easier to retrieve/delete them later.