flowcept 0.9.8__tar.gz → 0.9.9__tar.gz

{flowcept-0.9.8 → flowcept-0.9.9}/PKG-INFO

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

{flowcept-0.9.8 → flowcept-0.9.9}/resources/sample_settings.yaml

@@ -1,4 +1,4 @@
-flowcept_version: 0.9.8 # Version of the Flowcept package. This setting file is compatible with this version.
+flowcept_version: 0.9.9 # 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.8 → flowcept-0.9.9}/src/flowcept/agents/gui/agent_gui.py

@@ -37,9 +37,11 @@ display_ai_msg(GREETING)
 def main():
     """Main Agent GUI function."""
     st.caption(
-        "💡 Tip: Ask about workflow metrics, generate plots, or summarize data. "
-        "Inputs are mapped to `used` and outputs to `generated` fields. "
-        "Use @record <your query guidance> if you have custom guidance."
+        "💡 Quick help\n"
+        "Ask about workflow metrics, plots, or summaries.\n\n"
+        "Inputs → used, outputs → generated.\n"
+        "Commands: @record <note> add • @show records list • reset context clear • save context save.\n"
+        "Tip: set result = df to run ad-hoc Python on the in-memory DataFrame."
     )
 
     user_input = st.chat_input("Send a message")

{flowcept-0.9.8 → flowcept-0.9.9}/src/flowcept/agents/prompts/general_prompts.py

@@ -23,9 +23,9 @@ ROUTING_PROMPT = (
     "You are a routing assistant for a provenance AI agent. "
     "Given the following user message, classify it into one of the following routes:\n"
     "- small_talk: if it's casual conversation or some random word (e.g., 'hausdn', 'a', hello, how are you, what can you do, what's your name)\n"
+    "- in_context_query: if the user is querying the provenance data questions about tasks or data in running workflow (or a workflow that ran recently) or if the user mentions the in-memory 'df' or a dataframe. I expect that most of the interactions will fall in this category.\n"
     "- plot: if user is requesting plots (e.g., plot, chart, visualize)\n"
     # "- in_context_query: if the user asks questions about tasks or data in running workflow (or a workflow that ran recently) or if the user mentions the in-memory 'df' or a dataframe.\n"
-    "- in_context_query: if the user is querying the provenance data questions about tasks or data in running workflow (or a workflow that ran recently) or if the user mentions the in-memory 'df' or a dataframe.\n"
     # "- historical_prov_query: if the user wants to query historical provenance data\n"
     "- in_chat_query: if the user appears to be asking about something that has said recently in this chat.\n"
     "- unknown: if you don't know.\n"

{flowcept-0.9.8 → flowcept-0.9.9}/src/flowcept/agents/prompts/in_memory_query_prompts.py

@@ -180,6 +180,8 @@ QUERY_GUIDELINES = """
       -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)`
       
+      -If user explicitly asks to display or show all columns or fields, do not project on any particular field or column. Just show all of them.
+      
       -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.8 → flowcept-0.9.9}/src/flowcept/agents/tools/general_tools.py

@@ -73,6 +73,57 @@ def record_guidance(message: str) -> ToolResult:
     return ToolResult(code=201, result=f"Ok. I recorded in my memory: {message}")
 
 
+@mcp_flowcept.tool()
+def show_records() -> ToolResult:
+    """
+    Lists all recorded user guidance.
+    """
+    try:
+        ctx = mcp_flowcept.get_context()
+        custom_guidance: List = ctx.request_context.lifespan_context.custom_guidance
+        if not custom_guidance:
+            message = "There is no recorded user guidance."
+        else:
+            message = "This is the list of custom guidance I have in my memory:\n"
+            message += "\n".join(f" - {msg}" for msg in custom_guidance)
+
+        return ToolResult(code=201, result=message)
+    except Exception as e:
+        return ToolResult(code=499, result=str(e))
+
+
+@mcp_flowcept.tool()
+def reset_records() -> ToolResult:
+    """
+    Resets all recorded user guidance.
+    """
+    try:
+        ctx = mcp_flowcept.get_context()
+        ctx.request_context.lifespan_context.custom_guidance = []
+        return ToolResult(code=201, result="Custom guidance reset.")
+    except Exception as e:
+        return ToolResult(code=499, result=str(e))
+
+@mcp_flowcept.tool()
+def show_records() -> ToolResult:
+    """
+    Lists all recorded user guidance.
+    """
+    try:
+        ctx = mcp_flowcept.get_context()
+        custom_guidance: List = ctx.request_context.lifespan_context.custom_guidance
+        if not custom_guidance:
+            message = "There is no recorded user guidance."
+        else:
+            message = "This is the list of custom guidance I have in my memory:"
+            message += "\n".join(f"- {msg}" for msg in custom_guidance)
+
+        return ToolResult(code=201, result=message)
+    except Exception as e:
+        return ToolResult(code=499, result=str(e))
+
+
+
 @mcp_flowcept.tool()
 def prompt_handler(message: str) -> ToolResult:
     """
@@ -95,6 +146,10 @@ def prompt_handler(message: str) -> ToolResult:
 
     if "@record" in message:
         return record_guidance(message)
+    if "@show records" in message:
+        return show_records()
+    if "@reset records" in message:
+        return reset_records(message)
 
     llm = build_llm_model()
 
@@ -105,12 +160,12 @@ def prompt_handler(message: str) -> ToolResult:
         prompt = SMALL_TALK_PROMPT + message
         response = llm.invoke(prompt)
         return ToolResult(code=201, result=response)
+    elif route == "in_context_query":
+        return run_df_query(llm, message, plot=False)
     elif route == "plot":
         return run_df_query(llm, message, plot=True)
     elif route == "historical_prov_query":
         return ToolResult(code=201, result="We need to query the Provenance Database. Feature coming soon.")
-    elif route == "in_context_query":
-        return run_df_query(llm, message, plot=False)
     elif route == "in_chat_query":
         prompt = SMALL_TALK_PROMPT + message
         response = llm.invoke(prompt)

{flowcept-0.9.8 → flowcept-0.9.9}/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.8"
+__version__ = "0.9.9"