SimplerLLM 0.3.2.4__tar.gz → 0.3.2.6__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SimplerLLM
-Version: 0.3.2.4
+Version: 0.3.2.6
 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.6}/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.6}/SimplerLLM/tools/json_helpers.py

@@ -1,8 +1,6 @@
 import re
 import json
 from pydantic import BaseModel, ValidationError
-from typing import get_type_hints
-from pydantic import BaseModel
 from typing import Type, get_type_hints, List, get_origin, get_args, Union, Dict, Any, Literal
 from enum import Enum
 from datetime import datetime, date, time
@@ -77,7 +75,6 @@ def __json_extend_search(text, span):
 
 
 
-@DeprecationWarning
 def __extend_search_deprecated(text, span):
     # Extend the search to try to capture nested structures
     start, end = span

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: SimplerLLM
-Version: 0.3.2.4
+Version: 0.3.2.6
 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.6}/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.6",
     author="Hasan Aboul Hasan",
     author_email="hasan@learnwithhasan.com",
     description="An easy-to-use Library for interacting with language models.",