local-deep-research 0.1.19__tar.gz → 0.1.20__tar.gz

{local_deep_research-0.1.19/src/local_deep_research.egg-info → local_deep_research-0.1.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: local-deep-research
-Version: 0.1.19
+Version: 0.1.20
 Summary: AI-powered research assistant with deep, iterative analysis using LLMs and web searches
 Author-email: LearningCircuit <185559241+LearningCircuit@users.noreply.github.com>, HashedViking <6432677+HashedViking@users.noreply.github.com>
 License: MIT License

{local_deep_research-0.1.19 → local_deep_research-0.1.20}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "local-deep-research"
-version = "0.1.19"
+version = "0.1.20"
 description = "AI-powered research assistant with deep, iterative analysis using LLMs and web searches"
 readme = "README.md"
 requires-python = ">=3.8"
@@ -51,6 +51,7 @@ dependencies = [
     "lxml>=4.9.2",
     "pdfplumber>=0.9.0",
     "unstructured>=0.10.0",
+
 ]
 
 [project.urls]

{local_deep_research-0.1.19 → local_deep_research-0.1.20}/src/local_deep_research/defaults/llm_config.py

@@ -115,38 +115,21 @@ def get_llm(model_name=None, temperature=None, provider=None):
             return get_fallback_model(temperature)
     
     elif provider == "lmstudio":
-        try:
-            # Import LM Studio package
-            import lmstudio
-            from langchain_core.language_models import BaseLLM
-            
-            # Get LM Studio URL from settings
+
+            # LM Studio supports OpenAI API format, so we can use ChatOpenAI directly
             lmstudio_url = settings.llm.get('lmstudio_url', "http://localhost:1234")
             
-            # Create LM Studio LLM instance
-            model = lmstudio.llm(model_name)
-            
-            # Return a LangChain compatible wrapper
-            class LMStudioLLM(BaseLLM):
-                def _call(self, prompt, stop=None, **kwargs):
-                    result = model.complete(prompt, temperature=temperature)
-                    return result.completion
-                    
-                @property
-                def _identifying_params(self):
-                    return {"model_name": model_name}
-                    
-                @property
-                def _llm_type(self):
-                    return "lmstudio"
-                    
-            return LMStudioLLM()
-        except ImportError:
-            logger.error("LM Studio package not installed. Run 'pip install lmstudio'")
-            raise ImportError("LM Studio package not installed. Run 'pip install lmstudio'")
-    
+            return ChatOpenAI(
+                model=model_name,
+                api_key="lm-studio",  # LM Studio doesn't require a real API key
+                base_url=f"{lmstudio_url}/v1",  # Use the configured URL with /v1 endpoint
+                temperature=temperature,
+                max_tokens=settings.llm.max_tokens
+            )
+
+ 
     elif provider == "llamacpp":
-        try:
+
             # Import LlamaCpp
             from langchain_community.llms import LlamaCpp
             
@@ -171,9 +154,6 @@ def get_llm(model_name=None, temperature=None, provider=None):
                 f16_kv=f16_kv,
                 verbose=True
             )
-        except ImportError:
-            logger.error("LlamaCpp package not installed. Run 'pip install llama-cpp-python'")
-            raise ImportError("LlamaCpp package not installed. Run 'pip install llama-cpp-python'")
     
     else:
         return get_fallback_model(temperature)
@@ -275,11 +255,10 @@ def is_vllm_available():
 def is_lmstudio_available():
     """Check if LM Studio is available"""
     try:
-        import lmstudio
         import requests
         lmstudio_url = settings.llm.get('lmstudio_url', 'http://localhost:1234')
-        # Try to connect to check if running
-        response = requests.get(f"{lmstudio_url}/health", timeout=1.0)
+        # LM Studio typically uses OpenAI-compatible endpoints
+        response = requests.get(f"{lmstudio_url}/v1/models", timeout=1.0)
         return response.status_code == 200
     except:
         return False

{local_deep_research-0.1.19 → local_deep_research-0.1.20/src/local_deep_research.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: local-deep-research
-Version: 0.1.19
+Version: 0.1.20
 Summary: AI-powered research assistant with deep, iterative analysis using LLMs and web searches
 Author-email: LearningCircuit <185559241+LearningCircuit@users.noreply.github.com>, HashedViking <6432677+HashedViking@users.noreply.github.com>
 License: MIT License