olca 0.2.35__py3-none-any.whl → 0.2.37__py3-none-any.whl

olca/olcacli.py

@@ -135,11 +135,19 @@ def extract_extra_directories_from_olca_config_system_and_user_input(system_inst
 
 def print_stream(stream):
     for s in stream:
-        message = s["messages"][-1]
-        if isinstance(message, tuple):
-            print(message)
-        else:
-            message.pretty_print()
+        try:
+            # Handle different response formats
+            if isinstance(s, dict) and "messages" in s:
+                message = s["messages"][-1]
+            else:
+                message = s
+                
+            if isinstance(message, tuple):
+                print(message)
+            else:
+                print(message.content)
+        except Exception as e:
+            print(s)
 
 def prepare_input(user_input, system_instructions,append_prompt=True, human=False):
     appended_prompt = system_instructions + SYSTEM_PROMPT_APPEND if append_prompt else system_instructions
@@ -192,21 +200,10 @@ def main():
     tracing_manager = TracingManager(config)
     callbacks = tracing_manager.get_callbacks()
     
-    # Check for tracing flag in config and CLI
-    tracing_enabled = config.get('tracing', False) or args.tracing or os.getenv("LANGCHAIN_TRACING_V2") == "true"
-    if tracing_enabled:
-        os.environ["LANGCHAIN_TRACING_V2"] = "true"
-        if not os.getenv("LANGCHAIN_API_KEY"):
-            print("Error: LANGCHAIN_API_KEY environment variable is required for tracing. Please set it up at : https://smith.langchain.com/settings")
-            exit(1)
-        # Initialize LangSmith client
-        import langsmith
-        LANGSMITH_API_KEY = os.getenv("LANGSMITH_API_KEY")
-        LANGCHAIN_API_KEY = os.getenv("LANGCHAIN_API_KEY")
-        if not LANGSMITH_API_KEY and not LANGCHAIN_API_KEY:
-            print("Error: LANGSMITH_API_KEY or LANGCHAIN_API_KEY environment variable is not set.")
-            exit(1)
-        client = langsmith.Client(api_key=LANGSMITH_API_KEY or LANGCHAIN_API_KEY)
+    # Remove old tracing setup
+    tracing_enabled = config.get('tracing', False) or args.tracing
+    if tracing_enabled and not callbacks:
+        print("Warning: Tracing enabled but no handlers configured")
 
     try:
             
@@ -292,7 +289,10 @@ def main():
     
 
     try:
-        print_stream(graph.stream(inputs, config={"callbacks": callbacks} if callbacks else None))
+        graph_config = {"callbacks": callbacks} if callbacks else {}
+        if recursion_limit:
+            graph_config["recursion_limit"] = recursion_limit
+        print_stream(graph.stream(inputs, config=graph_config))
     except GraphRecursionError as e:
         #print(f"Error: {e}")
         print("Recursion limit reached. Please increase the 'recursion_limit' in the olca_config.yaml file.")

olca/tracing.py

@@ -1,6 +1,5 @@
 import os
 from langfuse.callback import CallbackHandler as LangfuseCallbackHandler
-import langsmith
 
 class TracingManager:
     def __init__(self, config):
@@ -16,9 +15,7 @@ class TracingManager:
             return
 
         if 'langsmith' in providers:
-            handler = self._setup_langsmith()
-            if handler:
-                self.handlers.append(handler)
+            self._setup_langsmith()
 
         if 'langfuse' in providers:
             handler = self._setup_langfuse()
@@ -26,14 +23,9 @@ class TracingManager:
                 self.handlers.append(handler)
 
     def _setup_langsmith(self):
-        os.environ["LANGCHAIN_TRACING_V2"] = "true"
         api_key = os.getenv("LANGCHAIN_API_KEY") or os.getenv("LANGSMITH_API_KEY")
-        
-        if not api_key:
-            print("Warning: LANGCHAIN_API_KEY/LANGSMITH_API_KEY not set for LangSmith tracing")
-            return None
-            
-        return langsmith.Client(api_key=api_key)
+        if api_key:
+            os.environ["LANGCHAIN_TRACING_V2"] = "true"
 
     def _setup_langfuse(self):
         if not (os.getenv("LANGFUSE_PUBLIC_KEY") and os.getenv("LANGFUSE_SECRET_KEY")):

{olca-0.2.35.dist-info → olca-0.2.37.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: olca
-Version: 0.2.35
+Version: 0.2.37
 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.37.dist-info/RECORD

@@ -0,0 +1,11 @@
+olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
+olca/fusewill_cli.py,sha256=Gf8CaYs7Uo4NH8QfgRNYalpmSUo047p9rzdkvIABHi8,7872
+olca/fusewill_utils.py,sha256=IOIElqWCIsNzePlS1FZa5_35vySYLwbMUGW6UhNefIc,6065
+olca/olcacli.py,sha256=yUHnASmkXn2FygwA-MhTzgaJOhD4yEvuqIp7ghote3o,14834
+olca/tracing.py,sha256=A6K9K_mhCPS1_NncFv2-7hbeMkqfKT3XgUCERusbIYE,1248
+olca-0.2.37.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+olca-0.2.37.dist-info/METADATA,sha256=oQql8pKoyL4SKiOlvHlUnmjEKYQwbF0P76iGH328i3M,25311
+olca-0.2.37.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+olca-0.2.37.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
+olca-0.2.37.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
+olca-0.2.37.dist-info/RECORD,,

olca-0.2.35.dist-info/RECORD

@@ -1,11 +0,0 @@
-olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
-olca/fusewill_cli.py,sha256=Gf8CaYs7Uo4NH8QfgRNYalpmSUo047p9rzdkvIABHi8,7872
-olca/fusewill_utils.py,sha256=IOIElqWCIsNzePlS1FZa5_35vySYLwbMUGW6UhNefIc,6065
-olca/olcacli.py,sha256=0t8HBP-tLwgRCOi9z0gLgA_2UD5YB--_V8izDWPMJbQ,15143
-olca/tracing.py,sha256=D8RL6QnBdjCyjxyt-3LHTbq4NVL_6gXYjjw20ewQX2I,1536
-olca-0.2.35.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-olca-0.2.35.dist-info/METADATA,sha256=Wr-OlwhfER3iUCvqsDrWyb4lrugjhlueCg46jcYrK1o,25311
-olca-0.2.35.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-olca-0.2.35.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
-olca-0.2.35.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
-olca-0.2.35.dist-info/RECORD,,