olca 0.2.66__py3-none-any.whl → 0.2.71__py3-none-any.whl

olca/olcacli.py

@@ -161,9 +161,6 @@ def main():
     # Load environment variables first
     load_environment()
     
-    # Initialize Langfuse if needed
-    langfuse = initialize_langfuse(debug=True if args.debug else False)
-    
     if args.init:
         if os.path.exists(olca_config_file):
             print("Error: Configuration file already exists. Cannot run 'olca init'.")
@@ -179,6 +176,12 @@ def main():
         return
 
     config = load_config(olca_config_file)
+
+    tracing_providers = config.get('tracing_providers', [])
+    if 'langfuse' in tracing_providers:
+        langfuse = initialize_langfuse(debug=True if args.debug else False)
+    else:
+        langfuse = None
     
     # Initialize tracing
     tracing_manager = TracingManager(config)
@@ -225,8 +228,18 @@ def main():
     provider, base_model, host = parse_model_uri(model_name)
     
     if provider == "ollama":
-        from langchain_ollama import OllamaLLM
-        model = OllamaLLM(model=base_model, base_url=host if host else None)
+        import ollama
+        from langchain_ollama import ChatOllama
+        try:
+            ollama.Client(host=host if host else None).show(base_model)
+        except:
+            print(f"Model {base_model} not found, pulling it...")
+            pull_stream = ollama.Client(host=host if host else None).pull(base_model, stream=True)
+            for chunk in pull_stream:
+                pass
+            print(f"\nPulled {base_model}")
+
+        model = ChatOllama(model=base_model, host=host if host else None)
         
     elif provider == "openai":
         from langchain_openai import ChatOpenAI

olca/olcahelper.py

@@ -43,9 +43,15 @@ def initialize_config_file():
             "temperature": float(input("temperature [0]: ") or default_temperature),
             "human": input("human [true]: ").lower() in ["true", "yes", "y", "1", ""] or use_default_human_input,
             "tracing": input("tracing [true]: ").lower() in ["true", "yes", "y", "1", ""] or use_default_tracing,
-            "tracing_providers": ["langsmith", "langfuse"]
         }
-        
+
+        tracing_providers = []
+        if input("Use langsmith for tracing? [Y/n]: ").lower() in ["y", "yes", ""]:
+            tracing_providers.append("langsmith")
+        if input("Use langfuse for tracing? [Y/n]: ").lower() in ["y", "yes", ""]:
+            tracing_providers.append("langfuse")
+        config["tracing_providers"] = tracing_providers
+
         user_system_instructions = input(f"system_instructions [{default_system_instructions}]: ")
         user_system_instructions = user_system_instructions or default_system_instructions
         user_system_instructions = user_system_instructions.replace("\n", " ").replace("\r", " ").replace("\t", " ")

olca/tracing.py

@@ -1,6 +1,4 @@
 import os
-from langfuse.callback import CallbackHandler as LangfuseCallbackHandler
-from langfuse import Langfuse
 from olca.utils import initialize_langfuse
 import warnings
 
@@ -35,6 +33,8 @@ class TracingManager:
             os.environ["LANGCHAIN_TRACING_V2"] = "true"
 
     def _setup_langfuse(self):
+        from langfuse.callback import CallbackHandler as LangfuseCallbackHandler
+        from langfuse import Langfuse
         self.langfuse = initialize_langfuse()
         if not self.langfuse:
             print("Warning: Missing Langfuse environment variables")

olca/utils.py

@@ -2,7 +2,6 @@ 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"))
@@ -13,6 +12,7 @@ def load_environment():
         dotenv.load_dotenv(dotenv_path=os.path.expanduser("~/.env"))
 
 def initialize_langfuse( debug=False):
+    from langfuse import Langfuse
     required_vars = ["LANGFUSE_PUBLIC_KEY", "LANGFUSE_SECRET_KEY", "LANGFUSE_HOST"]
     if not all(os.getenv(var) for var in required_vars):
         return None

{olca-0.2.66.dist-info → olca-0.2.71.dist-info}/METADATA

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

@@ -0,0 +1,15 @@
+olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
+olca/fusewill_cli.py,sha256=7etgdpM64-jXGLlALG7PtwICAzoLKYGMBRvHf5nMLic,17119
+olca/fusewill_utils.py,sha256=sNCe4LhpSaEOY5olMxW1nTv7CKiT__dc2lGomelE6mY,18207
+olca/oiv.py,sha256=TyhbgjzE9B5woacln8AtITLv1X7iv9ZivJBdk0MyP4U,4610
+olca/olcacli.py,sha256=wwn4mL1kGonigI9w6UEsnFJDWQ6yONxqvogjrCj-Og4,11246
+olca/olcahelper.py,sha256=v16ETPboJFdJdkL1iAJ86_oAhkFo5jMFfx5oivOG8kA,4115
+olca/prompts.py,sha256=q7lvDZ8n_K6ea2nu3K5R2fgBgQYrCIEjh0cA9fUlMFI,3697
+olca/tracing.py,sha256=cEfPNZoI-PbMHFpX8bvY7OajeW4yvZBMycqwj6sRpDE,1637
+olca/utils.py,sha256=kDxKwD51234OvA6PIW9IfAND8DHwd8lSyOvSf4abvPE,916
+olca-0.2.71.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
+olca-0.2.71.dist-info/METADATA,sha256=fbrb7yvxwGwURri-_bic01CVFQrcMRjrqWRRYh0GoG0,25808
+olca-0.2.71.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+olca-0.2.71.dist-info/entry_points.txt,sha256=W2UkinpMZYsCLH54must9ahwbzCRl-rekcZ7rL5u3wA,123
+olca-0.2.71.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
+olca-0.2.71.dist-info/RECORD,,

olca-0.2.66.dist-info/RECORD

@@ -1,15 +0,0 @@
-olca/__init__.py,sha256=3QyLLAys_KiiDIe-cfO_7QyY7di_qCaCS-sVziW2BOw,23
-olca/fusewill_cli.py,sha256=7etgdpM64-jXGLlALG7PtwICAzoLKYGMBRvHf5nMLic,17119
-olca/fusewill_utils.py,sha256=sNCe4LhpSaEOY5olMxW1nTv7CKiT__dc2lGomelE6mY,18207
-olca/oiv.py,sha256=TyhbgjzE9B5woacln8AtITLv1X7iv9ZivJBdk0MyP4U,4610
-olca/olcacli.py,sha256=oftilrp9v0QOV5tJeb6o8nwO54gjjlwWVfDU5JpoXJc,10758
-olca/olcahelper.py,sha256=s9KwEYAJCbnKSXIsj5HCL-OHxkSsmL08ZBgoluxmP1s,3828
-olca/prompts.py,sha256=q7lvDZ8n_K6ea2nu3K5R2fgBgQYrCIEjh0cA9fUlMFI,3697
-olca/tracing.py,sha256=GSuXIKmIMQ9V2gLkKcin142Z2QpyS6obsnj8Tf-1zHc,1621
-olca/utils.py,sha256=zM94HDMDYF95Yd9ubeOK6vuepbQN4kDFh0rTvaVFagI,912
-olca-0.2.66.dist-info/LICENSE,sha256=gXf5dRMhNSbfLPYYTY_5hsZ1r7UU1OaKQEAQUhuIBkM,18092
-olca-0.2.66.dist-info/METADATA,sha256=rnjt54tCa8STAp7ZbD8VKHsW5AJtAh2NGQ9Dyjym2mQ,25808
-olca-0.2.66.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-olca-0.2.66.dist-info/entry_points.txt,sha256=W2UkinpMZYsCLH54must9ahwbzCRl-rekcZ7rL5u3wA,123
-olca-0.2.66.dist-info/top_level.txt,sha256=bGDtAReS-xlS0F6MM-DyD0IQUqjNdWmgemnM3vNtrpI,5
-olca-0.2.66.dist-info/RECORD,,