SimplerLLM 0.3.2.4__tar.gz → 0.3.2.5__tar.gz

{simplerllm-0.3.2.4 → simplerllm-0.3.2.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SimplerLLM
-Version: 0.3.2.4
+Version: 0.3.2.5
 Summary: An easy-to-use Library for interacting with language models.
 Home-page: https://github.com/hassancs91/SimplerLLM
 Author: Hasan Aboul Hasan

{simplerllm-0.3.2.4 → simplerllm-0.3.2.5}/SimplerLLM/language/llm/reliable.py

@@ -5,17 +5,20 @@ from SimplerLLM.utils.custom_verbose import verbose_print
 from SimplerLLM.language.llm_providers.llm_response_models import LLMFullResponse
 
 class ReliableLLM:
-    def __init__(self, primary_llm: LLM, secondary_llm: LLM, verbose=False):
+    def __init__(self, primary_llm: LLM, secondary_llm: LLM, verbose=False, validation_max_tokens: int = 4000):
         """
         Initialize ReliableLLM with primary and secondary LLM providers.
-        
+
         Args:
             primary_llm (LLM): The primary LLM provider to use first
             secondary_llm (LLM): The secondary LLM provider to use as fallback
+            verbose (bool): Enable verbose logging
+            validation_max_tokens (int): Max tokens for provider validation test (default: 4000)
         """
         self.primary_llm = primary_llm
         self.secondary_llm = secondary_llm
         self.verbose = verbose
+        self.validation_max_tokens = validation_max_tokens
         
         if self.verbose:
             verbose_print("Initializing ReliableLLM with fallback support", "info")
@@ -38,7 +41,7 @@ class ReliableLLM:
                 verbose_print("Validating primary provider...", "info")
             response = self.primary_llm.generate_response(
                 prompt="test",
-                max_tokens=1
+                max_tokens=self.validation_max_tokens
             )
             if response is None:
                 self.primary_valid = False
@@ -55,7 +58,7 @@ class ReliableLLM:
                 verbose_print("Validating secondary provider...", "info")
             response = self.secondary_llm.generate_response(
                 prompt="test",
-                max_tokens=1
+                max_tokens=self.validation_max_tokens
             )
             if response is None:
                 self.secondary_valid = False

{simplerllm-0.3.2.4 → simplerllm-0.3.2.5}/SimplerLLM.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SimplerLLM
-Version: 0.3.2.4
+Version: 0.3.2.5
 Summary: An easy-to-use Library for interacting with language models.
 Home-page: https://github.com/hassancs91/SimplerLLM
 Author: Hasan Aboul Hasan

{simplerllm-0.3.2.4 → simplerllm-0.3.2.5}/setup.py

@@ -20,7 +20,7 @@ with open("README.md", encoding="utf-8") as f:
 
 setup(
     name="SimplerLLM",
-    version="0.3.2.4",
+    version="0.3.2.5",
     author="Hasan Aboul Hasan",
     author_email="hasan@learnwithhasan.com",
     description="An easy-to-use Library for interacting with language models.",