flowcept 0.9.4__tar.gz → 0.9.6__tar.gz

{flowcept-0.9.4 → flowcept-0.9.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flowcept
-Version: 0.9.4
+Version: 0.9.6
 Summary: Capture and query workflow provenance data using data observability
 Author: Oak Ridge National Laboratory
 License-Expression: MIT

{flowcept-0.9.4 → flowcept-0.9.6}/resources/sample_settings.yaml

@@ -1,4 +1,4 @@
-flowcept_version: 0.9.4 # Version of the Flowcept package. This setting file is compatible with this version.
+flowcept_version: 0.9.6 # 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.

{flowcept-0.9.4 → flowcept-0.9.6}/src/flowcept/agents/prompts/in_memory_query_prompts.py

@@ -176,6 +176,9 @@ QUERY_GUIDELINES = """
       -To select the first (or earliest) N workflow executions, use or adapt the following: `df.groupby('workflow_id', as_index=False).agg({{"started_at": 'min'}}).sort_values(by='started_at', ascending=True).head(N)['workflow_id']` - utilize `started_at` to sort!     
       -To select the last (or latest or most recent) N workflow executions, use or adapt the following: `df.groupby('workflow_id', as_index=False).agg({{"ended_at": 'max'}}).sort_values(by='ended_at', ascending=False).head(N)['workflow_id']` - utilize `ended_at` to sort!
       
+      -To select the first or earliest or initial tasks, use or adapt the following: `df.sort_values(by='started_at', ascending=True)`
+      -To select the last or final or most recent tasks, use or adapt the following: `df.sort_values(by='ended_at', ascending=False)`
+      
       -WHEN the user requests a "summary" of activities, you must incorporate relevant summary statistics such as min, max, and mean, into the code you generate.
       -Do NOT use df[0] or df[integer value] or df[df[<field name>].idxmax()] or df[df[<field name>].idxmin()] because these are obviously not valid Pandas Code!
       -**Do NOT use any of those: df[df['started_at'].idxmax()], df[df['started_at'].idxmin()], df[df['ended_at'].idxmin()], df[df['ended_at'].idxmax()]. Those are not valid Pandas Code.**

{flowcept-0.9.4 → flowcept-0.9.6}/src/flowcept/agents/tools/general_tools.py

@@ -93,7 +93,6 @@ def prompt_handler(message: str) -> ToolResult:
     if "@record" in message:
         return record_guidance(message)
 
-
     llm = build_llm_model()
 
     prompt = ROUTING_PROMPT + message

{flowcept-0.9.4 → flowcept-0.9.6}/src/flowcept/agents/tools/in_memory_queries/in_memory_queries_tools.py

@@ -106,7 +106,7 @@ def run_df_query(llm, query: str, plot=False) -> ToolResult:
 
 
 @mcp_flowcept.tool()
-def generate_plot_code(llm, query, dynamic_schema, value_examples, df) -> ToolResult:
+def generate_plot_code(llm, query, dynamic_schema, value_examples, df, custom_user_guidance=None) -> ToolResult:
     """
     Generate DataFrame and plotting code from a natural language query using an LLM.
 

{flowcept-0.9.4 → flowcept-0.9.6}/src/flowcept/agents/tools/in_memory_queries/pandas_agent_utils.py

@@ -142,7 +142,7 @@ def format_result_df(result_df) -> str:
     """
     if isinstance(result_df, pd.DataFrame):
         if not len(result_df):
-            raise Exception("Empty DataFrame")
+            return "Empty Results"
         if len(result_df) > 100:
             print("Result set is too long. We are only going to send the head.")  # TODO log
             # TODO deal with very long results later

{flowcept-0.9.4 → flowcept-0.9.6}/src/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.4"
+__version__ = "0.9.6"