olca 0.2.38__py3-none-any.whl → 0.2.40__py3-none-any.whl

olca/olcacli.py

@@ -5,7 +5,7 @@ import dotenv
 from langchain import hub
 import argparse
 import yaml
-import fusewill_utils as fu
+from olca.utils import load_environment, initialize_langfuse
 
 #jgwill/olca1
 #olca1_prompt = hub.pull("jgwill/olca1") #Future use
@@ -137,6 +137,10 @@ def extract_extra_directories_from_olca_config_system_and_user_input(system_inst
 def print_stream(stream):
     for s in stream:
         try:
+            # Skip Langfuse internal state messages
+            if isinstance(s, dict) and 'keep_alive' in s and 'states' in s:
+                continue
+                
             # Handle different response formats
             if isinstance(s, dict) and "messages" in s:
                 message = s["messages"][-1]
@@ -145,10 +149,16 @@ def print_stream(stream):
                 
             if isinstance(message, tuple):
                 print(message)
-            else:
+            elif hasattr(message, 'content'):
                 print(message.content)
+            else:
+                # Only print if it's not a system message
+                if not (isinstance(s, dict) and any(k in s for k in ['keep_alive', 'states'])):
+                    print(s)
         except Exception as e:
-            print(s)
+            # Only print if it's not a system message
+            if not (isinstance(s, dict) and any(k in s for k in ['keep_alive', 'states'])):
+                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
@@ -177,15 +187,6 @@ def _parse_args():
 
 from olca.tracing import TracingManager
 
-def load_environment():
-    # Load .env from current directory
-    dotenv.load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
-    
-    # Try loading from home directory if keys are missing
-    if not all([os.getenv(key) for key in ["LANGFUSE_PUBLIC_KEY", "LANGFUSE_SECRET_KEY", "LANGFUSE_HOST", 
-                                          "LANGCHAIN_API_KEY", "OPENAI_API_KEY"]]):
-        dotenv.load_dotenv(dotenv_path=os.path.expanduser("~/.env"))
-
 def main():
     args = _parse_args()
     olca_config_file = 'olca.yml'
@@ -193,6 +194,9 @@ def main():
     # Load environment variables first
     load_environment()
     
+    # Initialize Langfuse if needed
+    langfuse = initialize_langfuse()
+    
     if args.init:
         if os.path.exists(olca_config_file):
             print("Error: Configuration file already exists. Cannot run 'olca init'.")

olca/tracing.py

@@ -1,6 +1,7 @@
 import os
 from langfuse.callback import CallbackHandler as LangfuseCallbackHandler
 from langfuse import Langfuse
+from olca.utils import initialize_langfuse
 
 class TracingManager:
     def __init__(self, config):
@@ -29,16 +30,11 @@ class TracingManager:
             os.environ["LANGCHAIN_TRACING_V2"] = "true"
 
     def _setup_langfuse(self):
-        required_vars = ["LANGFUSE_PUBLIC_KEY", "LANGFUSE_SECRET_KEY", "LANGFUSE_HOST"]
-        if not all(os.getenv(var) for var in required_vars):
+        langfuse = initialize_langfuse()
+        if not langfuse:
             print("Warning: Missing Langfuse environment variables")
             return None
             
-        Langfuse(
-            public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
-            secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
-            host=os.getenv("LANGFUSE_HOST")
-        )
         return LangfuseCallbackHandler()
 
     def get_callbacks(self):

olca/utils.py

@@ -0,0 +1,24 @@
+import os
+import sys
+import dotenv
+import webbrowser
+from langfuse import Langfuse
+
+def load_environment():
+    dotenv.load_dotenv(dotenv_path=os.path.join(os.getcwd(), ".env"))
+    
+    # Try loading from home directory if variables are still not set
+    if not all([os.getenv(key) for key in ["LANGFUSE_PUBLIC_KEY", "LANGFUSE_SECRET_KEY", "LANGFUSE_HOST", 
+                                          "LANGCHAIN_API_KEY", "OPENAI_API_KEY"]]):
+        dotenv.load_dotenv(dotenv_path=os.path.expanduser("~/.env"))
+
+def initialize_langfuse():
+    required_vars = ["LANGFUSE_PUBLIC_KEY", "LANGFUSE_SECRET_KEY", "LANGFUSE_HOST"]
+    if not all(os.getenv(var) for var in required_vars):
+        return None
+    
+    return Langfuse(
+        public_key=os.getenv("LANGFUSE_PUBLIC_KEY"),
+        secret_key=os.getenv("LANGFUSE_SECRET_KEY"),
+        host=os.getenv("LANGFUSE_HOST")
+    )

{olca-0.2.38.dist-info → olca-0.2.40.dist-info}/METADATA

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

@@ -0,0 +1,12 @@
+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=whzbvrhJkiIoHkTPL_mRH0ZzZvmUKwZ2LiL65s32Vzw,15980
+olca/tracing.py,sha256=FnToJVhD7XxP0lBe_Ca7bQJYDHEs9bSaFqZx7rpbSaU,1269
+olca/utils.py,sha256=l01DkaDw5ENl8tzN8X4Ftgntz321AEAY1gpRX8_5q6Q,879
+olca-0.2.40.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+olca-0.2.40.dist-info/METADATA,sha256=C7JU43sZtW1oLo_0OGjScVspSsNKkdB5STHVfFegGd8,25311
+olca-0.2.40.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+olca-0.2.40.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
+olca-0.2.40.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
+olca-0.2.40.dist-info/RECORD,,

olca-0.2.38.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=VXXGfUAxrWTI_EM-TRpDNo2Cop3MRZWgQfN_S0HJwNc,15777
-olca/tracing.py,sha256=OexV2i7d7-tLN9XYTL9D9gGh0y0uYcfQK7_eIDNX2Iw,1495
-olca-0.2.38.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-olca-0.2.38.dist-info/METADATA,sha256=YHvmI1c0vXi6yjFquHAWj1ZKBSuOMcHTJOBZiUQz100,25311
-olca-0.2.38.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-olca-0.2.38.dist-info/entry_points.txt,sha256=AhP5FMv6vnOq9C76V_vxRVIO50smnZXG4RIY47oD2_U,103
-olca-0.2.38.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
-olca-0.2.38.dist-info/RECORD,,