olca 0.2.55__tar.gz → 0.2.56__tar.gz

{olca-0.2.55 → olca-0.2.56}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: olca
-Version: 0.2.55
+Version: 0.2.56
 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.55 → olca-0.2.56}/olca/fusewill_cli.py

@@ -92,6 +92,10 @@ def main():
     parser_list_by_score.add_argument('--max_value', type=float, help='Maximum score value')
     parser_list_by_score.add_argument('-L','--limit', type=int, default=100, help='Number of traces to fetch')
 
+    # list_scores command
+    parser_list_scores = subparsers.add_parser('list_scores', help='List all scores', aliases=['ls'])
+    parser_list_scores.add_argument('-o', '--output', type=str, help='Output JSON file path')
+
     args = parser.parse_args()
 
     if args.command == 'list_traces' or args.command == 'lt':
@@ -159,6 +163,20 @@ def main():
         for trace in traces:
             print_trace(trace)
             #print(f"Trace ID: {trace.id}, Name: {trace.name}")
+    elif args.command == 'list_scores' or args.command == 'ls':
+        scores = fu.list_scores()
+        if scores:
+            if args.output:
+                try:
+                    with open(args.output, 'w') as f:
+                        json.dump(scores, f, indent=2)
+                    print(f"Scores written to {os.path.realpath(args.output)}")
+                except Exception as e:
+                    print(f"Error writing to file {args.output}: {e}")
+            else:
+                print(json.dumps(scores, indent=2))
+        else:
+            print("No scores found.")
     else:
         parser.print_help()
         exit(1)

{olca-0.2.55 → olca-0.2.56}/olca/fusewill_utils.py

@@ -43,7 +43,11 @@ langfuse = Langfuse(
     secret_key=os.environ.get("LANGFUSE_SECRET_KEY"),
     host=os.environ.get("LANGFUSE_HOST")
 )
-
+def dummy():
+    o=langfuse.get_observations()
+    g=langfuse.get_generations()
+    prompts_list=langfuse.get_dataset()
+    
 def open_trace_in_browser(trace_id):
     base_url = os.environ.get("LANGFUSE_HOST")
     project_id = os.environ.get("LANGFUSE_PROJECT_ID")
@@ -68,6 +72,20 @@ def get_score_by_id(score_id):
         print(f"Error retrieving score {score_id}: {e}")
         return None
 
+def list_scores():
+    """Retrieve all score configurations."""
+    base_url = os.environ.get("LANGFUSE_HOST")
+    public_key = os.environ.get("LANGFUSE_PUBLIC_KEY")
+    secret_key = os.environ.get("LANGFUSE_SECRET_KEY")
+    url = f"{base_url}/api/public/score-configs"
+    try:
+        response = requests.get(url, auth=(public_key, secret_key))
+        response.raise_for_status()
+        return response.json()
+    except Exception as e:
+        print(f"Error retrieving scores: {e}")
+        return None
+
 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>")
@@ -157,7 +175,7 @@ def create_prompt(name, prompt_text, model_name, temperature, labels=None, suppo
         }
     )
 def get_prompt(name, label="production"):
-    return langfuse.get_prompt(name=name,label="production")
+    return langfuse.get_prompt(name=name,label=label)
   
 def update_prompt(name, new_prompt_text):
     prompt = langfuse.get_prompt(name=name)

{olca-0.2.55 → olca-0.2.56}/olca/olcacli.py

@@ -8,9 +8,6 @@ import yaml
 from olca.utils import load_environment, initialize_langfuse
 from olca.tracing import TracingManager
 from olca.olcahelper import setup_required_directories, initialize_config_file
-import time
-import httpx
-from openai.error import APIConnectionError
 
 #jgwill/olca1
 #olca1_prompt = hub.pull("jgwill/olca1") #Future use
@@ -308,19 +305,7 @@ def main():
         graph_config = {"callbacks": callbacks} if callbacks else {}
         if recursion_limit:
             graph_config["recursion_limit"] = recursion_limit
-        
-        retry_attempts = 3
-        for attempt in range(retry_attempts):
-            try:
-                print_stream(graph.stream(inputs, config=graph_config))
-                break
-            except (httpx.ConnectError, APIConnectionError) as e:
-                if attempt < retry_attempts - 1:
-                    print(f"Network error encountered: {e}. Retrying in 5 seconds...")
-                    time.sleep(5)
-                else:
-                    print("Failed to connect after multiple attempts. Exiting.")
-                    raise e
+        print_stream(graph.stream(inputs, config=graph_config))
     except GraphRecursionError as e:
         print("Recursion limit reached. Please increase the 'recursion_limit' in the olca_config.yaml file.")
         print("For troubleshooting, visit: https://python.langchain.com/docs/troubleshooting/errors/GRAPH_RECURSION_LIMIT")

{olca-0.2.55 → olca-0.2.56}/olca.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: olca
-Version: 0.2.55
+Version: 0.2.56
 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.55 → olca-0.2.56}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "olca"
-version = "0.2.55"
+version = "0.2.56"
 
 description = "A Python package for experimental usage of Langchain and Human-in-the-Loop"
 readme = "README.md"

{olca-0.2.55 → olca-0.2.56}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='olca',
-    version = "0.2.55",
+    version = "0.2.56",
     author='Jean GUillaume ISabelle',
     author_email='jgi@jgwill.com',
     description='A Python package for experimenting with Langchain agent and interactivity in Terminal modalities.',