olca 0.2.32__py3-none-any.whl → 0.2.33__py3-none-any.whl

olca/fusewill_cli.py

@@ -9,20 +9,36 @@ from fusewill_utils import (
     create_prompt,
     update_prompt,
     delete_dataset,
-    get_trace_by_id
+    get_trace_by_id,
+    open_trace_in_browser,
+    print_traces,
+    print_trace
 )
 import dotenv
 import json
+import sys, termios, tty
 #dotenv.load_dotenv()
 
+def get_single_char_input():
+    fd = sys.stdin.fileno()
+    old_settings = termios.tcgetattr(fd)
+    try:
+        tty.setraw(fd)
+        ch = sys.stdin.read(1)
+    finally:
+        termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
+    return ch
+
 def main():
     parser = argparse.ArgumentParser(description="Langfuse CLI Wrapper")
     subparsers = parser.add_subparsers(dest="command", help="Available commands")
 
     # list_traces command
-    parser_list = subparsers.add_parser('list_traces', help='List traces')
+    parser_list = subparsers.add_parser('list_traces', help='List traces',aliases=['lt'])
     parser_list.add_argument('--limit', type=int, default=100, help='Number of traces to fetch')
     parser_list.add_argument('--output_dir', type=str, default='../output/traces', help='Directory to save traces')
+    parser_list.add_argument('-C','--comments',  action='store_true', help='Show comments from the traces', default=False)
+    parser_list.add_argument('-W','--browse-interact', action='store_true', help='Ask user to open each trace in browser')
 
     # create_dataset command
     parser_create_dataset = subparsers.add_parser('create_dataset', help='Create a new dataset')
@@ -76,8 +92,29 @@ def main():
 
     args = parser.parse_args()
 
-    if args.command == 'list_traces':
-        list_traces(limit=args.limit, output_dir=args.output_dir)
+    if args.command == 'list_traces' or args.command == 'lt':
+        show_comments_flag = args.comments if args.comments else False
+        traces = list_traces(
+            limit=args.limit, 
+            output_dir=args.output_dir
+        )
+        if not args.browse_interact:
+            print_traces(traces, show_comments=show_comments_flag)
+        else:
+            for trace in traces.data:
+                print_trace(trace, show_comments=show_comments_flag)
+                print("Open this trace in browser (Y/N/Q)? ", end='', flush=True)
+                try:
+                    resp = get_single_char_input().lower()
+                except KeyboardInterrupt:
+                    print("\nExiting.")
+                    sys.exit(0)
+                print(resp)  # Echo the character
+                if resp == 'y':
+                    open_trace_in_browser(trace.id)
+                elif resp == 'q':
+                    print("Quitting.")
+                    break
     elif args.command == 'create_dataset':
         metadata = json.loads(args.metadata)
         create_dataset(name=args.name, description=args.description, metadata=metadata)
@@ -121,6 +158,7 @@ def main():
             print(f"Trace ID: {trace.id}, Name: {trace.name}")
     else:
         parser.print_help()
+        exit(1)
 
 if __name__ == '__main__':
     main()

olca/fusewill_utils.py

@@ -2,12 +2,13 @@ import os
 import sys
 import json
 import dotenv
+import webbrowser
 
 # Load .env from the current working directory
 dotenv.load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
 
 # Try loading from home directory if variables are still not set
-if not os.environ.get("LANGFUSE_PUBLIC_KEY") or not os.environ.get("LANGFUSE_SECRET_KEY"):
+if not os.environ.get("LANGFUSE_PUBLIC_KEY") or not os.environ.get("LANGFUSE_SECRET_KEY") or not os.environ.get("LANGFUSE_HOST"):
     dotenv.load_dotenv(dotenv_path=os.path.expanduser("~/.env"))
 
 # Final check before exiting
@@ -16,6 +17,8 @@ if not os.environ.get("LANGFUSE_PUBLIC_KEY"):
     missing_vars.append("LANGFUSE_PUBLIC_KEY")
 if not os.environ.get("LANGFUSE_SECRET_KEY"):
     missing_vars.append("LANGFUSE_SECRET_KEY")
+if not os.environ.get("LANGFUSE_HOST"):
+    missing_vars.append("LANGFUSE_HOST")
 
 if missing_vars:
     print(f"Error: {', '.join(missing_vars)} not found.")
@@ -28,17 +31,49 @@ sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
 import json
 
 import dotenv
+_DEBUG_=False
+if _DEBUG_:
+    print(os.environ.get("LANGFUSE_PUBLIC_KEY"))
+    print(os.environ.get("LANGFUSE_SECRET_KEY"))
+    print(os.environ.get("LANGFUSE_HOST"))
 
-langfuse = Langfuse()
+langfuse = Langfuse(
+    public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),
+    secret_key=os.environ.get("LANGFUSE_SECRET_KEY"),
+    host=os.environ.get("LANGFUSE_HOST")
+)
 
-def list_traces(limit=100, output_dir="../output/traces"):
+def open_trace_in_browser(trace_id):
+    base_url = os.environ.get("LANGFUSE_HOST")
+    project_id = os.environ.get("LANGFUSE_PROJECT_ID")
+    if not base_url or not project_id:
+        print("Missing LANGFUSE_HOST or LANGFUSE_PROJECT_ID")
+        return
+    full_url = f"{base_url}/project/{project_id}/traces/{trace_id}"
+    print(f"Opening {full_url}")
+    webbrowser.open(full_url)
+
+def print_trace(trace, show_comments=False):
+    print(f"<Trace \n\tat=\"{trace.createdAt}\" \n\tid=\"{trace.id}\" \n\tname=\"{trace.name}\" \n\tsession_id=\"{trace.session_id}\" \n\tprojectId=\"{trace.projectId}\" >")
+    print(f"<Input><CDATA[[\n{trace.input}\n]]></Input>")
+    print(f"<Output><CDATA[[\n{trace.output}\n]]></Output>")
+    print(f"<Metadata>{trace.metadata}</Metadata>")
+    if trace.scores:
+        print("<Scores>")
+        for score in trace.scores:
+            print(f"{score}")
+        print("</Scores>")
+    if show_comments and hasattr(trace, "comments"):
+        print(f"<Comments>\n{trace.comments}\n</Comments>")
+    print("</Trace>")
+
+def print_traces(traces, show_comments=False):
+    for trace in traces.data:
+        print_trace(trace, show_comments)
+
+def list_traces(limit=100, output_dir="../output/traces", show_comments=False):
     traces = langfuse.get_traces(limit=limit)
     os.makedirs(output_dir, exist_ok=True)
-    for trace in traces.data:
-        print(f"-----Trace ID: {trace.id}--Name: {trace.name}----------")
-        print(f"<output>{trace.output}</output>")
-        print(f"<Metadata>{trace.metadata}</Metadata>")
-        print("---")
     return traces
 
 def list_traces_by_score(score_name, min_value=None, max_value=None, limit=100):
@@ -62,11 +97,16 @@ def add_score_to_a_trace(trace_id, generation_id, name, value, data_type="NUMERI
         comment=comment
     )
 
-def create_score(name, data_type, description=""):
-    langfuse.create_score(
+def create_score(name, data_type, description="", possible_values=None, min_value=None, max_value=None):
+    langfuse.score(
         name=name,
+        value="",  # Provide a placeholder value
         data_type=data_type,
-        description=description
+        description=description,
+        # For categorical:
+        **({"possible_values": possible_values} if data_type == "CATEGORICAL" and possible_values else {}),
+        # For numeric:
+        **({"min_value": min_value, "max_value": max_value} if data_type == "NUMERIC" and min_value is not None and max_value is not None else {})
     )
 
 def score_exists(name):

{olca-0.2.32.dist-info → olca-0.2.33.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: olca
-Version: 0.2.32
+Version: 0.2.33
 Summary: A Python package for experimental usage of Langchain and Human-in-the-Loop
 Home-page: https://github.com/jgwill/olca
 Author: Jean GUillaume ISabelle

olca-0.2.33.dist-info/RECORD

@@ -0,0 +1,10 @@
+olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
+olca/fusewill_cli.py,sha256=-hq-uCQfx9Tkb8TTTJNgQS7hgigS9XDaNBL9CKAouMU,7415
+olca/fusewill_utils.py,sha256=KBCa8Qntfj-M1WzS8hD6X8O_rN4JoL6ZW_CYCJ8X1pU,5224
+olca/olcacli.py,sha256=s-dqJRE6xqQLG_qSwp8prahRV3vZNueEaK7OXgmU9wc,14885
+olca-0.2.33.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+olca-0.2.33.dist-info/METADATA,sha256=HHf928Hr4wplrAu3shaof-TAoluinuRCcVHHj7bfhlI,25311
+olca-0.2.33.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+olca-0.2.33.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
+olca-0.2.33.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
+olca-0.2.33.dist-info/RECORD,,

olca-0.2.32.dist-info/RECORD

@@ -1,10 +0,0 @@
-olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
-olca/fusewill_cli.py,sha256=ZFW2Oeq-gQpAaTIxzsztQWDMv7-R9WJTVcSspbTxpPk,5926
-olca/fusewill_utils.py,sha256=TyyCVRkWeXaS14vh0BXGK-tj0ZPZ3_uQNnAOM-IE9FA,3405
-olca/olcacli.py,sha256=s-dqJRE6xqQLG_qSwp8prahRV3vZNueEaK7OXgmU9wc,14885
-olca-0.2.32.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-olca-0.2.32.dist-info/METADATA,sha256=GioYe8rCVBdvGIw8QaIjkiMMQqvZfjYtRw85_z9ZcS4,25311
-olca-0.2.32.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-olca-0.2.32.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
-olca-0.2.32.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
-olca-0.2.32.dist-info/RECORD,,