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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (8) hide show
  1. flowcept/agents/agents_utils.py +15 -0
  2. flowcept/version.py +1 -1
  3. {flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/METADATA +1 -1
  4. {flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/RECORD +8 -8
  5. resources/sample_settings.yaml +1 -1
  6. {flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/WHEEL +0 -0
  7. {flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/entry_points.txt +0 -0
  8. {flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/licenses/LICENSE +0 -0
flowcept/agents/agents_utils.py CHANGED
@@ -166,6 +166,21 @@ def build_llm_model(
166
166
  from flowcept.agents.llms.gemini25 import Gemini25LLM
167
167
 
168
168
  llm = Gemini25LLM(**_model_kwargs)
169
+ elif _service_provider == "openai":
170
+ from langchain_openai import ChatOpenAI
171
+
172
+ api_key = os.environ.get("OPENAI_API_KEY", AGENT.get("api_key"))
173
+ base_url = os.environ.get("OPENAI_BASE_URL", AGENT.get("llm_server_url") or None) # optional
174
+ org = os.environ.get("OPENAI_ORG_ID", AGENT.get("organization", None)) # optional
175
+
176
+ init_kwargs = {"api_key": api_key}
177
+ if base_url:
178
+ init_kwargs["base_url"] = base_url
179
+ if org:
180
+ init_kwargs["organization"] = org
181
+
182
+ # IMPORTANT: use the merged kwargs so `model` and temps flow through
183
+ llm = ChatOpenAI(**init_kwargs, **_model_kwargs)
169
184
 
170
185
  else:
171
186
  raise Exception("Currently supported providers are sambanova, openai, azure, and google.")
flowcept/version.py CHANGED
@@ -4,4 +4,4 @@
4
4
  # The expected format is: <Major>.<Minor>.<Patch>
5
5
  # This file is supposed to be automatically modified by the CI Bot.
6
6
  # See .github/workflows/version_bumper.py
7
- __version__ = "0.9.15"
7
+ __version__ = "0.9.16"
{flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flowcept
3
- Version: 0.9.15
3
+ Version: 0.9.16
4
4
  Summary: Capture and query workflow provenance data using data observability
5
5
  Author: Oak Ridge National Laboratory
6
6
  License-Expression: MIT
{flowcept-0.9.15.dist-info → flowcept-0.9.16.dist-info}/RECORD RENAMED
@@ -1,10 +1,10 @@
1
1
  flowcept/__init__.py,sha256=tvVZKyymdqv3qOsgpAyDppBlUiBc0ag4QF21IcS-mVk,2449
2
2
  flowcept/cli.py,sha256=d3hogRpuMwQFIqw_fsYD4y074o3AFslBjlkRvTthOVM,25702
3
3
  flowcept/configs.py,sha256=DBkYx0CAaDSl8x2EJY1665PFY80eCp9PEriYH-BNwL4,8781
4
- flowcept/version.py,sha256=lpsttQT_qTX1Bsw13GeGYe5P2OBF2EKw2PC8dx5YBLw,307
4
+ flowcept/version.py,sha256=X_noM4WnOGvGvCrAd_pXE-NBhOzr3f_98p4cBho_IKA,307
5
5
  flowcept/agents/__init__.py,sha256=8eeD2CiKBtHiDsWdrHK_UreIkKlTq4dUbhHDyzw372o,175
6
6
  flowcept/agents/agent_client.py,sha256=UiBQkC9WE2weLZR2OTkEOEQt9-zqQOkPwRA17HfI-jk,2027
7
- flowcept/agents/agents_utils.py,sha256=Az5lvWTsBHs_3sWWwy7jSdDjNn-PvZ7KmYd79wxvdyU,6666
7
+ flowcept/agents/agents_utils.py,sha256=MXrzAFwCnBQF7s28QLN3OzWEIryOl-a0zeVQIbzNxD4,7330
8
8
  flowcept/agents/dynamic_schema_tracker.py,sha256=TsmXRRkyUkqB-0bEgmeqSms8xj1tMMJeYvjoaO2mtwI,6829
9
9
  flowcept/agents/flowcept_agent.py,sha256=1sidjnNMdG0S6lUKBvml7ZfIb6o3u7zc6HNogsJbl9g,871
10
10
  flowcept/agents/flowcept_ctx_manager.py,sha256=-WmkddzzFY2dnU9LbZaoY4-5RcSAQH4FziEJgcC5LEI,7083
@@ -94,9 +94,9 @@ flowcept/instrumentation/flowcept_loop.py,sha256=nF7Sov-DCDapyYvS8zx-1ZFrnjc3CPg
94
94
  flowcept/instrumentation/flowcept_task.py,sha256=EmKODpjl8usNklKSVmsKYyCa6gC_QMqKhAr3DKaw44s,8199
95
95
  flowcept/instrumentation/flowcept_torch.py,sha256=kkZQRYq6cDBpdBU6J39_4oKRVkhyF3ODlz8ydV5WGKw,23455
96
96
  flowcept/instrumentation/task_capture.py,sha256=1g9EtLdqsTB0RHsF-eRmA2Xh9l_YqTd953d4v89IC24,8287
97
- resources/sample_settings.yaml,sha256=-j2tPycvT2xICEBeIX7Poye7R42SycEA6_hY3H6VeWE,6881
98
- flowcept-0.9.15.dist-info/METADATA,sha256=OU2QaLrjb7FWoIt7b4CbCYLU4qsnyCoLPHRC1uRXI2w,32896
99
- flowcept-0.9.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
100
- flowcept-0.9.15.dist-info/entry_points.txt,sha256=i8q67WE0201rVxYI2lyBtS52shvgl93x2Szp4q8zMlw,47
101
- flowcept-0.9.15.dist-info/licenses/LICENSE,sha256=r5-2P6tFTuRGWT5TiX32s1y0tnp4cIqBEC1QjTaXe2k,1086
102
- flowcept-0.9.15.dist-info/RECORD,,
97
+ resources/sample_settings.yaml,sha256=M994vAtdMeEzjfJ0v56ibYEtEIXciNSy9iZqXuWAJcA,6881
98
+ flowcept-0.9.16.dist-info/METADATA,sha256=PM5oaTzN7PT46XwOFs480asKZeq76Y_km1mc72zdCEs,32896
99
+ flowcept-0.9.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
100
+ flowcept-0.9.16.dist-info/entry_points.txt,sha256=i8q67WE0201rVxYI2lyBtS52shvgl93x2Szp4q8zMlw,47
101
+ flowcept-0.9.16.dist-info/licenses/LICENSE,sha256=r5-2P6tFTuRGWT5TiX32s1y0tnp4cIqBEC1QjTaXe2k,1086
102
+ flowcept-0.9.16.dist-info/RECORD,,
resources/sample_settings.yaml CHANGED
@@ -1,4 +1,4 @@
1
- flowcept_version: 0.9.15 # Version of the Flowcept package. This setting file is compatible with this version.
1
+ flowcept_version: 0.9.16 # Version of the Flowcept package. This setting file is compatible with this version.
2
2
 
3
3
  project:
4
4
  debug: true # Toggle debug mode. This will add a property `debug: true` to all saved data, making it easier to retrieve/delete them later.